src/test/java/com/android/tools/r8/inspection/InspectionApiTest.java

// Copyright (c) 2020, 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.inspection;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNull;

import com.android.tools.r8.TestBase;
import com.android.tools.r8.TestParameters;
import com.android.tools.r8.TestParametersCollection;
import com.android.tools.r8.inspector.Inspector;
import com.android.tools.r8.references.ClassReference;
import com.android.tools.r8.references.FieldReference;
import com.android.tools.r8.references.MethodReference;
import com.android.tools.r8.references.Reference;
import com.android.tools.r8.utils.StringUtils;
import java.util.Collections;
import java.util.HashSet;
import java.util.Set;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.junit.runners.Parameterized;

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

  static final String EXPECTED = StringUtils.lines("Hello, world");

  private final TestParameters parameters;

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

  public InspectionApiTest(TestParameters parameters) {
    this.parameters = parameters;
  }

  @Test
  public void testD8() throws Exception {
    if (parameters.isCfRuntime()) {
      testForJvm()
          .addProgramClasses(TestClass.class)
          .run(parameters.getRuntime(), TestClass.class)
          .assertSuccessWithOutput(EXPECTED);
      return;
    }
    testForD8()
        .addProgramClasses(TestClass.class)
        .setMinApi(parameters.getApiLevel())
        .apply(b -> b.getBuilder().addOutputInspection(this::inspection))
        .run(parameters.getRuntime(), TestClass.class)
        .assertSuccessWithOutput(EXPECTED);
    assertFound(false);
  }

  @Test
  public void testR8() throws Exception {
    testForR8(parameters.getBackend())
        .addProgramClasses(TestClass.class)
        .addKeepClassAndMembersRules(TestClass.class)
        .setMinApi(parameters.getApiLevel())
        .apply(b -> b.getBuilder().addOutputInspection(this::inspection))
        .run(parameters.getRuntime(), TestClass.class)
        .assertSuccessWithOutput(EXPECTED);
    assertFound(true);
  }

  ClassReference foundClass = null;
  FieldReference foundField = null;
  Set<MethodReference> foundMethods = new HashSet<>();

  private void inspection(Inspector inspector) {
    inspector.forEachClass(
        classInspector -> {
          assertNull(foundClass);
          foundClass = classInspector.getClassReference();
          classInspector.forEachField(
              fieldInspector -> {
                assertNull(foundField);
                foundField = fieldInspector.getFieldReference();
              });
          classInspector.forEachMethod(
              methodInspector -> {
                foundMethods.add(methodInspector.getMethodReference());
              });
        });
  }

  private void assertFound(boolean isR8) throws Exception {
    assertEquals(Reference.classFromClass(TestClass.class), foundClass);
    assertEquals(Reference.fieldFromField(TestClass.class.getDeclaredField("foo")), foundField);

    Set<MethodReference> expectedMethods = new HashSet<>();
    expectedMethods.add(
        Reference.methodFromMethod(TestClass.class.getDeclaredMethod("main", String[].class)));
    expectedMethods.add(Reference.methodFromMethod(TestClass.class.getDeclaredConstructor()));
    if (!isR8) {
      expectedMethods.add(
          Reference.method(
              Reference.classFromClass(TestClass.class),
              "<clinit>",
              Collections.emptyList(),
              null));
    }
    assertEquals(expectedMethods, foundMethods);
  }

  static class TestClass {
    public static int foo = 42;

    public static void main(String[] args) {
      System.out.println("Hello, world");
    }
  }
}