| // 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 org.junit.Assert.assertEquals; |
| import static org.junit.Assert.assertFalse; |
| import static org.junit.Assert.assertTrue; |
| import static org.junit.Assume.assumeTrue; |
| |
| import com.android.tools.r8.utils.codeinspector.ClassSubject; |
| import com.android.tools.r8.utils.codeinspector.CodeInspector; |
| import java.util.concurrent.atomic.AtomicInteger; |
| import org.junit.Test; |
| |
| public class KotlinClassStaticizerTest extends AbstractR8KotlinTestBase { |
| |
| @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); |
| assertTrue(inspector.clazz("class_staticizer.Regular$Companion").isPresent()); |
| assertTrue(inspector.clazz("class_staticizer.Derived$Companion").isPresent()); |
| |
| ClassSubject utilClass = inspector.clazz("class_staticizer.Util"); |
| assertTrue(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); |
| assertFalse(inspector.clazz("class_staticizer.Regular$Companion").isPresent()); |
| assertFalse(inspector.clazz("class_staticizer.Derived$Companion").isPresent()); |
| |
| ClassSubject utilClass = inspector.clazz("class_staticizer.Util"); |
| assertTrue(utilClass.isPresent()); |
| utilClass.forAllMethods(method -> assertTrue(method.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; |
| }, inspector); |
| } |
| } |