src/test/java/com/android/tools/r8/kotlin/KotlinClassStaticizerTest.java

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

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 static org.junit.Assume.assumeTrue;

import com.android.tools.r8.ToolHelper.KotlinTargetVersion;
import com.android.tools.r8.utils.codeinspector.ClassSubject;
import com.android.tools.r8.utils.codeinspector.CodeInspector;
import com.android.tools.r8.utils.codeinspector.FoundMethodSubject;
import java.util.concurrent.atomic.AtomicInteger;
import org.junit.Test;

public class KotlinClassStaticizerTest extends AbstractR8KotlinTestBase {

  public KotlinClassStaticizerTest(
      KotlinTargetVersion targetVersion, boolean allowAccessModification) {
    super(targetVersion, allowAccessModification);
  }

  @Test
  public void testCompanionAndRegularObjects() throws Exception {
    assumeTrue("Only work with -allowaccessmodification", allowAccessModification);
    final String mainClassName = "class_staticizer.MainKt";

    // Without class staticizer.
    runTest(
        "class_staticizer",
        mainClassName,
        false,
        app -> {
          CodeInspector inspector = new CodeInspector(app);
          assertThat(inspector.clazz("class_staticizer.Regular$Companion"), isPresent());
          assertThat(inspector.clazz("class_staticizer.Derived$Companion"), isPresent());

          ClassSubject utilClass = inspector.clazz("class_staticizer.Util");
          assertThat(utilClass, isPresent());
          AtomicInteger nonStaticMethodCount = new AtomicInteger();
          utilClass.forAllMethods(
              method -> {
                if (!method.isStatic()) {
                  nonStaticMethodCount.incrementAndGet();
                }
              });
          assertEquals(4, nonStaticMethodCount.get());
        });

    // With class staticizer.
    runTest(
        "class_staticizer",
        mainClassName,
        true,
        app -> {
          CodeInspector inspector = new CodeInspector(app);
          assertThat(inspector.clazz("class_staticizer.Regular$Companion"), not(isPresent()));
          assertThat(inspector.clazz("class_staticizer.Derived$Companion"), not(isPresent()));

          ClassSubject utilClass = inspector.clazz("class_staticizer.Util");
          assertThat(utilClass, isPresent());
          assertTrue(utilClass.allMethods().stream().allMatch(FoundMethodSubject::isStatic));
        });
  }

  protected void runTest(String folder, String mainClass,
      boolean enabled, AndroidAppInspector inspector) throws Exception {
    runTest(
        folder,
        mainClass,
        null,
        options -> {
          options.enableClassInlining = false;
          options.enableClassStaticizer = enabled;
          options.enableInliningOfInvokesWithNullableReceivers = false;
        },
        inspector);
  }
}