src/test/java/com/android/tools/r8/dex/container/LegacyResourceShrinkerTest.java

// Copyright (c) 2023, 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.dex.container;

import static org.hamcrest.CoreMatchers.containsString;
import static org.hamcrest.MatcherAssert.assertThat;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertTrue;

import com.android.tools.r8.ResourceShrinker;
import com.android.tools.r8.ResourceShrinker.ReferenceChecker;
import com.android.tools.r8.TestParameters;
import com.android.tools.r8.ToolHelper;
import com.android.tools.r8.utils.AndroidApiLevel;
import com.android.tools.r8.utils.BooleanUtils;
import com.android.tools.r8.utils.IntBox;
import java.nio.file.Path;
import java.util.List;
import org.junit.BeforeClass;
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 LegacyResourceShrinkerTest extends DexContainerFormatTestBase {

  @Parameter(0)
  public TestParameters parameters;

  @Parameter(1)
  public boolean enableDexContainerInResourceShrinker;

  @Parameters(name = "{0}, enableDexContainerInResourceShrinker: {1}")
  public static List<Object[]> data() {
    return buildParameters(getTestParameters().withNoneRuntime().build(), BooleanUtils.values());
  }

  private static Path inputA;
  private static Path inputB;

  @BeforeClass
  public static void generateTestApplications() throws Throwable {
    // Build two applications in different packages both with required multidex due to number
    // of methods.
    inputA = getStaticTemp().getRoot().toPath().resolve("application_a.jar");
    inputB = getStaticTemp().getRoot().toPath().resolve("application_b.jar");

    generateApplication(inputA, "a", 10);
    generateApplication(inputB, "b", 10);
  }

  @Test
  public void test() throws Exception {
    Path outputBoth =
        testForD8(Backend.DEX)
            .addProgramFiles(inputA, inputB)
            .setMinApi(AndroidApiLevel.L)
            .addOptionsModification(
                options -> options.getTestingOptions().dexContainerExperiment = true)
            .compile()
            .writeToZip();
    validateSingleContainerDex(outputBoth);
    ResourceShrinker.Command command =
        new ResourceShrinker.Builder().addProgramFiles(outputBoth).build();

    try {
      IntBox classCount = new IntBox();
      ToolHelper.runLegacyResourceShrinker(
          new ResourceShrinker.Builder().addProgramFiles(outputBoth),
          options -> options.testing.dexContainerExperiment = enableDexContainerInResourceShrinker,
          new ReferenceChecker() {
            @Override
            public boolean shouldProcess(String internalName) {
              classCount.increment();
              return true;
            }

            @Override
            public void referencedInt(int value) {}

            @Override
            public void referencedString(String value) {}

            @Override
            public void referencedStaticField(String internalName, String fieldName) {}

            @Override
            public void referencedMethod(
                String internalName, String methodName, String methodDescriptor) {}
          });
      assertTrue(enableDexContainerInResourceShrinker);
      assertEquals(20000, classCount.get());
    } catch (RuntimeException e) {
      assertThat(
          e.getMessage(),
          containsString("Experimental container DEX version V41 is not supported"));
      assertFalse(enableDexContainerInResourceShrinker);
    }
  }
}