| // 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.d8; |
| |
| import static org.junit.Assert.assertEquals; |
| import static org.junit.Assert.assertTrue; |
| |
| import com.android.tools.r8.D8TestBuilder; |
| import com.android.tools.r8.OutputMode; |
| import com.android.tools.r8.TestBase; |
| import com.android.tools.r8.TestParameters; |
| import com.android.tools.r8.TestParametersCollection; |
| import com.android.tools.r8.utils.AndroidApiLevel; |
| import com.android.tools.r8.utils.FileUtils; |
| import com.google.common.collect.ImmutableList; |
| import java.nio.file.Files; |
| import java.nio.file.Path; |
| import java.util.HashMap; |
| import java.util.Map; |
| import org.junit.Test; |
| import org.junit.runner.RunWith; |
| import org.junit.runners.Parameterized; |
| |
| @RunWith(Parameterized.class) |
| public class ProguardMapSortByTest extends TestBase { |
| private final TestParameters parameters; |
| |
| @Parameterized.Parameters(name = "{0}") |
| public static TestParametersCollection data() { |
| return getTestParameters().withDexRuntimes().withAllApiLevels().build(); |
| } |
| |
| public ProguardMapSortByTest(TestParameters parameters) { |
| this.parameters = parameters; |
| } |
| |
| @Test |
| public void testMapSorting() throws Exception { |
| Path mappingFile = temp.newFile().toPath(); |
| // Sort the classes, so that they are in 3 packages _not_ sorted by their original name. |
| FileUtils.writeTextFile( |
| mappingFile, |
| "com.A.a -> " + C.class.getTypeName() + ":", |
| "com.A.b -> " + B.class.getTypeName() + ":", |
| "com.B.a -> " + D.class.getTypeName() + ":", |
| "com.B.b -> " + E.class.getTypeName() + ":", |
| "com.C.a -> " + A.class.getTypeName() + ":"); |
| D8TestBuilder d8TestBuilder = |
| testForD8() |
| .addOptionsModification( |
| internalOptions -> internalOptions.testing.limitNumberOfClassesPerDex = 2); |
| if (parameters.getApiLevel().isLessThan(AndroidApiLevel.L)) { |
| d8TestBuilder.addMainDexListClasses(A.class); |
| } |
| Path outputDir = temp.newFolder().toPath(); |
| d8TestBuilder |
| .release() |
| .addProgramClasses(A.class, B.class, C.class, D.class, E.class) |
| .setMinApi(parameters.getApiLevel()) |
| .apply(b -> b.getBuilder().setProguardInputMapFile(mappingFile)) |
| .run(parameters.getRuntime(), A.class) |
| .assertSuccessWithOutputLines("Hello world!") |
| .app() |
| .writeToDirectory(outputDir, OutputMode.DexIndexed); |
| Map<String, String> mapping = new HashMap<>(); |
| for (String dexFile : ImmutableList.of("classes.dex", "classes2.dex", "classes3.dex")) { |
| Path resolvedDexFile = outputDir.resolve(dexFile); |
| assertTrue(Files.exists(resolvedDexFile)); |
| classNamesFromDexFile(resolvedDexFile).forEach(name -> mapping.put(name, dexFile)); |
| } |
| // Check that the classes are grouped by their original package name. |
| assertEquals(mapping.get(B.class.getTypeName()), mapping.get(C.class.getTypeName())); |
| assertEquals(mapping.get(D.class.getTypeName()), mapping.get(E.class.getTypeName())); |
| assertTrue( |
| mapping.values().stream().filter(s -> s.equals(mapping.get(A.class.getTypeName()))).count() |
| == 1); |
| } |
| |
| static class A { |
| public static void main(String[] args) { |
| System.out.println("Hello world!"); |
| } |
| } |
| |
| static class B { |
| public static void foo() { |
| System.out.println("foo"); |
| } |
| } |
| |
| static class C { |
| public static void bar() { |
| System.out.println("bar"); |
| } |
| } |
| |
| static class D { |
| public static void foobar() { |
| System.out.println("foobar"); |
| } |
| } |
| |
| static class E { |
| public static void barfoo() { |
| System.out.println("barfoo"); |
| } |
| } |
| } |