src/test/java/com/android/tools/r8/shaking/ClassKindTest.java

// Copyright (c) 2017, 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.shaking;

import com.android.tools.r8.TestBase;
import com.android.tools.r8.shaking.classkinds.Annotation;
import com.android.tools.r8.shaking.classkinds.Class;
import com.android.tools.r8.shaking.classkinds.Enum;
import com.android.tools.r8.shaking.classkinds.Interface;
import com.android.tools.r8.utils.AndroidApp;
import com.android.tools.r8.utils.codeinspector.CodeInspector;
import com.google.common.collect.ImmutableList;
import com.google.common.collect.Sets;
import java.util.Collection;
import java.util.HashSet;
import java.util.List;
import org.junit.Assert;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.junit.runners.Parameterized;
import org.junit.runners.Parameterized.Parameters;

@RunWith(Parameterized.class)
public class ClassKindTest extends TestBase {

  private static List<java.lang.Class> CLASSES_TO_INCLUDE = ImmutableList.of(
      Annotation.class, Class.class, Enum.class, Interface.class);

  public ClassKindTest(String config, List<java.lang.Class<?>> classes) {
    this.config = config;
    this.classes = classes;
  }

  @Parameters(name = "{0}")
  public static Collection<Object[]> parameters() {
    return ImmutableList.copyOf(new Object[][]{
        {"-keep interface *", ImmutableList.of(Interface.class)},
        {"-keep class *", CLASSES_TO_INCLUDE},
        {"-keep enum *", ImmutableList.of(Enum.class)},
        {"-keep @interface *", ImmutableList.of(Annotation.class)},
        {"-keep !interface *", ImmutableList.of(Enum.class, Annotation.class, Class.class)},
        {"-keep !enum *", ImmutableList.of(Interface.class, Annotation.class, Class.class)},
        {"-keep !@interface *", ImmutableList.of(Interface.class, Enum.class, Class.class)},
        {"-keep !class *", ImmutableList.of()}
    });
  }

  public final String config;
  public final List<java.lang.Class<?>> classes;

  @Test
  public void run() throws Exception {
    AndroidApp app;
    try {
      app = compileWithR8(CLASSES_TO_INCLUDE, config);
    } catch (AssertionError e) {
      // Compilation will fail if we produce no classes.
      Assert.assertTrue(classes.isEmpty());
      return;
    }
    CodeInspector inspector = new CodeInspector(app);
    HashSet<java.lang.Class<?>> expected = Sets.newHashSet(classes);
    CLASSES_TO_INCLUDE.forEach(c -> {
      Assert.assertEquals(expected.contains(c), inspector.clazz(c).isPresent());
    });
  }
}