src/test/java/com/android/tools/r8/compilerapi/desugardependencies/DesugarDependenciesTest.java

// Copyright (c) 2022, 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.compilerapi.desugardependencies;

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

import com.android.tools.r8.D8;
import com.android.tools.r8.D8Command;
import com.android.tools.r8.DesugarGraphConsumer;
import com.android.tools.r8.DexIndexedConsumer;
import com.android.tools.r8.TestParameters;
import com.android.tools.r8.compilerapi.CompilerApiTest;
import com.android.tools.r8.compilerapi.CompilerApiTestRunner;
import com.android.tools.r8.origin.Origin;
import java.util.Set;
import java.util.concurrent.ConcurrentHashMap;
import org.junit.Test;

public class DesugarDependenciesTest extends CompilerApiTestRunner {

  public DesugarDependenciesTest(TestParameters parameters) {
    super(parameters);
  }

  @Override
  public Class<? extends CompilerApiTest> binaryTestClass() {
    return ApiTest.class;
  }

  @Test
  public void testDesugarDependencies() throws Exception {
    ApiTest test = new ApiTest(ApiTest.PARAMETERS);
    runTest(test::run);
  }

  private interface Runner {
    void run() throws Exception;
  }

  private void runTest(Runner test) throws Exception {
    test.run();
  }

  public static class ApiTest extends CompilerApiTest {

    public ApiTest(Object parameters) {
      super(parameters);
    }

    public void run() throws Exception {
      D8.run(
          D8Command.builder()
              .addClassProgramData(getBytesForClass(getMockClass()), Origin.unknown())
              .addLibraryFiles(getJava8RuntimeJar())
              .setProgramConsumer(DexIndexedConsumer.emptyConsumer())
              .setDesugarGraphConsumer(
                  new DesugarGraphConsumer() {
                    private final Set<Origin> desugaringUnit = ConcurrentHashMap.newKeySet();

                    @Override
                    public void acceptProgramNode(Origin node) {
                      desugaringUnit.add(node);
                    }

                    @Override
                    public void accept(Origin dependent, Origin dependency) {
                      assertTrue(desugaringUnit.contains(dependent));
                    }

                    @Override
                    public void finished() {
                      // Input unit contains just the mock class.
                      assertEquals(1, desugaringUnit.size());
                    }
                  })
              .build());
    }

    @Test
    public void testDesugarDependencies() throws Exception {
      run();
    }
  }
}