|  | // Copyright 2016 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 cc | 
|  |  | 
|  | import ( | 
|  | "path/filepath" | 
|  | "strconv" | 
|  | "strings" | 
|  |  | 
|  | "github.com/google/blueprint/proptools" | 
|  |  | 
|  | "android/soong/android" | 
|  | "android/soong/bazel" | 
|  | "android/soong/bazel/cquery" | 
|  | "android/soong/tradefed" | 
|  | "android/soong/ui/metrics/bp2build_metrics_proto" | 
|  | ) | 
|  |  | 
|  | // TestLinkerProperties properties to be registered via the linker | 
|  | type TestLinkerProperties struct { | 
|  | // if set, build against the gtest library. Defaults to true. | 
|  | Gtest *bool | 
|  |  | 
|  | // if set, use the isolated gtest runner. Defaults to true if gtest is also true and the arch is Windows, false | 
|  | // otherwise. | 
|  | Isolated *bool | 
|  | } | 
|  |  | 
|  | // TestInstallerProperties properties to be registered via the installer | 
|  | type TestInstallerProperties struct { | 
|  | // list of compatibility suites (for example "cts", "vts") that the module should be installed into. | 
|  | Test_suites []string `android:"arch_variant"` | 
|  | } | 
|  |  | 
|  | // Test option struct. | 
|  | type TestOptions struct { | 
|  | android.CommonTestOptions | 
|  |  | 
|  | // The UID that you want to run the test as on a device. | 
|  | Run_test_as *string | 
|  |  | 
|  | // A list of free-formed strings without spaces that categorize the test. | 
|  | Test_suite_tag []string | 
|  |  | 
|  | // a list of extra test configuration files that should be installed with the module. | 
|  | Extra_test_configs []string `android:"path,arch_variant"` | 
|  |  | 
|  | // Add ShippingApiLevelModuleController to auto generated test config. If the device properties | 
|  | // for the shipping api level is less than the min_shipping_api_level, skip this module. | 
|  | Min_shipping_api_level *int64 | 
|  |  | 
|  | // Add ShippingApiLevelModuleController to auto generated test config. If any of the device | 
|  | // shipping api level and vendor api level properties are less than the | 
|  | // vsr_min_shipping_api_level, skip this module. | 
|  | // As this includes the shipping api level check, it is not allowed to define | 
|  | // min_shipping_api_level at the same time with this property. | 
|  | Vsr_min_shipping_api_level *int64 | 
|  |  | 
|  | // Add MinApiLevelModuleController with ro.vndk.version property. If ro.vndk.version has an | 
|  | // integer value and the value is less than the min_vndk_version, skip this module. | 
|  | Min_vndk_version *int64 | 
|  |  | 
|  | // Extra <option> tags to add to the auto generated test xml file under the test runner, e.g., GTest. | 
|  | // The "key" is optional in each of these. | 
|  | Test_runner_options []tradefed.Option | 
|  | } | 
|  |  | 
|  | type TestBinaryProperties struct { | 
|  | // Create a separate binary for each source file.  Useful when there is | 
|  | // global state that can not be torn down and reset between each test suite. | 
|  | Test_per_src *bool | 
|  |  | 
|  | // Disables the creation of a test-specific directory when used with | 
|  | // relative_install_path. Useful if several tests need to be in the same | 
|  | // directory, but test_per_src doesn't work. | 
|  | No_named_install_directory *bool | 
|  |  | 
|  | // list of files or filegroup modules that provide data that should be installed alongside | 
|  | // the test | 
|  | Data []string `android:"path,arch_variant"` | 
|  |  | 
|  | // list of shared library modules that should be installed alongside the test | 
|  | Data_libs []string `android:"arch_variant"` | 
|  |  | 
|  | // list of binary modules that should be installed alongside the test | 
|  | Data_bins []string `android:"arch_variant"` | 
|  |  | 
|  | // the name of the test configuration (for example "AndroidTest.xml") that should be | 
|  | // installed with the module. | 
|  | Test_config *string `android:"path,arch_variant"` | 
|  |  | 
|  | // the name of the test configuration template (for example "AndroidTestTemplate.xml") that | 
|  | // should be installed with the module. | 
|  | Test_config_template *string `android:"path,arch_variant"` | 
|  |  | 
|  | // Test options. | 
|  | Test_options TestOptions | 
|  |  | 
|  | // Add RootTargetPreparer to auto generated test config. This guarantees the test to run | 
|  | // with root permission. | 
|  | Require_root *bool | 
|  |  | 
|  | // Add RunCommandTargetPreparer to stop framework before the test and start it after the test. | 
|  | Disable_framework *bool | 
|  |  | 
|  | // Flag to indicate whether or not to create test config automatically. If AndroidTest.xml | 
|  | // doesn't exist next to the Android.bp, this attribute doesn't need to be set to true | 
|  | // explicitly. | 
|  | Auto_gen_config *bool | 
|  |  | 
|  | // Add parameterized mainline modules to auto generated test config. The options will be | 
|  | // handled by TradeFed to download and install the specified modules on the device. | 
|  | Test_mainline_modules []string | 
|  |  | 
|  | // Install the test into a folder named for the module in all test suites. | 
|  | Per_testcase_directory *bool | 
|  | } | 
|  |  | 
|  | func init() { | 
|  | android.RegisterModuleType("cc_test", TestFactory) | 
|  | android.RegisterModuleType("cc_test_library", TestLibraryFactory) | 
|  | android.RegisterModuleType("cc_benchmark", BenchmarkFactory) | 
|  | android.RegisterModuleType("cc_test_host", TestHostFactory) | 
|  | android.RegisterModuleType("cc_benchmark_host", BenchmarkHostFactory) | 
|  | } | 
|  |  | 
|  | // cc_test generates a test config file and an executable binary file to test | 
|  | // specific functionality on a device. The executable binary gets an implicit | 
|  | // static_libs dependency on libgtests unless the gtest flag is set to false. | 
|  | func TestFactory() android.Module { | 
|  | module := NewTest(android.HostAndDeviceSupported, true) | 
|  | module.bazelHandler = &ccTestBazelHandler{module: module} | 
|  | return module.Init() | 
|  | } | 
|  |  | 
|  | // cc_test_library creates an archive of files (i.e. .o files) which is later | 
|  | // referenced by another module (such as cc_test, cc_defaults or cc_test_library) | 
|  | // for archiving or linking. | 
|  | func TestLibraryFactory() android.Module { | 
|  | module := NewTestLibrary(android.HostAndDeviceSupported) | 
|  | return module.Init() | 
|  | } | 
|  |  | 
|  | // cc_benchmark compiles an executable binary that performs benchmark testing | 
|  | // of a specific component in a device. Additional files such as test suites | 
|  | // and test configuration are installed on the side of the compiled executed | 
|  | // binary. | 
|  | func BenchmarkFactory() android.Module { | 
|  | module := NewBenchmark(android.HostAndDeviceSupported) | 
|  | return module.Init() | 
|  | } | 
|  |  | 
|  | // cc_test_host compiles a test host binary. | 
|  | func TestHostFactory() android.Module { | 
|  | module := NewTest(android.HostSupported, true) | 
|  | return module.Init() | 
|  | } | 
|  |  | 
|  | // cc_benchmark_host compiles an executable binary that performs benchmark | 
|  | // testing of a specific component in the host. Additional files such as | 
|  | // test suites and test configuration are installed on the side of the | 
|  | // compiled executed binary. | 
|  | func BenchmarkHostFactory() android.Module { | 
|  | module := NewBenchmark(android.HostSupported) | 
|  | return module.Init() | 
|  | } | 
|  |  | 
|  | type testPerSrc interface { | 
|  | testPerSrc() bool | 
|  | srcs() []string | 
|  | isAllTestsVariation() bool | 
|  | setSrc(string, string) | 
|  | unsetSrc() | 
|  | } | 
|  |  | 
|  | func (test *testBinary) testPerSrc() bool { | 
|  | return Bool(test.Properties.Test_per_src) | 
|  | } | 
|  |  | 
|  | func (test *testBinary) srcs() []string { | 
|  | return test.baseCompiler.Properties.Srcs | 
|  | } | 
|  |  | 
|  | func (test *testBinary) dataPaths() []android.DataPath { | 
|  | return test.data | 
|  | } | 
|  |  | 
|  | func (test *testBinary) isAllTestsVariation() bool { | 
|  | stem := test.binaryDecorator.Properties.Stem | 
|  | return stem != nil && *stem == "" | 
|  | } | 
|  |  | 
|  | func (test *testBinary) setSrc(name, src string) { | 
|  | test.baseCompiler.Properties.Srcs = []string{src} | 
|  | test.binaryDecorator.Properties.Stem = StringPtr(name) | 
|  | } | 
|  |  | 
|  | func (test *testBinary) unsetSrc() { | 
|  | test.baseCompiler.Properties.Srcs = nil | 
|  | test.binaryDecorator.Properties.Stem = StringPtr("") | 
|  | } | 
|  |  | 
|  | func (test *testBinary) testBinary() bool { | 
|  | return true | 
|  | } | 
|  |  | 
|  | var _ testPerSrc = (*testBinary)(nil) | 
|  |  | 
|  | func TestPerSrcMutator(mctx android.BottomUpMutatorContext) { | 
|  | if m, ok := mctx.Module().(*Module); ok { | 
|  | if test, ok := m.linker.(testPerSrc); ok { | 
|  | numTests := len(test.srcs()) | 
|  | if test.testPerSrc() && numTests > 0 { | 
|  | if duplicate, found := android.CheckDuplicate(test.srcs()); found { | 
|  | mctx.PropertyErrorf("srcs", "found a duplicate entry %q", duplicate) | 
|  | return | 
|  | } | 
|  | testNames := make([]string, numTests) | 
|  | for i, src := range test.srcs() { | 
|  | testNames[i] = strings.TrimSuffix(filepath.Base(src), filepath.Ext(src)) | 
|  | } | 
|  | // In addition to creating one variation per test source file, | 
|  | // create an additional "all tests" variation named "", and have it | 
|  | // depends on all other test_per_src variations. This is useful to | 
|  | // create subsequent dependencies of a given module on all | 
|  | // test_per_src variations created above: by depending on | 
|  | // variation "", that module will transitively depend on all the | 
|  | // other test_per_src variations without the need to know their | 
|  | // name or even their number. | 
|  | testNames = append(testNames, "") | 
|  | tests := mctx.CreateLocalVariations(testNames...) | 
|  | allTests := tests[numTests] | 
|  | allTests.(*Module).linker.(testPerSrc).unsetSrc() | 
|  | // Prevent the "all tests" variation from being installable nor | 
|  | // exporting to Make, as it won't create any output file. | 
|  | allTests.(*Module).Properties.PreventInstall = true | 
|  | allTests.(*Module).Properties.HideFromMake = true | 
|  | for i, src := range test.srcs() { | 
|  | tests[i].(*Module).linker.(testPerSrc).setSrc(testNames[i], src) | 
|  | mctx.AddInterVariantDependency(testPerSrcDepTag, allTests, tests[i]) | 
|  | } | 
|  | mctx.AliasVariation("") | 
|  | } | 
|  | } | 
|  | } | 
|  | } | 
|  |  | 
|  | type testDecorator struct { | 
|  | LinkerProperties    TestLinkerProperties | 
|  | InstallerProperties TestInstallerProperties | 
|  | installer           *baseInstaller | 
|  | linker              *baseLinker | 
|  | } | 
|  |  | 
|  | func (test *testDecorator) gtest() bool { | 
|  | return BoolDefault(test.LinkerProperties.Gtest, true) | 
|  | } | 
|  |  | 
|  | func (test *testDecorator) isolated(ctx android.EarlyModuleContext) bool { | 
|  | return BoolDefault(test.LinkerProperties.Isolated, false) | 
|  | } | 
|  |  | 
|  | // NOTE: Keep this in sync with cc/cc_test.bzl#gtest_copts | 
|  | func (test *testDecorator) linkerFlags(ctx ModuleContext, flags Flags) Flags { | 
|  | if !test.gtest() { | 
|  | return flags | 
|  | } | 
|  |  | 
|  | flags.Local.CFlags = append(flags.Local.CFlags, "-DGTEST_HAS_STD_STRING") | 
|  | if ctx.Host() { | 
|  | flags.Local.CFlags = append(flags.Local.CFlags, "-O0", "-g") | 
|  |  | 
|  | switch ctx.Os() { | 
|  | case android.Windows: | 
|  | flags.Local.CFlags = append(flags.Local.CFlags, "-DGTEST_OS_WINDOWS") | 
|  | case android.Linux: | 
|  | flags.Local.CFlags = append(flags.Local.CFlags, "-DGTEST_OS_LINUX") | 
|  | case android.Darwin: | 
|  | flags.Local.CFlags = append(flags.Local.CFlags, "-DGTEST_OS_MAC") | 
|  | } | 
|  | } else { | 
|  | flags.Local.CFlags = append(flags.Local.CFlags, "-DGTEST_OS_LINUX_ANDROID") | 
|  | } | 
|  |  | 
|  | return flags | 
|  | } | 
|  |  | 
|  | func (test *testDecorator) linkerDeps(ctx BaseModuleContext, deps Deps) Deps { | 
|  | if test.gtest() { | 
|  | if ctx.useSdk() && ctx.Device() { | 
|  | deps.StaticLibs = append(deps.StaticLibs, "libgtest_main_ndk_c++", "libgtest_ndk_c++") | 
|  | } else if test.isolated(ctx) { | 
|  | deps.StaticLibs = append(deps.StaticLibs, "libgtest_isolated_main") | 
|  | // The isolated library requires liblog, but adding it | 
|  | // as a static library means unit tests cannot override | 
|  | // liblog functions. Instead make it a shared library | 
|  | // dependency. | 
|  | deps.SharedLibs = append(deps.SharedLibs, "liblog") | 
|  | } else { | 
|  | deps.StaticLibs = append(deps.StaticLibs, "libgtest_main", "libgtest") | 
|  | } | 
|  | } | 
|  |  | 
|  | return deps | 
|  | } | 
|  |  | 
|  | func (test *testDecorator) linkerProps() []interface{} { | 
|  | return []interface{}{&test.LinkerProperties} | 
|  | } | 
|  |  | 
|  | func (test *testDecorator) installerProps() []interface{} { | 
|  | return []interface{}{&test.InstallerProperties} | 
|  | } | 
|  |  | 
|  | func NewTestInstaller() *baseInstaller { | 
|  | return NewBaseInstaller("nativetest", "nativetest64", InstallInData) | 
|  | } | 
|  |  | 
|  | type testBinary struct { | 
|  | *testDecorator | 
|  | *binaryDecorator | 
|  | *baseCompiler | 
|  | Properties       TestBinaryProperties | 
|  | data             []android.DataPath | 
|  | testConfig       android.Path | 
|  | extraTestConfigs android.Paths | 
|  | } | 
|  |  | 
|  | func (test *testBinary) linkerProps() []interface{} { | 
|  | props := append(test.testDecorator.linkerProps(), test.binaryDecorator.linkerProps()...) | 
|  | props = append(props, &test.Properties) | 
|  | return props | 
|  | } | 
|  |  | 
|  | func (test *testBinary) linkerDeps(ctx DepsContext, deps Deps) Deps { | 
|  | deps = test.testDecorator.linkerDeps(ctx, deps) | 
|  | deps = test.binaryDecorator.linkerDeps(ctx, deps) | 
|  | deps.DataLibs = append(deps.DataLibs, test.Properties.Data_libs...) | 
|  | deps.DataBins = append(deps.DataBins, test.Properties.Data_bins...) | 
|  | return deps | 
|  | } | 
|  |  | 
|  | func (test *testBinary) linkerFlags(ctx ModuleContext, flags Flags) Flags { | 
|  | flags = test.binaryDecorator.linkerFlags(ctx, flags) | 
|  | flags = test.testDecorator.linkerFlags(ctx, flags) | 
|  | return flags | 
|  | } | 
|  |  | 
|  | func (test *testBinary) installerProps() []interface{} { | 
|  | return append(test.baseInstaller.installerProps(), test.testDecorator.installerProps()...) | 
|  | } | 
|  |  | 
|  | func (test *testBinary) install(ctx ModuleContext, file android.Path) { | 
|  | dataSrcPaths := android.PathsForModuleSrc(ctx, test.Properties.Data) | 
|  |  | 
|  | for _, dataSrcPath := range dataSrcPaths { | 
|  | test.data = append(test.data, android.DataPath{SrcPath: dataSrcPath}) | 
|  | } | 
|  |  | 
|  | ctx.VisitDirectDepsWithTag(dataLibDepTag, func(dep android.Module) { | 
|  | depName := ctx.OtherModuleName(dep) | 
|  | linkableDep, ok := dep.(LinkableInterface) | 
|  | if !ok { | 
|  | ctx.ModuleErrorf("data_lib %q is not a LinkableInterface module", depName) | 
|  | } | 
|  | if linkableDep.OutputFile().Valid() { | 
|  | test.data = append(test.data, | 
|  | android.DataPath{SrcPath: linkableDep.OutputFile().Path(), | 
|  | RelativeInstallPath: linkableDep.RelativeInstallPath()}) | 
|  | } | 
|  | }) | 
|  | ctx.VisitDirectDepsWithTag(dataBinDepTag, func(dep android.Module) { | 
|  | depName := ctx.OtherModuleName(dep) | 
|  | linkableDep, ok := dep.(LinkableInterface) | 
|  | if !ok { | 
|  | ctx.ModuleErrorf("data_bin %q is not a LinkableInterface module", depName) | 
|  | } | 
|  | if linkableDep.OutputFile().Valid() { | 
|  | test.data = append(test.data, | 
|  | android.DataPath{SrcPath: linkableDep.OutputFile().Path(), | 
|  | RelativeInstallPath: linkableDep.RelativeInstallPath()}) | 
|  | } | 
|  | }) | 
|  |  | 
|  | useVendor := ctx.inVendor() || ctx.useVndk() | 
|  | testInstallBase := getTestInstallBase(useVendor) | 
|  | configs := getTradefedConfigOptions(ctx, &test.Properties, test.isolated(ctx), ctx.Device()) | 
|  |  | 
|  | test.testConfig = tradefed.AutoGenTestConfig(ctx, tradefed.AutoGenTestConfigOptions{ | 
|  | TestConfigProp:         test.Properties.Test_config, | 
|  | TestConfigTemplateProp: test.Properties.Test_config_template, | 
|  | TestSuites:             test.testDecorator.InstallerProperties.Test_suites, | 
|  | Config:                 configs, | 
|  | TestRunnerOptions:      test.Properties.Test_options.Test_runner_options, | 
|  | AutoGenConfig:          test.Properties.Auto_gen_config, | 
|  | TestInstallBase:        testInstallBase, | 
|  | DeviceTemplate:         "${NativeTestConfigTemplate}", | 
|  | HostTemplate:           "${NativeHostTestConfigTemplate}", | 
|  | }) | 
|  |  | 
|  | test.extraTestConfigs = android.PathsForModuleSrc(ctx, test.Properties.Test_options.Extra_test_configs) | 
|  |  | 
|  | test.binaryDecorator.baseInstaller.dir = "nativetest" | 
|  | test.binaryDecorator.baseInstaller.dir64 = "nativetest64" | 
|  |  | 
|  | if !Bool(test.Properties.No_named_install_directory) { | 
|  | test.binaryDecorator.baseInstaller.relative = ctx.ModuleName() | 
|  | } else if String(test.binaryDecorator.baseInstaller.Properties.Relative_install_path) == "" { | 
|  | ctx.PropertyErrorf("no_named_install_directory", "Module install directory may only be disabled if relative_install_path is set") | 
|  | } | 
|  |  | 
|  | if ctx.Host() && test.gtest() && test.Properties.Test_options.Unit_test == nil { | 
|  | test.Properties.Test_options.Unit_test = proptools.BoolPtr(true) | 
|  | } | 
|  | test.binaryDecorator.baseInstaller.install(ctx, file) | 
|  | } | 
|  |  | 
|  | func getTestInstallBase(useVendor bool) string { | 
|  | // TODO: (b/167308193) Switch to /data/local/tests/unrestricted as the default install base. | 
|  | testInstallBase := "/data/local/tmp" | 
|  | if useVendor { | 
|  | testInstallBase = "/data/local/tests/vendor" | 
|  | } | 
|  | return testInstallBase | 
|  | } | 
|  |  | 
|  | func getTradefedConfigOptions(ctx android.EarlyModuleContext, properties *TestBinaryProperties, isolated bool, device bool) []tradefed.Config { | 
|  | var configs []tradefed.Config | 
|  |  | 
|  | for _, module := range properties.Test_mainline_modules { | 
|  | configs = append(configs, tradefed.Option{Name: "config-descriptor:metadata", Key: "mainline-param", Value: module}) | 
|  | } | 
|  | if device { | 
|  | if Bool(properties.Require_root) { | 
|  | configs = append(configs, tradefed.Object{"target_preparer", "com.android.tradefed.targetprep.RootTargetPreparer", nil}) | 
|  | } else { | 
|  | var options []tradefed.Option | 
|  | options = append(options, tradefed.Option{Name: "force-root", Value: "false"}) | 
|  | configs = append(configs, tradefed.Object{"target_preparer", "com.android.tradefed.targetprep.RootTargetPreparer", options}) | 
|  | } | 
|  | if Bool(properties.Disable_framework) { | 
|  | var options []tradefed.Option | 
|  | configs = append(configs, tradefed.Object{"target_preparer", "com.android.tradefed.targetprep.StopServicesSetup", options}) | 
|  | } | 
|  | } | 
|  | if isolated { | 
|  | configs = append(configs, tradefed.Option{Name: "not-shardable", Value: "true"}) | 
|  | } | 
|  | if properties.Test_options.Run_test_as != nil { | 
|  | configs = append(configs, tradefed.Option{Name: "run-test-as", Value: String(properties.Test_options.Run_test_as)}) | 
|  | } | 
|  | for _, tag := range properties.Test_options.Test_suite_tag { | 
|  | configs = append(configs, tradefed.Option{Name: "test-suite-tag", Value: tag}) | 
|  | } | 
|  | if properties.Test_options.Min_shipping_api_level != nil { | 
|  | if properties.Test_options.Vsr_min_shipping_api_level != nil { | 
|  | ctx.PropertyErrorf("test_options.min_shipping_api_level", "must not be set at the same time as 'vsr_min_shipping_api_level'.") | 
|  | } | 
|  | var options []tradefed.Option | 
|  | options = append(options, tradefed.Option{Name: "min-api-level", Value: strconv.FormatInt(int64(*properties.Test_options.Min_shipping_api_level), 10)}) | 
|  | configs = append(configs, tradefed.Object{"module_controller", "com.android.tradefed.testtype.suite.module.ShippingApiLevelModuleController", options}) | 
|  | } | 
|  | if properties.Test_options.Vsr_min_shipping_api_level != nil { | 
|  | var options []tradefed.Option | 
|  | options = append(options, tradefed.Option{Name: "vsr-min-api-level", Value: strconv.FormatInt(int64(*properties.Test_options.Vsr_min_shipping_api_level), 10)}) | 
|  | configs = append(configs, tradefed.Object{"module_controller", "com.android.tradefed.testtype.suite.module.ShippingApiLevelModuleController", options}) | 
|  | } | 
|  | if properties.Test_options.Min_vndk_version != nil { | 
|  | var options []tradefed.Option | 
|  | options = append(options, tradefed.Option{Name: "min-api-level", Value: strconv.FormatInt(int64(*properties.Test_options.Min_vndk_version), 10)}) | 
|  | options = append(options, tradefed.Option{Name: "api-level-prop", Value: "ro.vndk.version"}) | 
|  | configs = append(configs, tradefed.Object{"module_controller", "com.android.tradefed.testtype.suite.module.MinApiLevelModuleController", options}) | 
|  | } | 
|  | return configs | 
|  | } | 
|  |  | 
|  | func NewTest(hod android.HostOrDeviceSupported, bazelable bool) *Module { | 
|  | module, binary := newBinary(hod, bazelable) | 
|  | module.bazelable = bazelable | 
|  | module.multilib = android.MultilibBoth | 
|  | binary.baseInstaller = NewTestInstaller() | 
|  |  | 
|  | test := &testBinary{ | 
|  | testDecorator: &testDecorator{ | 
|  | linker:    binary.baseLinker, | 
|  | installer: binary.baseInstaller, | 
|  | }, | 
|  | binaryDecorator: binary, | 
|  | baseCompiler:    NewBaseCompiler(), | 
|  | } | 
|  | module.compiler = test | 
|  | module.linker = test | 
|  | module.installer = test | 
|  | return module | 
|  | } | 
|  |  | 
|  | type testLibrary struct { | 
|  | *testDecorator | 
|  | *libraryDecorator | 
|  | } | 
|  |  | 
|  | func (test *testLibrary) testLibrary() bool { | 
|  | return true | 
|  | } | 
|  |  | 
|  | func (test *testLibrary) linkerProps() []interface{} { | 
|  | var props []interface{} | 
|  | props = append(props, test.testDecorator.linkerProps()...) | 
|  | return append(props, test.libraryDecorator.linkerProps()...) | 
|  | } | 
|  |  | 
|  | func (test *testLibrary) linkerDeps(ctx DepsContext, deps Deps) Deps { | 
|  | deps = test.testDecorator.linkerDeps(ctx, deps) | 
|  | deps = test.libraryDecorator.linkerDeps(ctx, deps) | 
|  | return deps | 
|  | } | 
|  |  | 
|  | func (test *testLibrary) linkerFlags(ctx ModuleContext, flags Flags) Flags { | 
|  | flags = test.libraryDecorator.linkerFlags(ctx, flags) | 
|  | flags = test.testDecorator.linkerFlags(ctx, flags) | 
|  | return flags | 
|  | } | 
|  |  | 
|  | func (test *testLibrary) installerProps() []interface{} { | 
|  | return append(test.baseInstaller.installerProps(), test.testDecorator.installerProps()...) | 
|  | } | 
|  |  | 
|  | func NewTestLibrary(hod android.HostOrDeviceSupported) *Module { | 
|  | module, library := NewLibrary(android.HostAndDeviceSupported) | 
|  | library.baseInstaller = NewTestInstaller() | 
|  | test := &testLibrary{ | 
|  | testDecorator: &testDecorator{ | 
|  | linker:    library.baseLinker, | 
|  | installer: library.baseInstaller, | 
|  | }, | 
|  | libraryDecorator: library, | 
|  | } | 
|  | module.linker = test | 
|  | module.installer = test | 
|  | module.bazelable = true | 
|  | return module | 
|  | } | 
|  |  | 
|  | type BenchmarkProperties struct { | 
|  | // list of files or filegroup modules that provide data that should be installed alongside | 
|  | // the test | 
|  | Data []string `android:"path"` | 
|  |  | 
|  | // list of compatibility suites (for example "cts", "vts") that the module should be | 
|  | // installed into. | 
|  | Test_suites []string `android:"arch_variant"` | 
|  |  | 
|  | // the name of the test configuration (for example "AndroidTest.xml") that should be | 
|  | // installed with the module. | 
|  | Test_config *string `android:"path,arch_variant"` | 
|  |  | 
|  | // the name of the test configuration template (for example "AndroidTestTemplate.xml") that | 
|  | // should be installed with the module. | 
|  | Test_config_template *string `android:"path,arch_variant"` | 
|  |  | 
|  | // Add RootTargetPreparer to auto generated test config. This guarantees the test to run | 
|  | // with root permission. | 
|  | Require_root *bool | 
|  |  | 
|  | // Flag to indicate whether or not to create test config automatically. If AndroidTest.xml | 
|  | // doesn't exist next to the Android.bp, this attribute doesn't need to be set to true | 
|  | // explicitly. | 
|  | Auto_gen_config *bool | 
|  | } | 
|  |  | 
|  | type benchmarkDecorator struct { | 
|  | *binaryDecorator | 
|  | Properties BenchmarkProperties | 
|  | data       android.Paths | 
|  | testConfig android.Path | 
|  | } | 
|  |  | 
|  | func (benchmark *benchmarkDecorator) benchmarkBinary() bool { | 
|  | return true | 
|  | } | 
|  |  | 
|  | func (benchmark *benchmarkDecorator) linkerProps() []interface{} { | 
|  | props := benchmark.binaryDecorator.linkerProps() | 
|  | props = append(props, &benchmark.Properties) | 
|  | return props | 
|  | } | 
|  |  | 
|  | func (benchmark *benchmarkDecorator) linkerDeps(ctx DepsContext, deps Deps) Deps { | 
|  | deps = benchmark.binaryDecorator.linkerDeps(ctx, deps) | 
|  | deps.StaticLibs = append(deps.StaticLibs, "libgoogle-benchmark") | 
|  | return deps | 
|  | } | 
|  |  | 
|  | func (benchmark *benchmarkDecorator) install(ctx ModuleContext, file android.Path) { | 
|  | benchmark.data = android.PathsForModuleSrc(ctx, benchmark.Properties.Data) | 
|  |  | 
|  | var configs []tradefed.Config | 
|  | if Bool(benchmark.Properties.Require_root) { | 
|  | configs = append(configs, tradefed.Object{"target_preparer", "com.android.tradefed.targetprep.RootTargetPreparer", nil}) | 
|  | } | 
|  | benchmark.testConfig = tradefed.AutoGenTestConfig(ctx, tradefed.AutoGenTestConfigOptions{ | 
|  | TestConfigProp:         benchmark.Properties.Test_config, | 
|  | TestConfigTemplateProp: benchmark.Properties.Test_config_template, | 
|  | TestSuites:             benchmark.Properties.Test_suites, | 
|  | Config:                 configs, | 
|  | AutoGenConfig:          benchmark.Properties.Auto_gen_config, | 
|  | DeviceTemplate:         "${NativeBenchmarkTestConfigTemplate}", | 
|  | HostTemplate:           "${NativeBenchmarkTestConfigTemplate}", | 
|  | }) | 
|  |  | 
|  | benchmark.binaryDecorator.baseInstaller.dir = filepath.Join("benchmarktest", ctx.ModuleName()) | 
|  | benchmark.binaryDecorator.baseInstaller.dir64 = filepath.Join("benchmarktest64", ctx.ModuleName()) | 
|  | benchmark.binaryDecorator.baseInstaller.install(ctx, file) | 
|  | } | 
|  |  | 
|  | func NewBenchmark(hod android.HostOrDeviceSupported) *Module { | 
|  | module, binary := newBinary(hod, false) | 
|  | module.multilib = android.MultilibBoth | 
|  | binary.baseInstaller = NewBaseInstaller("benchmarktest", "benchmarktest64", InstallInData) | 
|  |  | 
|  | benchmark := &benchmarkDecorator{ | 
|  | binaryDecorator: binary, | 
|  | } | 
|  | module.linker = benchmark | 
|  | module.installer = benchmark | 
|  | return module | 
|  | } | 
|  |  | 
|  | type ccTestBazelHandler struct { | 
|  | module *Module | 
|  | } | 
|  |  | 
|  | var _ BazelHandler = (*ccTestBazelHandler)(nil) | 
|  |  | 
|  | // The top level target named $label is a test_suite target, | 
|  | // not the internal cc_test executable target. | 
|  | // | 
|  | // This is to ensure `b test //$label` runs the test_suite target directly, | 
|  | // which depends on tradefed_test targets, instead of the internal cc_test | 
|  | // target, which doesn't have tradefed integrations. | 
|  | // | 
|  | // However, for cquery, we want the internal cc_test executable target, which | 
|  | // has the suffix "__tf_internal". | 
|  | func mixedBuildsTestLabel(label string) string { | 
|  | return label + "__tf_internal" | 
|  | } | 
|  |  | 
|  | func (handler *ccTestBazelHandler) QueueBazelCall(ctx android.BaseModuleContext, label string) { | 
|  | bazelCtx := ctx.Config().BazelContext | 
|  | bazelCtx.QueueBazelRequest(mixedBuildsTestLabel(label), cquery.GetCcUnstrippedInfo, android.GetConfigKey(ctx)) | 
|  | } | 
|  |  | 
|  | func (handler *ccTestBazelHandler) ProcessBazelQueryResponse(ctx android.ModuleContext, label string) { | 
|  | bazelCtx := ctx.Config().BazelContext | 
|  | info, err := bazelCtx.GetCcUnstrippedInfo(mixedBuildsTestLabel(label), android.GetConfigKey(ctx)) | 
|  | if err != nil { | 
|  | ctx.ModuleErrorf(err.Error()) | 
|  | return | 
|  | } | 
|  |  | 
|  | var outputFilePath android.Path = android.PathForBazelOut(ctx, info.OutputFile) | 
|  | if len(info.TidyFiles) > 0 { | 
|  | handler.module.tidyFiles = android.PathsForBazelOut(ctx, info.TidyFiles) | 
|  | outputFilePath = android.AttachValidationActions(ctx, outputFilePath, handler.module.tidyFiles) | 
|  | } | 
|  | handler.module.outputFile = android.OptionalPathForPath(outputFilePath) | 
|  | handler.module.linker.(*testBinary).unstrippedOutputFile = android.PathForBazelOut(ctx, info.UnstrippedOutput) | 
|  |  | 
|  | handler.module.setAndroidMkVariablesFromCquery(info.CcAndroidMkInfo) | 
|  | } | 
|  |  | 
|  | // binaryAttributes contains Bazel attributes corresponding to a cc test | 
|  | type testBinaryAttributes struct { | 
|  | binaryAttributes | 
|  |  | 
|  | Gtest *bool | 
|  |  | 
|  | tidyAttributes | 
|  | tradefed.TestConfigAttributes | 
|  |  | 
|  | Runs_on bazel.StringListAttribute | 
|  | } | 
|  |  | 
|  | // testBinaryBp2build is the bp2build converter for cc_test modules. A cc_test's | 
|  | // dependency graph and compilation/linking steps are functionally similar to a | 
|  | // cc_binary, but has additional dependencies on test deps like gtest, and | 
|  | // produces additional runfiles like XML plans for Tradefed orchestration | 
|  | // | 
|  | // TODO(b/244432609): handle `isolated` property. | 
|  | // TODO(b/244432134): handle custom runpaths for tests that assume runfile layouts not | 
|  | // default to bazel. (see linkerInit function) | 
|  | func testBinaryBp2build(ctx android.Bp2buildMutatorContext, m *Module) { | 
|  | var testBinaryAttrs testBinaryAttributes | 
|  | testBinaryAttrs.binaryAttributes = binaryBp2buildAttrs(ctx, m) | 
|  |  | 
|  | var data bazel.LabelListAttribute | 
|  | var tags bazel.StringListAttribute | 
|  |  | 
|  | testBinaryProps := m.GetArchVariantProperties(ctx, &TestBinaryProperties{}) | 
|  | for axis, configToProps := range testBinaryProps { | 
|  | for config, props := range configToProps { | 
|  | if p, ok := props.(*TestBinaryProperties); ok { | 
|  | // Combine data, data_bins and data_libs into a single 'data' attribute. | 
|  | var combinedData bazel.LabelList | 
|  | combinedData.Append(android.BazelLabelForModuleSrc(ctx, p.Data)) | 
|  | combinedData.Append(android.BazelLabelForModuleDeps(ctx, p.Data_bins)) | 
|  | combinedData.Append(android.BazelLabelForModuleDeps(ctx, p.Data_libs)) | 
|  | data.SetSelectValue(axis, config, combinedData) | 
|  | tags.SetSelectValue(axis, config, p.Test_options.Tags) | 
|  |  | 
|  | // TODO: b/300117121 - handle bp2build conversion of non-unit tests | 
|  | // default to true to only handle non-nil falses | 
|  | if !BoolDefault(p.Test_options.Unit_test, true) { | 
|  | ctx.MarkBp2buildUnconvertible(bp2build_metrics_proto.UnconvertedReasonType_PROPERTY_UNSUPPORTED, "Host unit_test = false") | 
|  | return | 
|  | } | 
|  | } | 
|  | } | 
|  | } | 
|  |  | 
|  | // The logic comes from https://cs.android.com/android/platform/superproject/main/+/0df8153267f96da877febc5332240fa06ceb8533:build/soong/cc/sanitize.go;l=488 | 
|  | var features bazel.StringListAttribute | 
|  | curFeatures := testBinaryAttrs.binaryAttributes.Features.SelectValue(bazel.OsArchConfigurationAxis, bazel.OsArchAndroidArm64) | 
|  | var newFeatures []string | 
|  | if !android.InList("memtag_heap", curFeatures) && !android.InList("-memtag_heap", curFeatures) { | 
|  | newFeatures = append(newFeatures, "memtag_heap") | 
|  | if !android.InList("diag_memtag_heap", curFeatures) && !android.InList("-diag_memtag_heap", curFeatures) { | 
|  | newFeatures = append(newFeatures, "diag_memtag_heap") | 
|  | } | 
|  | } | 
|  |  | 
|  | features.SetSelectValue(bazel.OsArchConfigurationAxis, bazel.OsArchAndroidArm64, newFeatures) | 
|  | testBinaryAttrs.binaryAttributes.Features.Append(features) | 
|  | testBinaryAttrs.binaryAttributes.Features.DeduplicateAxesFromBase() | 
|  |  | 
|  | m.convertTidyAttributes(ctx, &testBinaryAttrs.tidyAttributes) | 
|  |  | 
|  | testBinary := m.linker.(*testBinary) | 
|  | gtest := testBinary.gtest() | 
|  | gtestIsolated := testBinary.isolated(ctx) | 
|  | // Use the underling bool pointer for Gtest in attrs | 
|  | // This ensures that if this property is not set in Android.bp file, it will not be set in BUILD file either | 
|  | // cc_test macro will default gtest to True | 
|  | testBinaryAttrs.Gtest = testBinary.LinkerProperties.Gtest | 
|  |  | 
|  | addImplicitGtestDeps(ctx, &testBinaryAttrs, gtest, gtestIsolated) | 
|  |  | 
|  | var unitTest *bool | 
|  |  | 
|  | for _, testProps := range m.GetProperties() { | 
|  | if p, ok := testProps.(*TestBinaryProperties); ok { | 
|  | useVendor := false // TODO Bug: 262914724 | 
|  | testInstallBase := getTestInstallBase(useVendor) | 
|  | testConfigAttributes := tradefed.GetTestConfigAttributes( | 
|  | ctx, | 
|  | p.Test_config, | 
|  | p.Test_options.Extra_test_configs, | 
|  | p.Auto_gen_config, | 
|  | p.Test_options.Test_suite_tag, | 
|  | p.Test_config_template, | 
|  | getTradefedConfigOptions(ctx, p, gtestIsolated, true), | 
|  | &testInstallBase, | 
|  | ) | 
|  | testBinaryAttrs.TestConfigAttributes = testConfigAttributes | 
|  | unitTest = p.Test_options.Unit_test | 
|  | } | 
|  | } | 
|  |  | 
|  | testBinaryAttrs.Runs_on = bazel.MakeStringListAttribute(android.RunsOn( | 
|  | m.ModuleBase.HostSupported(), | 
|  | m.ModuleBase.DeviceSupported(), | 
|  | gtest || (unitTest != nil && *unitTest))) | 
|  |  | 
|  | // TODO (b/262914724): convert to tradefed_cc_test and tradefed_cc_test_host | 
|  | ctx.CreateBazelTargetModule( | 
|  | bazel.BazelTargetModuleProperties{ | 
|  | Rule_class:        "cc_test", | 
|  | Bzl_load_location: "//build/bazel/rules/cc:cc_test.bzl", | 
|  | }, | 
|  | android.CommonAttributes{ | 
|  | Name: m.Name(), | 
|  | Data: data, | 
|  | Tags: tags, | 
|  | }, | 
|  | &testBinaryAttrs) | 
|  | } | 
|  |  | 
|  | // cc_test that builds using gtest needs some additional deps | 
|  | // addImplicitGtestDeps makes these deps explicit in the generated BUILD files | 
|  | func addImplicitGtestDeps(ctx android.Bp2buildMutatorContext, attrs *testBinaryAttributes, gtest, gtestIsolated bool) { | 
|  | addDepsAndDedupe := func(lla *bazel.LabelListAttribute, modules []string) { | 
|  | moduleLabels := android.BazelLabelForModuleDeps(ctx, modules) | 
|  | lla.Value.Append(moduleLabels) | 
|  | // Dedupe | 
|  | lla.Value = bazel.FirstUniqueBazelLabelList(lla.Value) | 
|  | } | 
|  | // this must be kept in sync with Soong's implementation in: | 
|  | // https://cs.android.com/android/_/android/platform/build/soong/+/460fb2d6d546b5ab493a7e5479998c4933a80f73:cc/test.go;l=300-313;drc=ec7314336a2b35ea30ce5438b83949c28e3ac429;bpv=1;bpt=0 | 
|  | if gtest { | 
|  | // TODO - b/244433197: Handle canUseSdk | 
|  | if gtestIsolated { | 
|  | addDepsAndDedupe(&attrs.Deps, []string{"libgtest_isolated_main"}) | 
|  | addDepsAndDedupe(&attrs.Dynamic_deps, []string{"liblog"}) | 
|  | } else { | 
|  | addDepsAndDedupe(&attrs.Deps, []string{ | 
|  | "libgtest_main", | 
|  | "libgtest", | 
|  | }) | 
|  | } | 
|  | } | 
|  | } |