src/test/java/com/android/tools/r8/assistant/ReflectiveCallExtractorTest.java

// 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.assistant;

import static com.android.tools.r8.assistant.ReflectiveCallExtractor.extractReflectiveCalls;
import static com.android.tools.r8.assistant.ReflectiveCallExtractor.printMethods;

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.dump.CompilerDump;
import com.android.tools.r8.graph.DexItemFactory;
import com.android.tools.r8.graph.DexMethod;
import com.android.tools.r8.graph.DexType;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Map;
import java.util.Set;
import org.junit.Assert;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.junit.runners.Parameterized;
import org.junit.runners.Parameterized.Parameter;
import org.junit.runners.Parameterized.Parameters;

@RunWith(Parameterized.class)
public class ReflectiveCallExtractorTest extends TestBase {

  @Parameter(0)
  public TestParameters parameters;

  @Parameters(name = "{0}")
  public static TestParametersCollection data() {
    return getTestParameters().withNoneRuntime().build();
  }

  @Test
  public void testGson() throws Exception {
    test(ToolHelper.GSON, 20, 8);
  }

  @Test
  public void testGuava() throws Exception {
    test(ToolHelper.GUAVA_JRE, 28, 11);
  }

  @Test
  public void testJacoco() throws Exception {
    test(ToolHelper.JACOCO_AGENT, 12, 0);
  }

  @Test
  public void testNowInAndroid() throws Exception {
    Path zip =
        Paths.get(
            ToolHelper.THIRD_PARTY_DIR,
            "opensource-apps",
            "android",
            "nowinandroid",
            "dump_app.zip");
    Path programArchive =
        CompilerDump.fromArchive(zip, temp.newFolder().toPath()).getProgramArchive();
    test(programArchive, 35, 18);
  }

  private void test(Path jar, int success, int failure) throws Exception {
    DexItemFactory factory = new DexItemFactory();
    Map<DexType, Collection<DexMethod>> reflectiveMethods = extractReflectiveCalls(jar, factory);
    Set<DexMethod> instrumentedMethodsForTesting =
        new InstrumentedReflectiveMethodList(factory).getInstrumentedMethodsForTesting();
    int supported = 0;
    int unsupported = 0;
    for (DexType dexType : reflectiveMethods.keySet()) {
      Collection<DexMethod> methods = reflectiveMethods.get(dexType);
      ArrayList<DexMethod> toRemove = new ArrayList<>();
      for (DexMethod dexMethod : methods) {
        if (instrumentedMethodsForTesting.contains(dexMethod)) {
          toRemove.add(dexMethod);
          supported++;
        } else {
          unsupported++;
        }
      }
      methods.removeAll(toRemove);
    }
    Assert.assertEquals(success, supported);
    Assert.assertEquals("Missing :\n" + printMethods(reflectiveMethods), failure, unsupported);
  }
}