src/test/java/com/android/tools/r8/naming/applymapping/ApplyMappingSignatureTest.java

// Copyright (c) 2019, 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.naming.applymapping;

import static com.android.tools.r8.utils.codeinspector.Matchers.isPresent;
import static org.hamcrest.MatcherAssert.assertThat;
import static org.hamcrest.core.StringContains.containsString;

import com.android.tools.r8.CompilationFailedException;
import com.android.tools.r8.R8TestBuilder;
import com.android.tools.r8.TestBase;
import com.android.tools.r8.ToolHelper;
import com.android.tools.r8.utils.codeinspector.ClassSubject;
import com.android.tools.r8.utils.codeinspector.FieldSubject;
import java.io.IOException;
import java.util.Set;
import java.util.concurrent.ExecutionException;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.junit.runners.Parameterized;

/** This test reproduces b/132593471 on d8-1.4. */
@RunWith(Parameterized.class)
public class ApplyMappingSignatureTest extends TestBase {

  public interface A {}

  public static class SignatureTest {
    Set<A> field;

    public static void main(String[] args) {
      System.out.print("HELLO");
    }
  }

  @Parameterized.Parameters(name = "{0}")
  public static Backend[] data() {
    return Backend.values();
  }

  private Backend backend;

  public ApplyMappingSignatureTest(Backend backend) {
    this.backend = backend;
  }

  @Test
  public void testApplyMappingWithSignatureRenaming()
      throws IOException, CompilationFailedException, ExecutionException {
    R8TestBuilder r8TestBuilder =
        testForR8(backend)
            .addProgramClasses(A.class, SignatureTest.class)
            .addKeepAllClassesRule()
            .addKeepMainRule(SignatureTest.class)
            .addKeepAttributes("Signature", "InnerClasses", "EnclosingMethod")
            .addApplyMapping(A.class.getTypeName() + " -> foo:");
    if (backend == Backend.DEX) {
      r8TestBuilder.setMinApi(ToolHelper.getMinApiLevelForDexVm());
    }
    r8TestBuilder
        .run(SignatureTest.class)
        .assertSuccessWithOutput("HELLO")
        .inspect(
            inspector -> {
              ClassSubject clazz = inspector.clazz(SignatureTest.class);
              assertThat(clazz, isPresent());
              FieldSubject field = clazz.uniqueFieldWithName("field");
              assertThat(field, isPresent());
              assertThat(field.getSignatureAnnotationValue(), containsString("foo"));
            });
  }
}