src/test/java/com/android/tools/r8/benchmarks/appdumps/ComposeSamplesBenchmarks.java

// Copyright (c) 2024, 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.benchmarks.appdumps;

import static com.android.tools.r8.DiagnosticsMatcher.diagnosticMessage;
import static com.android.tools.r8.DiagnosticsMatcher.diagnosticType;
import static org.hamcrest.CoreMatchers.allOf;
import static org.hamcrest.CoreMatchers.containsString;

import com.android.tools.r8.R8FullTestBuilder;
import com.android.tools.r8.R8PartialTestBuilder;
import com.android.tools.r8.R8PartialTestCompileResult;
import com.android.tools.r8.TestParameters;
import com.android.tools.r8.ToolHelper;
import com.android.tools.r8.benchmarks.BenchmarkBase;
import com.android.tools.r8.benchmarks.BenchmarkConfig;
import com.android.tools.r8.errors.InterfaceDesugarMissingTypeDiagnostic;
import com.google.common.collect.ImmutableList;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.List;
import org.junit.Ignore;
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 ComposeSamplesBenchmarks extends BenchmarkBase {

  private static final Path dir =
      Paths.get(ToolHelper.THIRD_PARTY_DIR, "opensource-apps/android/compose-samples");

  public ComposeSamplesBenchmarks(BenchmarkConfig config, TestParameters parameters) {
    super(config, parameters);
  }

  @Parameters(name = "{0}")
  public static List<Object[]> data() {
    return parametersFromConfigs(configs());
  }

  public static List<BenchmarkConfig> configs() {
    return ImmutableList.of(
        AppDumpBenchmarkBuilder.builder()
            .setName("CraneApp")
            .setDumpDependencyPath(dir.resolve("crane"))
            .setFromRevision(16457)
            .buildR8(),
        AppDumpBenchmarkBuilder.builder()
            .setName("CraneAppPartial")
            .setDumpDependencyPath(dir.resolve("crane"))
            .setFromRevision(16457)
            .buildR8WithPartialShrinking(),
        AppDumpBenchmarkBuilder.builder()
            .setName("JetLaggedApp")
            .setDumpDependencyPath(dir.resolve("jetlagged"))
            .setFromRevision(16457)
            .buildR8(),
        AppDumpBenchmarkBuilder.builder()
            .setName("JetLaggedAppPartial")
            .setDumpDependencyPath(dir.resolve("jetlagged"))
            .setFromRevision(16457)
            .buildR8WithPartialShrinking(),
        AppDumpBenchmarkBuilder.builder()
            .setName("JetNewsApp")
            .setDumpDependencyPath(dir.resolve("jetnews"))
            .setFromRevision(16457)
            .buildR8(),
        AppDumpBenchmarkBuilder.builder()
            .setName("JetNewsAppPartial")
            .setDumpDependencyPath(dir.resolve("jetnews"))
            .setFromRevision(16457)
            .buildR8WithPartialShrinking(),
        AppDumpBenchmarkBuilder.builder()
            .setName("JetCasterApp")
            .setDumpDependencyPath(dir.resolve("jetcaster"))
            .setFromRevision(16457)
            .buildR8(ComposeSamplesBenchmarks::configureJetCasterApp),
        AppDumpBenchmarkBuilder.builder()
            .setName("JetCasterAppPartial")
            .setDumpDependencyPath(dir.resolve("jetcaster"))
            .setFromRevision(16457)
            .buildR8WithPartialShrinking(
                ComposeSamplesBenchmarks::configureJetCasterAppPartial,
                ComposeSamplesBenchmarks::inspectJetCasterAppPartial),
        AppDumpBenchmarkBuilder.builder()
            .setName("JetChatApp")
            .setDumpDependencyPath(dir.resolve("jetchat"))
            .setFromRevision(16457)
            .buildR8(ComposeSamplesBenchmarks::configureJetChatApp),
        AppDumpBenchmarkBuilder.builder()
            .setName("JetChatAppPartial")
            .setDumpDependencyPath(dir.resolve("jetchat"))
            .setFromRevision(16457)
            .buildR8WithPartialShrinking(
                ComposeSamplesBenchmarks::configureJetChatAppPartial,
                ComposeSamplesBenchmarks::inspectJetChatAppPartial),
        AppDumpBenchmarkBuilder.builder()
            .setName("JetSnackApp")
            .setDumpDependencyPath(dir.resolve("jetsnack"))
            .setFromRevision(16457)
            .buildR8(),
        AppDumpBenchmarkBuilder.builder()
            .setName("JetSnackAppPartial")
            .setDumpDependencyPath(dir.resolve("jetsnack"))
            .setFromRevision(16457)
            .buildR8WithPartialShrinking(),
        AppDumpBenchmarkBuilder.builder()
            .setName("OwlApp")
            .setDumpDependencyPath(dir.resolve("owl"))
            .setFromRevision(16457)
            .buildR8(),
        AppDumpBenchmarkBuilder.builder()
            .setName("OwlAppPartial")
            .setDumpDependencyPath(dir.resolve("owl"))
            .setFromRevision(16457)
            .buildR8WithPartialShrinking(),
        AppDumpBenchmarkBuilder.builder()
            .setName("ReplyApp")
            .setDumpDependencyPath(dir.resolve("reply"))
            .setFromRevision(16457)
            .buildR8(),
        AppDumpBenchmarkBuilder.builder()
            .setName("ReplyAppPartial")
            .setDumpDependencyPath(dir.resolve("reply"))
            .setFromRevision(16457)
            .buildR8WithPartialShrinking());
  }

  private static void configureJetCasterApp(R8FullTestBuilder testBuilder) {
    testBuilder
        .addDontWarn(
            "org.bouncycastle.jsse.BCSSLParameters",
            "org.bouncycastle.jsse.BCSSLSocket",
            "org.bouncycastle.jsse.provider.BouncyCastleJsseProvider",
            "org.conscrypt.Conscrypt",
            "org.conscrypt.Conscrypt$Version",
            "org.openjsse.javax.net.ssl.SSLParameters",
            "org.openjsse.javax.net.ssl.SSLSocket",
            "org.openjsse.net.ssl.OpenJSSE",
            "org.slf4j.impl.StaticLoggerBinder")
        .allowDiagnosticInfoMessages()
        .allowUnnecessaryDontWarnWildcards()
        .allowUnusedDontWarnPatterns()
        .allowUnusedProguardConfigurationRules()
        .addOptionsModification(
            options -> {
              options.getCfCodeAnalysisOptions().setAllowUnreachableCfBlocks(true);
              options.getOpenClosedInterfacesOptions().suppressAllOpenInterfaces();
            });
  }

  private static void configureJetCasterAppPartial(R8PartialTestBuilder testBuilder) {
    testBuilder
        .allowDiagnosticMessages()
        .allowUnnecessaryDontWarnWildcards()
        .allowUnusedDontWarnPatterns()
        .allowUnusedProguardConfigurationRules()
        .addR8PartialOptionsModification(
            options -> options.getOpenClosedInterfacesOptions().disallowOpenInterfaces());
  }

  private static void inspectJetCasterAppPartial(R8PartialTestCompileResult compileResult) {
    compileResult.inspectDiagnosticMessages(
        diagnostics ->
            diagnostics
                .assertWarningsMatch(
                    allOf(
                        diagnosticType(InterfaceDesugarMissingTypeDiagnostic.class),
                        diagnosticMessage(
                            containsString("androidx.compose.animation.tooling.ComposeAnimation"))),
                    allOf(
                        diagnosticType(InterfaceDesugarMissingTypeDiagnostic.class),
                        diagnosticMessage(containsString("androidx.paging.PositionalDataSource"))),
                    allOf(
                        diagnosticType(InterfaceDesugarMissingTypeDiagnostic.class),
                        diagnosticMessage(
                            containsString("java.lang.instrument.ClassFileTransformer"))),
                    allOf(
                        diagnosticType(InterfaceDesugarMissingTypeDiagnostic.class),
                        diagnosticMessage(
                            containsString("org.conscrypt.ConscryptHostnameVerifier"))),
                    allOf(
                        diagnosticType(InterfaceDesugarMissingTypeDiagnostic.class),
                        diagnosticMessage(containsString("org.jaxen.DefaultNavigator"))),
                    allOf(
                        diagnosticType(InterfaceDesugarMissingTypeDiagnostic.class),
                        diagnosticMessage(containsString("org.jaxen.NamespaceContext"))),
                    allOf(
                        diagnosticType(InterfaceDesugarMissingTypeDiagnostic.class),
                        diagnosticMessage(containsString("org.jaxen.VariableContext"))),
                    allOf(
                        diagnosticType(InterfaceDesugarMissingTypeDiagnostic.class),
                        diagnosticMessage(containsString("sun.misc.SignalHandler"))))
                .assertNoErrors());
  }

  private static void configureJetChatApp(R8FullTestBuilder testBuilder) {
    testBuilder
        .allowDiagnosticInfoMessages()
        .allowUnnecessaryDontWarnWildcards()
        .allowUnusedDontWarnPatterns()
        .allowUnusedProguardConfigurationRules();
  }

  private static void configureJetChatAppPartial(R8PartialTestBuilder testBuilder) {
    testBuilder
        .allowDiagnosticMessages()
        .allowUnnecessaryDontWarnWildcards()
        .allowUnusedDontWarnPatterns()
        .allowUnusedProguardConfigurationRules();
  }

  private static void inspectJetChatAppPartial(R8PartialTestCompileResult compileResult) {
    compileResult.inspectDiagnosticMessages(
        diagnostics ->
            diagnostics
                .assertWarningsMatch(
                    allOf(
                        diagnosticType(InterfaceDesugarMissingTypeDiagnostic.class),
                        diagnosticMessage(
                            containsString("androidx.compose.animation.tooling.ComposeAnimation"))),
                    allOf(
                        diagnosticType(InterfaceDesugarMissingTypeDiagnostic.class),
                        diagnosticMessage(
                            containsString("java.lang.instrument.ClassFileTransformer"))),
                    allOf(
                        diagnosticType(InterfaceDesugarMissingTypeDiagnostic.class),
                        diagnosticMessage(containsString("sun.misc.SignalHandler"))))
                .assertNoErrors());
  }

  @Ignore
  @Test
  @Override
  public void testBenchmarks() throws Exception {
    super.testBenchmarks();
  }

  @Test
  public void testCraneApp() throws Exception {
    testBenchmarkWithName("CraneApp");
  }

  @Test
  public void testCraneAppPartial() throws Exception {
    testBenchmarkWithName("CraneAppPartial");
  }

  @Test
  public void testJetLaggedApp() throws Exception {
    testBenchmarkWithName("JetLaggedApp");
  }

  @Test
  public void testJetLaggedAppPartial() throws Exception {
    testBenchmarkWithName("JetLaggedAppPartial");
  }

  @Test
  public void testJetNewsApp() throws Exception {
    testBenchmarkWithName("JetNewsApp");
  }

  @Test
  public void testJetNewsAppPartial() throws Exception {
    testBenchmarkWithName("JetNewsAppPartial");
  }

  @Test
  public void testJetCasterApp() throws Exception {
    testBenchmarkWithName("JetCasterApp");
  }

  @Test
  public void testJetCasterAppPartial() throws Exception {
    testBenchmarkWithName("JetCasterAppPartial");
  }

  @Test
  public void testJetChatApp() throws Exception {
    testBenchmarkWithName("JetChatApp");
  }

  @Test
  public void testJetChatAppPartial() throws Exception {
    testBenchmarkWithName("JetChatAppPartial");
  }

  @Test
  public void testJetSnackApp() throws Exception {
    testBenchmarkWithName("JetSnackApp");
  }

  @Test
  public void testJetSnackAppPartial() throws Exception {
    testBenchmarkWithName("JetSnackAppPartial");
  }

  @Test
  public void testOwlApp() throws Exception {
    testBenchmarkWithName("OwlApp");
  }

  @Test
  public void testOwlAppPartial() throws Exception {
    testBenchmarkWithName("OwlAppPartial");
  }

  @Test
  public void testReplyApp() throws Exception {
    testBenchmarkWithName("ReplyApp");
  }

  @Test
  public void testReplyAppPartial() throws Exception {
    testBenchmarkWithName("ReplyAppPartial");
  }
}