| // Copyright 2020 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 bp2build |
| |
| import ( |
| "android/soong/android" |
| "io/ioutil" |
| "os" |
| "strings" |
| "testing" |
| ) |
| |
| func setUp() { |
| var err error |
| buildDir, err = ioutil.TempDir("", "bazel_queryview_test") |
| if err != nil { |
| panic(err) |
| } |
| } |
| |
| func tearDown() { |
| os.RemoveAll(buildDir) |
| } |
| |
| func TestMain(m *testing.M) { |
| run := func() int { |
| setUp() |
| defer tearDown() |
| |
| return m.Run() |
| } |
| |
| os.Exit(run()) |
| } |
| |
| func TestGenerateModuleRuleShims(t *testing.T) { |
| moduleTypeFactories := map[string]android.ModuleFactory{ |
| "custom": customModuleFactoryBase, |
| "custom_test": customTestModuleFactoryBase, |
| "custom_defaults": customDefaultsModuleFactoryBasic, |
| } |
| ruleShims := CreateRuleShims(moduleTypeFactories) |
| |
| if len(ruleShims) != 1 { |
| t.Errorf("Expected to generate 1 rule shim, but got %d", len(ruleShims)) |
| } |
| |
| ruleShim := ruleShims["bp2build"] |
| expectedRules := []string{ |
| "custom", |
| "custom_defaults", |
| "custom_test_", |
| } |
| |
| if len(ruleShim.rules) != len(expectedRules) { |
| t.Errorf("Expected %d rules, but got %d", len(expectedRules), len(ruleShim.rules)) |
| } |
| |
| for i, rule := range ruleShim.rules { |
| if rule != expectedRules[i] { |
| t.Errorf("Expected rule shim to contain %s, but got %s", expectedRules[i], rule) |
| } |
| } |
| expectedBzl := `load("//build/bazel/queryview_rules:providers.bzl", "SoongModuleInfo") |
| |
| def _custom_impl(ctx): |
| return [SoongModuleInfo()] |
| |
| custom = rule( |
| implementation = _custom_impl, |
| attrs = { |
| "soong_module_name": attr.string(mandatory = True), |
| "soong_module_variant": attr.string(), |
| "soong_module_deps": attr.label_list(providers = [SoongModuleInfo]), |
| "arch_paths": attr.string_list(), |
| "arch_paths_exclude": attr.string_list(), |
| # bazel_module start |
| # "label": attr.string(), |
| # "bp2build_available": attr.bool(), |
| # bazel_module end |
| "bool_prop": attr.bool(), |
| "bool_ptr_prop": attr.bool(), |
| "embedded_prop": attr.string(), |
| "int64_ptr_prop": attr.int(), |
| # nested_props start |
| # "nested_prop": attr.string(), |
| # nested_props end |
| # nested_props_ptr start |
| # "nested_prop": attr.string(), |
| # nested_props_ptr end |
| "one_to_many_prop": attr.bool(), |
| "other_embedded_prop": attr.string(), |
| "string_list_prop": attr.string_list(), |
| "string_prop": attr.string(), |
| "string_ptr_prop": attr.string(), |
| }, |
| ) |
| |
| def _custom_defaults_impl(ctx): |
| return [SoongModuleInfo()] |
| |
| custom_defaults = rule( |
| implementation = _custom_defaults_impl, |
| attrs = { |
| "soong_module_name": attr.string(mandatory = True), |
| "soong_module_variant": attr.string(), |
| "soong_module_deps": attr.label_list(providers = [SoongModuleInfo]), |
| "arch_paths": attr.string_list(), |
| "arch_paths_exclude": attr.string_list(), |
| "bool_prop": attr.bool(), |
| "bool_ptr_prop": attr.bool(), |
| "embedded_prop": attr.string(), |
| "int64_ptr_prop": attr.int(), |
| # nested_props start |
| # "nested_prop": attr.string(), |
| # nested_props end |
| # nested_props_ptr start |
| # "nested_prop": attr.string(), |
| # nested_props_ptr end |
| "one_to_many_prop": attr.bool(), |
| "other_embedded_prop": attr.string(), |
| "string_list_prop": attr.string_list(), |
| "string_prop": attr.string(), |
| "string_ptr_prop": attr.string(), |
| }, |
| ) |
| |
| def _custom_test__impl(ctx): |
| return [SoongModuleInfo()] |
| |
| custom_test_ = rule( |
| implementation = _custom_test__impl, |
| attrs = { |
| "soong_module_name": attr.string(mandatory = True), |
| "soong_module_variant": attr.string(), |
| "soong_module_deps": attr.label_list(providers = [SoongModuleInfo]), |
| "arch_paths": attr.string_list(), |
| "arch_paths_exclude": attr.string_list(), |
| "bool_prop": attr.bool(), |
| "bool_ptr_prop": attr.bool(), |
| "embedded_prop": attr.string(), |
| "int64_ptr_prop": attr.int(), |
| # nested_props start |
| # "nested_prop": attr.string(), |
| # nested_props end |
| # nested_props_ptr start |
| # "nested_prop": attr.string(), |
| # nested_props_ptr end |
| "one_to_many_prop": attr.bool(), |
| "other_embedded_prop": attr.string(), |
| "string_list_prop": attr.string_list(), |
| "string_prop": attr.string(), |
| "string_ptr_prop": attr.string(), |
| # test_prop start |
| # "test_string_prop": attr.string(), |
| # test_prop end |
| }, |
| ) |
| ` |
| |
| if ruleShim.content != expectedBzl { |
| t.Errorf( |
| "Expected the generated rule shim bzl to be:\n%s\nbut got:\n%s", |
| expectedBzl, |
| ruleShim.content) |
| } |
| } |
| |
| func TestGenerateSoongModuleBzl(t *testing.T) { |
| ruleShims := map[string]RuleShim{ |
| "file1": RuleShim{ |
| rules: []string{"a", "b"}, |
| content: "irrelevant", |
| }, |
| "file2": RuleShim{ |
| rules: []string{"c", "d"}, |
| content: "irrelevant", |
| }, |
| } |
| files := CreateBazelFiles(android.NullConfig("out", "out/soong"), ruleShims, make(map[string]BazelTargets), QueryView) |
| |
| var actualSoongModuleBzl BazelFile |
| for _, f := range files { |
| if f.Basename == "soong_module.bzl" { |
| actualSoongModuleBzl = f |
| } |
| } |
| |
| expectedLoad := `load("//build/bazel/queryview_rules:file1.bzl", "a", "b") |
| load("//build/bazel/queryview_rules:file2.bzl", "c", "d") |
| ` |
| expectedRuleMap := `soong_module_rule_map = { |
| "a": a, |
| "b": b, |
| "c": c, |
| "d": d, |
| }` |
| if !strings.Contains(actualSoongModuleBzl.Contents, expectedLoad) { |
| t.Errorf( |
| "Generated soong_module.bzl:\n\n%s\n\n"+ |
| "Could not find the load statement in the generated soong_module.bzl:\n%s", |
| actualSoongModuleBzl.Contents, |
| expectedLoad) |
| } |
| |
| if !strings.Contains(actualSoongModuleBzl.Contents, expectedRuleMap) { |
| t.Errorf( |
| "Generated soong_module.bzl:\n\n%s\n\n"+ |
| "Could not find the module -> rule map in the generated soong_module.bzl:\n%s", |
| actualSoongModuleBzl.Contents, |
| expectedRuleMap) |
| } |
| } |