Add a test for -checkdiscard
Bug: 162969014
Change-Id: I1bfef38c14c4a487d6a99cc0e02e485468c778e9
diff --git a/src/test/java/com/android/tools/r8/checkdiscarded/B162969014.java b/src/test/java/com/android/tools/r8/checkdiscarded/B162969014.java
new file mode 100644
index 0000000..6eb081f
--- /dev/null
+++ b/src/test/java/com/android/tools/r8/checkdiscarded/B162969014.java
@@ -0,0 +1,108 @@
+// 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.checkdiscarded;
+
+import static com.android.tools.r8.utils.codeinspector.Matchers.isPresent;
+import static org.hamcrest.CoreMatchers.not;
+import static org.hamcrest.MatcherAssert.assertThat;
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertTrue;
+
+import com.android.tools.r8.AssumeNoSideEffects;
+import com.android.tools.r8.CompilationFailedException;
+import com.android.tools.r8.NeverInline;
+import com.android.tools.r8.R8FullTestBuilder;
+import com.android.tools.r8.R8TestCompileResult;
+import com.android.tools.r8.TestBase;
+import com.android.tools.r8.TestParameters;
+import com.android.tools.r8.utils.BooleanUtils;
+import com.android.tools.r8.utils.codeinspector.ClassSubject;
+import com.android.tools.r8.utils.codeinspector.CodeInspector;
+import java.util.List;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.junit.runners.Parameterized;
+
+@RunWith(Parameterized.class)
+public class B162969014 extends TestBase {
+
+ private final boolean checkLogIsDiscarded;
+ private final TestParameters parameters;
+
+ @Parameterized.Parameters(name = "{1}, checkLogIsDiscarded: {0}")
+ public static List<Object[]> data() {
+ return buildParameters(
+ BooleanUtils.values(), getTestParameters().withAllRuntimesAndApiLevels().build());
+ }
+
+ public B162969014(boolean checkLogIsDiscarded, TestParameters parameters) {
+ this.checkLogIsDiscarded = checkLogIsDiscarded;
+ this.parameters = parameters;
+ }
+
+ @Test
+ public void test() throws Exception {
+ R8TestCompileResult compileResult;
+ try {
+ compileResult =
+ testForR8(parameters.getBackend())
+ .addInnerClasses(B162969014.class)
+ .addKeepMainRule(TestClass.class)
+ .apply(this::applyCheckDiscardedRule)
+ .enableAssumeNoSideEffectsAnnotations()
+ .enableInliningAnnotations()
+ .setMinApi(parameters.getApiLevel())
+ .compile();
+ } catch (CompilationFailedException e) {
+ assertTrue(checkLogIsDiscarded);
+ assertEquals(e.getCause().getMessage(), "Discard checks failed.");
+ return;
+ }
+
+ compileResult
+ .inspect(this::inspect)
+ .run(parameters.getRuntime(), TestClass.class)
+ .assertSuccessWithOutputLines("foo");
+ }
+
+ private void applyCheckDiscardedRule(R8FullTestBuilder builder) {
+ if (checkLogIsDiscarded) {
+ builder.addKeepRules(
+ "-checkdiscard class " + Log.class.getTypeName() + "{",
+ " public static void foo();",
+ " public static void bar();",
+ "}");
+ }
+ }
+
+ private void inspect(CodeInspector inspector) {
+ ClassSubject logClassSubject = inspector.clazz(Log.class);
+ assertThat(logClassSubject, isPresent());
+ assertThat(logClassSubject.uniqueMethodWithName("foo"), isPresent());
+ assertThat(logClassSubject.uniqueMethodWithName("bar"), not(isPresent()));
+ }
+
+ static class TestClass {
+
+ public static void main(String[] args) {
+ Log.foo();
+ Log.bar();
+ }
+ }
+
+ static class Log {
+
+ @NeverInline
+ public static void foo() {
+ System.out.println("foo");
+ }
+
+ @AssumeNoSideEffects
+ @NeverInline
+ public static void bar() {
+ System.out.println("bar");
+ }
+ }
+}
