|  | // Copyright 2017 Google Inc. All rights reserved. | 
|  | // | 
|  | // Licensed under the Apache License, Version 2.0 (the "License"); | 
|  | // you may not use this file except in compliance with the License. | 
|  | // You may obtain a copy of the License at | 
|  | // | 
|  | //     http://www.apache.org/licenses/LICENSE-2.0 | 
|  | // | 
|  | // Unless required by applicable law or agreed to in writing, software | 
|  | // distributed under the License is distributed on an "AS IS" BASIS, | 
|  | // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | 
|  | // See the License for the specific language governing permissions and | 
|  | // limitations under the License. | 
|  |  | 
|  | package java | 
|  |  | 
|  | import ( | 
|  | "strconv" | 
|  | "strings" | 
|  | "testing" | 
|  |  | 
|  | "android/soong/android" | 
|  | ) | 
|  |  | 
|  | func TestKotlin(t *testing.T) { | 
|  | ctx, _ := testJava(t, ` | 
|  | java_library { | 
|  | name: "foo", | 
|  | srcs: ["a.java", "b.kt"], | 
|  | } | 
|  |  | 
|  | java_library { | 
|  | name: "bar", | 
|  | srcs: ["b.kt"], | 
|  | libs: ["foo"], | 
|  | static_libs: ["baz"], | 
|  | } | 
|  |  | 
|  | java_library { | 
|  | name: "baz", | 
|  | srcs: ["c.java"], | 
|  | } | 
|  | `) | 
|  |  | 
|  | kotlinStdlib := ctx.ModuleForTests("kotlin-stdlib", "android_common"). | 
|  | Output("turbine-combined/kotlin-stdlib.jar").Output | 
|  | kotlinStdlibJdk7 := ctx.ModuleForTests("kotlin-stdlib-jdk7", "android_common"). | 
|  | Output("turbine-combined/kotlin-stdlib-jdk7.jar").Output | 
|  | kotlinStdlibJdk8 := ctx.ModuleForTests("kotlin-stdlib-jdk8", "android_common"). | 
|  | Output("turbine-combined/kotlin-stdlib-jdk8.jar").Output | 
|  | kotlinAnnotations := ctx.ModuleForTests("kotlin-annotations", "android_common"). | 
|  | Output("turbine-combined/kotlin-annotations.jar").Output | 
|  |  | 
|  | fooKotlinc := ctx.ModuleForTests("foo", "android_common").Rule("kotlinc") | 
|  | fooJavac := ctx.ModuleForTests("foo", "android_common").Rule("javac") | 
|  | fooJar := ctx.ModuleForTests("foo", "android_common").Output("combined/foo.jar") | 
|  | fooHeaderJar := ctx.ModuleForTests("foo", "android_common").Output("turbine-combined/foo.jar") | 
|  |  | 
|  | fooKotlincClasses := fooKotlinc.Output | 
|  | fooKotlincHeaderClasses := fooKotlinc.ImplicitOutput | 
|  |  | 
|  | if len(fooKotlinc.Inputs) != 2 || fooKotlinc.Inputs[0].String() != "a.java" || | 
|  | fooKotlinc.Inputs[1].String() != "b.kt" { | 
|  | t.Errorf(`foo kotlinc inputs %v != ["a.java", "b.kt"]`, fooKotlinc.Inputs) | 
|  | } | 
|  |  | 
|  | if len(fooJavac.Inputs) != 1 || fooJavac.Inputs[0].String() != "a.java" { | 
|  | t.Errorf(`foo inputs %v != ["a.java"]`, fooJavac.Inputs) | 
|  | } | 
|  |  | 
|  | if !strings.Contains(fooJavac.Args["classpath"], fooKotlincHeaderClasses.String()) { | 
|  | t.Errorf("foo classpath %v does not contain %q", | 
|  | fooJavac.Args["classpath"], fooKotlincHeaderClasses.String()) | 
|  | } | 
|  |  | 
|  | if !inList(fooKotlincClasses.String(), fooJar.Inputs.Strings()) { | 
|  | t.Errorf("foo jar inputs %v does not contain %q", | 
|  | fooJar.Inputs.Strings(), fooKotlincClasses.String()) | 
|  | } | 
|  |  | 
|  | if !inList(kotlinStdlib.String(), fooJar.Inputs.Strings()) { | 
|  | t.Errorf("foo jar inputs %v does not contain %v", | 
|  | fooJar.Inputs.Strings(), kotlinStdlib.String()) | 
|  | } | 
|  |  | 
|  | if !inList(kotlinStdlibJdk7.String(), fooJar.Inputs.Strings()) { | 
|  | t.Errorf("foo jar inputs %v does not contain %v", | 
|  | fooJar.Inputs.Strings(), kotlinStdlibJdk7.String()) | 
|  | } | 
|  |  | 
|  | if !inList(kotlinStdlibJdk8.String(), fooJar.Inputs.Strings()) { | 
|  | t.Errorf("foo jar inputs %v does not contain %v", | 
|  | fooJar.Inputs.Strings(), kotlinStdlibJdk8.String()) | 
|  | } | 
|  |  | 
|  | if !inList(kotlinAnnotations.String(), fooJar.Inputs.Strings()) { | 
|  | t.Errorf("foo jar inputs %v does not contain %v", | 
|  | fooJar.Inputs.Strings(), kotlinAnnotations.String()) | 
|  | } | 
|  |  | 
|  | if !inList(fooKotlincHeaderClasses.String(), fooHeaderJar.Inputs.Strings()) { | 
|  | t.Errorf("foo header jar inputs %v does not contain %q", | 
|  | fooHeaderJar.Inputs.Strings(), fooKotlincHeaderClasses.String()) | 
|  | } | 
|  |  | 
|  | bazHeaderJar := ctx.ModuleForTests("baz", "android_common").Output("turbine-combined/baz.jar") | 
|  | barKotlinc := ctx.ModuleForTests("bar", "android_common").Rule("kotlinc") | 
|  |  | 
|  | if len(barKotlinc.Inputs) != 1 || barKotlinc.Inputs[0].String() != "b.kt" { | 
|  | t.Errorf(`bar kotlinc inputs %v != ["b.kt"]`, barKotlinc.Inputs) | 
|  | } | 
|  |  | 
|  | if !inList(fooHeaderJar.Output.String(), barKotlinc.Implicits.Strings()) { | 
|  | t.Errorf(`expected %q in bar implicits %v`, | 
|  | fooHeaderJar.Output.String(), barKotlinc.Implicits.Strings()) | 
|  | } | 
|  |  | 
|  | if !inList(bazHeaderJar.Output.String(), barKotlinc.Implicits.Strings()) { | 
|  | t.Errorf(`expected %q in bar implicits %v`, | 
|  | bazHeaderJar.Output.String(), barKotlinc.Implicits.Strings()) | 
|  | } | 
|  | } | 
|  |  | 
|  | func TestKapt(t *testing.T) { | 
|  | bp := ` | 
|  | java_library { | 
|  | name: "foo", | 
|  | srcs: ["a.java", "b.kt"], | 
|  | plugins: ["bar", "baz"], | 
|  | errorprone: { | 
|  | extra_check_modules: ["my_check"], | 
|  | }, | 
|  | } | 
|  |  | 
|  | java_plugin { | 
|  | name: "bar", | 
|  | processor_class: "com.bar", | 
|  | srcs: ["b.java"], | 
|  | } | 
|  |  | 
|  | java_plugin { | 
|  | name: "baz", | 
|  | processor_class: "com.baz", | 
|  | srcs: ["b.java"], | 
|  | } | 
|  |  | 
|  | java_plugin { | 
|  | name: "my_check", | 
|  | srcs: ["b.java"], | 
|  | } | 
|  | ` | 
|  | t.Run("", func(t *testing.T) { | 
|  | ctx, _ := testJava(t, bp) | 
|  |  | 
|  | buildOS := ctx.Config().BuildOS.String() | 
|  |  | 
|  | foo := ctx.ModuleForTests("foo", "android_common") | 
|  | kaptStubs := foo.Rule("kapt") | 
|  | turbineApt := foo.Description("turbine apt") | 
|  | kotlinc := foo.Rule("kotlinc") | 
|  | javac := foo.Rule("javac") | 
|  |  | 
|  | bar := ctx.ModuleForTests("bar", buildOS+"_common").Rule("javac").Output.String() | 
|  | baz := ctx.ModuleForTests("baz", buildOS+"_common").Rule("javac").Output.String() | 
|  |  | 
|  | // Test that the kotlin and java sources are passed to kapt and kotlinc | 
|  | if len(kaptStubs.Inputs) != 2 || kaptStubs.Inputs[0].String() != "a.java" || kaptStubs.Inputs[1].String() != "b.kt" { | 
|  | t.Errorf(`foo kapt inputs %v != ["a.java", "b.kt"]`, kaptStubs.Inputs) | 
|  | } | 
|  | if len(kotlinc.Inputs) != 2 || kotlinc.Inputs[0].String() != "a.java" || kotlinc.Inputs[1].String() != "b.kt" { | 
|  | t.Errorf(`foo kotlinc inputs %v != ["a.java", "b.kt"]`, kotlinc.Inputs) | 
|  | } | 
|  |  | 
|  | // Test that only the java sources are passed to turbine-apt and javac | 
|  | if len(turbineApt.Inputs) != 1 || turbineApt.Inputs[0].String() != "a.java" { | 
|  | t.Errorf(`foo turbine apt inputs %v != ["a.java"]`, turbineApt.Inputs) | 
|  | } | 
|  | if len(javac.Inputs) != 1 || javac.Inputs[0].String() != "a.java" { | 
|  | t.Errorf(`foo inputs %v != ["a.java"]`, javac.Inputs) | 
|  | } | 
|  |  | 
|  | // Test that the kapt stubs jar is a dependency of turbine-apt | 
|  | if !inList(kaptStubs.Output.String(), turbineApt.Implicits.Strings()) { | 
|  | t.Errorf("expected %q in turbine-apt implicits %v", kaptStubs.Output.String(), kotlinc.Implicits.Strings()) | 
|  | } | 
|  |  | 
|  | // Test that the turbine-apt srcjar is a dependency of kotlinc and javac rules | 
|  | if !inList(turbineApt.Output.String(), kotlinc.Implicits.Strings()) { | 
|  | t.Errorf("expected %q in kotlinc implicits %v", turbineApt.Output.String(), kotlinc.Implicits.Strings()) | 
|  | } | 
|  | if !inList(turbineApt.Output.String(), javac.Implicits.Strings()) { | 
|  | t.Errorf("expected %q in javac implicits %v", turbineApt.Output.String(), javac.Implicits.Strings()) | 
|  | } | 
|  |  | 
|  | // Test that the turbine-apt srcjar is extracted by the kotlinc and javac rules | 
|  | if kotlinc.Args["srcJars"] != turbineApt.Output.String() { | 
|  | t.Errorf("expected %q in kotlinc srcjars %v", turbineApt.Output.String(), kotlinc.Args["srcJars"]) | 
|  | } | 
|  | if javac.Args["srcJars"] != turbineApt.Output.String() { | 
|  | t.Errorf("expected %q in javac srcjars %v", turbineApt.Output.String(), kotlinc.Args["srcJars"]) | 
|  | } | 
|  |  | 
|  | // Test that the processors are passed to kapt | 
|  | expectedProcessorPath := "-P plugin:org.jetbrains.kotlin.kapt3:apclasspath=" + bar + | 
|  | " -P plugin:org.jetbrains.kotlin.kapt3:apclasspath=" + baz | 
|  | if kaptStubs.Args["kaptProcessorPath"] != expectedProcessorPath { | 
|  | t.Errorf("expected kaptProcessorPath %q, got %q", expectedProcessorPath, kaptStubs.Args["kaptProcessorPath"]) | 
|  | } | 
|  | expectedProcessor := "-P plugin:org.jetbrains.kotlin.kapt3:processors=com.bar -P plugin:org.jetbrains.kotlin.kapt3:processors=com.baz" | 
|  | if kaptStubs.Args["kaptProcessor"] != expectedProcessor { | 
|  | t.Errorf("expected kaptProcessor %q, got %q", expectedProcessor, kaptStubs.Args["kaptProcessor"]) | 
|  | } | 
|  |  | 
|  | // Test that the processors are passed to turbine-apt | 
|  | expectedProcessorPath = "--processorpath " + bar + " " + baz | 
|  | if !strings.Contains(turbineApt.Args["turbineFlags"], expectedProcessorPath) { | 
|  | t.Errorf("expected turbine-apt processorpath %q, got %q", expectedProcessorPath, turbineApt.Args["turbineFlags"]) | 
|  | } | 
|  | expectedProcessor = "--processors com.bar com.baz" | 
|  | if !strings.Contains(turbineApt.Args["turbineFlags"], expectedProcessor) { | 
|  | t.Errorf("expected turbine-apt processor %q, got %q", expectedProcessor, turbineApt.Args["turbineFlags"]) | 
|  | } | 
|  |  | 
|  | // Test that the processors are not passed to javac | 
|  | if javac.Args["processorpath"] != "" { | 
|  | t.Errorf("expected processorPath '', got %q", javac.Args["processorpath"]) | 
|  | } | 
|  | if javac.Args["processor"] != "-proc:none" { | 
|  | t.Errorf("expected processor '-proc:none', got %q", javac.Args["processor"]) | 
|  | } | 
|  | }) | 
|  |  | 
|  | t.Run("errorprone", func(t *testing.T) { | 
|  | env := map[string]string{ | 
|  | "RUN_ERROR_PRONE": "true", | 
|  | } | 
|  |  | 
|  | result := android.GroupFixturePreparers( | 
|  | PrepareForTestWithJavaDefaultModules, | 
|  | android.FixtureMergeEnv(env), | 
|  | ).RunTestWithBp(t, bp) | 
|  |  | 
|  | buildOS := result.Config.BuildOS.String() | 
|  |  | 
|  | kapt := result.ModuleForTests("foo", "android_common").Rule("kapt") | 
|  | javac := result.ModuleForTests("foo", "android_common").Description("javac") | 
|  | errorprone := result.ModuleForTests("foo", "android_common").Description("errorprone") | 
|  |  | 
|  | bar := result.ModuleForTests("bar", buildOS+"_common").Description("javac").Output.String() | 
|  | baz := result.ModuleForTests("baz", buildOS+"_common").Description("javac").Output.String() | 
|  | myCheck := result.ModuleForTests("my_check", buildOS+"_common").Description("javac").Output.String() | 
|  |  | 
|  | // Test that the errorprone plugins are not passed to kapt | 
|  | expectedProcessorPath := "-P plugin:org.jetbrains.kotlin.kapt3:apclasspath=" + bar + | 
|  | " -P plugin:org.jetbrains.kotlin.kapt3:apclasspath=" + baz | 
|  | if kapt.Args["kaptProcessorPath"] != expectedProcessorPath { | 
|  | t.Errorf("expected kaptProcessorPath %q, got %q", expectedProcessorPath, kapt.Args["kaptProcessorPath"]) | 
|  | } | 
|  | expectedProcessor := "-P plugin:org.jetbrains.kotlin.kapt3:processors=com.bar -P plugin:org.jetbrains.kotlin.kapt3:processors=com.baz" | 
|  | if kapt.Args["kaptProcessor"] != expectedProcessor { | 
|  | t.Errorf("expected kaptProcessor %q, got %q", expectedProcessor, kapt.Args["kaptProcessor"]) | 
|  | } | 
|  |  | 
|  | // Test that the errorprone plugins are not passed to javac | 
|  | if javac.Args["processorpath"] != "" { | 
|  | t.Errorf("expected processorPath '', got %q", javac.Args["processorpath"]) | 
|  | } | 
|  | if javac.Args["processor"] != "-proc:none" { | 
|  | t.Errorf("expected processor '-proc:none', got %q", javac.Args["processor"]) | 
|  | } | 
|  |  | 
|  | // Test that the errorprone plugins are passed to errorprone | 
|  | expectedProcessorPath = "-processorpath " + myCheck | 
|  | if errorprone.Args["processorpath"] != expectedProcessorPath { | 
|  | t.Errorf("expected processorpath %q, got %q", expectedProcessorPath, errorprone.Args["processorpath"]) | 
|  | } | 
|  | if errorprone.Args["processor"] != "-proc:none" { | 
|  | t.Errorf("expected processor '-proc:none', got %q", errorprone.Args["processor"]) | 
|  | } | 
|  | }) | 
|  | } | 
|  |  | 
|  | func TestKaptEncodeFlags(t *testing.T) { | 
|  | // Compares the kaptEncodeFlags against the results of the example implementation at | 
|  | // https://kotlinlang.org/docs/reference/kapt.html#apjavac-options-encoding | 
|  | tests := []struct { | 
|  | in  [][2]string | 
|  | out string | 
|  | }{ | 
|  | { | 
|  | // empty input | 
|  | in:  [][2]string{}, | 
|  | out: "rO0ABXcEAAAAAA==", | 
|  | }, | 
|  | { | 
|  | // common input | 
|  | in: [][2]string{ | 
|  | {"-source", "1.8"}, | 
|  | {"-target", "1.8"}, | 
|  | }, | 
|  | out: "rO0ABXcgAAAAAgAHLXNvdXJjZQADMS44AActdGFyZ2V0AAMxLjg=", | 
|  | }, | 
|  | { | 
|  | // input that serializes to a 255 byte block | 
|  | in: [][2]string{ | 
|  | {"-source", "1.8"}, | 
|  | {"-target", "1.8"}, | 
|  | {"a", strings.Repeat("b", 218)}, | 
|  | }, | 
|  | out: "rO0ABXf/AAAAAwAHLXNvdXJjZQADMS44AActdGFyZ2V0AAMxLjgAAWEA2mJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJi", | 
|  | }, | 
|  | { | 
|  | // input that serializes to a 256 byte block | 
|  | in: [][2]string{ | 
|  | {"-source", "1.8"}, | 
|  | {"-target", "1.8"}, | 
|  | {"a", strings.Repeat("b", 219)}, | 
|  | }, | 
|  | out: "rO0ABXoAAAEAAAAAAwAHLXNvdXJjZQADMS44AActdGFyZ2V0AAMxLjgAAWEA22JiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYg==", | 
|  | }, | 
|  | { | 
|  | // input that serializes to a 257 byte block | 
|  | in: [][2]string{ | 
|  | {"-source", "1.8"}, | 
|  | {"-target", "1.8"}, | 
|  | {"a", strings.Repeat("b", 220)}, | 
|  | }, | 
|  | out: "rO0ABXoAAAEBAAAAAwAHLXNvdXJjZQADMS44AActdGFyZ2V0AAMxLjgAAWEA3GJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmJiYmI=", | 
|  | }, | 
|  | } | 
|  |  | 
|  | for i, test := range tests { | 
|  | t.Run(strconv.Itoa(i), func(t *testing.T) { | 
|  | got := kaptEncodeFlags(test.in) | 
|  | if got != test.out { | 
|  | t.Errorf("\nwant %q\n got %q", test.out, got) | 
|  | } | 
|  | }) | 
|  | } | 
|  | } | 
|  |  | 
|  | func TestKotlinCompose(t *testing.T) { | 
|  | result := android.GroupFixturePreparers( | 
|  | PrepareForTestWithJavaDefaultModules, | 
|  | ).RunTestWithBp(t, ` | 
|  | java_library { | 
|  | name: "androidx.compose.runtime_runtime", | 
|  | } | 
|  |  | 
|  | java_library_host { | 
|  | name: "androidx.compose.compiler_compiler-hosted", | 
|  | } | 
|  |  | 
|  | java_library { | 
|  | name: "withcompose", | 
|  | srcs: ["a.kt"], | 
|  | plugins: ["plugin"], | 
|  | static_libs: ["androidx.compose.runtime_runtime"], | 
|  | } | 
|  |  | 
|  | java_library { | 
|  | name: "nocompose", | 
|  | srcs: ["a.kt"], | 
|  | } | 
|  |  | 
|  | java_plugin { | 
|  | name: "plugin", | 
|  | } | 
|  | `) | 
|  |  | 
|  | buildOS := result.Config.BuildOS.String() | 
|  |  | 
|  | composeCompiler := result.ModuleForTests("androidx.compose.compiler_compiler-hosted", buildOS+"_common").Rule("combineJar").Output | 
|  | withCompose := result.ModuleForTests("withcompose", "android_common") | 
|  | noCompose := result.ModuleForTests("nocompose", "android_common") | 
|  |  | 
|  | android.AssertStringListContains(t, "missing compose compiler dependency", | 
|  | withCompose.Rule("kotlinc").Implicits.Strings(), composeCompiler.String()) | 
|  |  | 
|  | android.AssertStringDoesContain(t, "missing compose compiler plugin", | 
|  | withCompose.VariablesForTestsRelativeToTop()["kotlincFlags"], "-Xplugin="+composeCompiler.String()) | 
|  |  | 
|  | android.AssertStringListContains(t, "missing kapt compose compiler dependency", | 
|  | withCompose.Rule("kapt").Implicits.Strings(), composeCompiler.String()) | 
|  |  | 
|  | android.AssertStringListDoesNotContain(t, "unexpected compose compiler dependency", | 
|  | noCompose.Rule("kotlinc").Implicits.Strings(), composeCompiler.String()) | 
|  |  | 
|  | android.AssertStringDoesNotContain(t, "unexpected compose compiler plugin", | 
|  | noCompose.VariablesForTestsRelativeToTop()["kotlincFlags"], "-Xplugin="+composeCompiler.String()) | 
|  | } |