Reproduce trace references missing keep rules

Bug: b/397904338
Change-Id: I84ad19aa8af162b8c1067df74fcbf18e8892ebb3
diff --git a/src/test/java/com/android/tools/r8/tracereferences/TraceReferencesIndirectOverrideTest.java b/src/test/java/com/android/tools/r8/tracereferences/TraceReferencesIndirectOverrideTest.java
new file mode 100644
index 0000000..cc255ed
--- /dev/null
+++ b/src/test/java/com/android/tools/r8/tracereferences/TraceReferencesIndirectOverrideTest.java
@@ -0,0 +1,76 @@
+// Copyright (c) 2025, the R8 project authors. Please see the AUTHORS file
+// for details. All rights reserved. Use of this source code is governed by a
+// BSD-style license that can be found in the LICENSE file.
+package com.android.tools.r8.tracereferences;
+
+import static org.hamcrest.CoreMatchers.containsString;
+import static org.hamcrest.CoreMatchers.not;
+import static org.hamcrest.MatcherAssert.assertThat;
+
+import com.android.tools.r8.TestBase;
+import com.android.tools.r8.TestParameters;
+import com.android.tools.r8.TestParametersCollection;
+import com.android.tools.r8.ToolHelper;
+import com.android.tools.r8.desugar.desugaredlibrary.test.TestingKeepRuleConsumer;
+import com.android.tools.r8.utils.NopDiagnosticsHandler;
+import com.android.tools.r8.utils.ZipUtils.ZipBuilder;
+import java.nio.file.Path;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.junit.runners.Parameterized;
+import org.junit.runners.Parameterized.Parameters;
+
+@RunWith(Parameterized.class)
+public class TraceReferencesIndirectOverrideTest extends TestBase {
+
+  @Parameters(name = "{0}")
+  public static TestParametersCollection data() {
+    return getTestParameters().withNoneRuntime().build();
+  }
+
+  public TraceReferencesIndirectOverrideTest(TestParameters parameters) {
+    parameters.assertNoneRuntime();
+  }
+
+  @Test
+  public void test() throws Throwable {
+    Path targetJar =
+        ZipBuilder.builder(temp.newFile("target.jar").toPath())
+            .addFilesRelative(
+                ToolHelper.getClassPathForTests(),
+                ToolHelper.getClassFileForTestClass(A.class),
+                ToolHelper.getClassFileForTestClass(B.class))
+            .build();
+    Path sourceJar =
+        ZipBuilder.builder(temp.newFile("source.jar").toPath())
+            .addFilesRelative(
+                ToolHelper.getClassPathForTests(), ToolHelper.getClassFileForTestClass(C.class))
+            .build();
+    TestingKeepRuleConsumer testingKeepRuleConsumer = new TestingKeepRuleConsumer();
+    TraceReferences.run(
+        TraceReferencesCommand.builder(new NopDiagnosticsHandler())
+            .addLibraryFiles(ToolHelper.getMostRecentAndroidJar())
+            .addSourceFiles(sourceJar)
+            .addTargetFiles(targetJar)
+            .setConsumer(
+                TraceReferencesKeepRules.builder()
+                    .setOutputConsumer(testingKeepRuleConsumer)
+                    .build())
+            .build());
+    // TODO(b/397904338): shouldBeKept() should be kept.
+    assertThat(testingKeepRuleConsumer.get(), not(containsString("shouldBeKept")));
+  }
+
+  static class A {
+
+    public void shouldBeKept() {}
+  }
+
+  static class B extends A {}
+
+  static class C extends B {
+
+    @Override
+    public void shouldBeKept() {}
+  }
+}
diff --git a/src/test/testbase/java/com/android/tools/r8/TestDiagnosticMessages.java b/src/test/testbase/java/com/android/tools/r8/TestDiagnosticMessages.java
index d20d678..38a72d4 100644
--- a/src/test/testbase/java/com/android/tools/r8/TestDiagnosticMessages.java
+++ b/src/test/testbase/java/com/android/tools/r8/TestDiagnosticMessages.java
@@ -19,7 +19,7 @@
 import java.util.List;
 import org.hamcrest.Matcher;
 
-public abstract class TestDiagnosticMessages {
+public abstract class TestDiagnosticMessages implements DiagnosticsHandler {
 
   public <E extends Exception> TestDiagnosticMessages applyIf(
       boolean condition,