|  | // 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 dexpreopt | 
|  |  | 
|  | // This file contains unit tests for class loader context structure. | 
|  | // For class loader context tests involving .bp files, see TestUsesLibraries in java package. | 
|  |  | 
|  | import ( | 
|  | "fmt" | 
|  | "reflect" | 
|  | "sort" | 
|  | "strings" | 
|  | "testing" | 
|  |  | 
|  | "android/soong/android" | 
|  | ) | 
|  |  | 
|  | func TestCLC(t *testing.T) { | 
|  | // Construct class loader context with the following structure: | 
|  | // . | 
|  | // ├── 29 | 
|  | // │   ├── android.hidl.manager | 
|  | // │   └── android.hidl.base | 
|  | // │ | 
|  | // └── any | 
|  | //     ├── a'  (a single quotation mark (') is there to test escaping) | 
|  | //     ├── b | 
|  | //     ├── c | 
|  | //     ├── d | 
|  | //     │   ├── a2 | 
|  | //     │   ├── b2 | 
|  | //     │   └── c2 | 
|  | //     │       ├── a1 | 
|  | //     │       └── b1 | 
|  | //     ├── f | 
|  | //     ├── a3 | 
|  | //     └── b3 | 
|  | // | 
|  | ctx := testContext() | 
|  |  | 
|  | optional := false | 
|  |  | 
|  | m := make(ClassLoaderContextMap) | 
|  |  | 
|  | m.AddContext(ctx, AnySdkVersion, "a'", optional, buildPath(ctx, "a"), installPath(ctx, "a"), nil) | 
|  | m.AddContext(ctx, AnySdkVersion, "b", optional, buildPath(ctx, "b"), installPath(ctx, "b"), nil) | 
|  | m.AddContext(ctx, AnySdkVersion, "c", optional, buildPath(ctx, "c"), installPath(ctx, "c"), nil) | 
|  |  | 
|  | // Add some libraries with nested subcontexts. | 
|  |  | 
|  | m1 := make(ClassLoaderContextMap) | 
|  | m1.AddContext(ctx, AnySdkVersion, "a1", optional, buildPath(ctx, "a1"), installPath(ctx, "a1"), nil) | 
|  | m1.AddContext(ctx, AnySdkVersion, "b1", optional, buildPath(ctx, "b1"), installPath(ctx, "b1"), nil) | 
|  |  | 
|  | m2 := make(ClassLoaderContextMap) | 
|  | m2.AddContext(ctx, AnySdkVersion, "a2", optional, buildPath(ctx, "a2"), installPath(ctx, "a2"), nil) | 
|  | m2.AddContext(ctx, AnySdkVersion, "b2", optional, buildPath(ctx, "b2"), installPath(ctx, "b2"), nil) | 
|  | m2.AddContext(ctx, AnySdkVersion, "c2", optional, buildPath(ctx, "c2"), installPath(ctx, "c2"), m1) | 
|  |  | 
|  | m3 := make(ClassLoaderContextMap) | 
|  | m3.AddContext(ctx, AnySdkVersion, "a3", optional, buildPath(ctx, "a3"), installPath(ctx, "a3"), nil) | 
|  | m3.AddContext(ctx, AnySdkVersion, "b3", optional, buildPath(ctx, "b3"), installPath(ctx, "b3"), nil) | 
|  |  | 
|  | m.AddContext(ctx, AnySdkVersion, "d", optional, buildPath(ctx, "d"), installPath(ctx, "d"), m2) | 
|  | // When the same library is both in conditional and unconditional context, it should be removed | 
|  | // from conditional context. | 
|  | m.AddContext(ctx, 42, "f", optional, buildPath(ctx, "f"), installPath(ctx, "f"), nil) | 
|  | m.AddContext(ctx, AnySdkVersion, "f", optional, buildPath(ctx, "f"), installPath(ctx, "f"), nil) | 
|  |  | 
|  | // Merge map with implicit root library that is among toplevel contexts => does nothing. | 
|  | m.AddContextMap(m1, "c") | 
|  | // Merge map with implicit root library that is not among toplevel contexts => all subcontexts | 
|  | // of the other map are added as toplevel contexts. | 
|  | m.AddContextMap(m3, "m_g") | 
|  |  | 
|  | // Compatibility libraries with unknown install paths get default paths. | 
|  | m.AddContext(ctx, 29, AndroidHidlManager, optional, buildPath(ctx, AndroidHidlManager), nil, nil) | 
|  | m.AddContext(ctx, 29, AndroidHidlBase, optional, buildPath(ctx, AndroidHidlBase), nil, nil) | 
|  |  | 
|  | // Add "android.test.mock" to conditional CLC, observe that is gets removed because it is only | 
|  | // needed as a compatibility library if "android.test.runner" is in CLC as well. | 
|  | m.AddContext(ctx, 30, AndroidTestMock, optional, buildPath(ctx, AndroidTestMock), nil, nil) | 
|  |  | 
|  | valid, validationError := validateClassLoaderContext(m) | 
|  |  | 
|  | fixClassLoaderContext(m) | 
|  |  | 
|  | var actualNames []string | 
|  | var actualPaths android.Paths | 
|  | var haveUsesLibsReq, haveUsesLibsOpt []string | 
|  | if valid && validationError == nil { | 
|  | actualNames, actualPaths = ComputeClassLoaderContextDependencies(m) | 
|  | haveUsesLibsReq, haveUsesLibsOpt = m.UsesLibs() | 
|  | } | 
|  |  | 
|  | // Test that validation is successful (all paths are known). | 
|  | t.Run("validate", func(t *testing.T) { | 
|  | if !(valid && validationError == nil) { | 
|  | t.Errorf("invalid class loader context") | 
|  | } | 
|  | }) | 
|  |  | 
|  | // Test that all expected build paths are gathered. | 
|  | t.Run("names and paths", func(t *testing.T) { | 
|  | expectedNames := []string{ | 
|  | "a'", "a1", "a2", "a3", "android.hidl.base-V1.0-java", "android.hidl.manager-V1.0-java", "b", | 
|  | "b1", "b2", "b3", "c", "c2", "d", "f", | 
|  | } | 
|  | expectedPaths := []string{ | 
|  | "out/soong/android.hidl.manager-V1.0-java.jar", "out/soong/android.hidl.base-V1.0-java.jar", | 
|  | "out/soong/a.jar", "out/soong/b.jar", "out/soong/c.jar", "out/soong/d.jar", | 
|  | "out/soong/a2.jar", "out/soong/b2.jar", "out/soong/c2.jar", | 
|  | "out/soong/a1.jar", "out/soong/b1.jar", | 
|  | "out/soong/f.jar", "out/soong/a3.jar", "out/soong/b3.jar", | 
|  | } | 
|  | actualPathsStrs := actualPaths.Strings() | 
|  | // The order does not matter. | 
|  | sort.Strings(expectedNames) | 
|  | sort.Strings(actualNames) | 
|  | android.AssertArrayString(t, "", expectedNames, actualNames) | 
|  | sort.Strings(expectedPaths) | 
|  | sort.Strings(actualPathsStrs) | 
|  | android.AssertArrayString(t, "", expectedPaths, actualPathsStrs) | 
|  | }) | 
|  |  | 
|  | // Test the JSON passed to construct_context.py. | 
|  | t.Run("json", func(t *testing.T) { | 
|  | // The tree structure within each SDK version should be kept exactly the same when serialized | 
|  | // to JSON. The order matters because the Python script keeps the order within each SDK version | 
|  | // as is. | 
|  | // The JSON is passed to the Python script as a commandline flag, so quotation ('') and escaping | 
|  | // must be performed. | 
|  | android.AssertStringEquals(t, "", strings.TrimSpace(` | 
|  | '{"29":[{"Name":"android.hidl.manager-V1.0-java","Optional":false,"Host":"out/soong/android.hidl.manager-V1.0-java.jar","Device":"/system/framework/android.hidl.manager-V1.0-java.jar","Subcontexts":[]},{"Name":"android.hidl.base-V1.0-java","Optional":false,"Host":"out/soong/android.hidl.base-V1.0-java.jar","Device":"/system/framework/android.hidl.base-V1.0-java.jar","Subcontexts":[]}],"30":[],"42":[],"any":[{"Name":"a'\''","Optional":false,"Host":"out/soong/a.jar","Device":"/system/a.jar","Subcontexts":[]},{"Name":"b","Optional":false,"Host":"out/soong/b.jar","Device":"/system/b.jar","Subcontexts":[]},{"Name":"c","Optional":false,"Host":"out/soong/c.jar","Device":"/system/c.jar","Subcontexts":[]},{"Name":"d","Optional":false,"Host":"out/soong/d.jar","Device":"/system/d.jar","Subcontexts":[{"Name":"a2","Optional":false,"Host":"out/soong/a2.jar","Device":"/system/a2.jar","Subcontexts":[]},{"Name":"b2","Optional":false,"Host":"out/soong/b2.jar","Device":"/system/b2.jar","Subcontexts":[]},{"Name":"c2","Optional":false,"Host":"out/soong/c2.jar","Device":"/system/c2.jar","Subcontexts":[{"Name":"a1","Optional":false,"Host":"out/soong/a1.jar","Device":"/system/a1.jar","Subcontexts":[]},{"Name":"b1","Optional":false,"Host":"out/soong/b1.jar","Device":"/system/b1.jar","Subcontexts":[]}]}]},{"Name":"f","Optional":false,"Host":"out/soong/f.jar","Device":"/system/f.jar","Subcontexts":[]},{"Name":"a3","Optional":false,"Host":"out/soong/a3.jar","Device":"/system/a3.jar","Subcontexts":[]},{"Name":"b3","Optional":false,"Host":"out/soong/b3.jar","Device":"/system/b3.jar","Subcontexts":[]}]}' | 
|  | `), m.DumpForFlag()) | 
|  | }) | 
|  |  | 
|  | // Test for libraries that are added by the manifest_fixer. | 
|  | t.Run("uses libs", func(t *testing.T) { | 
|  | wantUsesLibsReq := []string{"a'", "b", "c", "d", "f", "a3", "b3"} | 
|  | wantUsesLibsOpt := []string{} | 
|  | if !reflect.DeepEqual(wantUsesLibsReq, haveUsesLibsReq) { | 
|  | t.Errorf("\nwant required uses libs: %s\nhave required uses libs: %s", wantUsesLibsReq, haveUsesLibsReq) | 
|  | } | 
|  | if !reflect.DeepEqual(wantUsesLibsOpt, haveUsesLibsOpt) { | 
|  | t.Errorf("\nwant optional uses libs: %s\nhave optional uses libs: %s", wantUsesLibsOpt, haveUsesLibsOpt) | 
|  | } | 
|  | }) | 
|  | } | 
|  |  | 
|  | func TestCLCJson(t *testing.T) { | 
|  | ctx := testContext() | 
|  | optional := false | 
|  | m := make(ClassLoaderContextMap) | 
|  | m.AddContext(ctx, 28, "a", optional, buildPath(ctx, "a"), installPath(ctx, "a"), nil) | 
|  | m.AddContext(ctx, 29, "b", optional, buildPath(ctx, "b"), installPath(ctx, "b"), nil) | 
|  | m.AddContext(ctx, 30, "c", optional, buildPath(ctx, "c"), installPath(ctx, "c"), nil) | 
|  | m.AddContext(ctx, AnySdkVersion, "d", optional, buildPath(ctx, "d"), installPath(ctx, "d"), nil) | 
|  | jsonCLC := toJsonClassLoaderContext(m) | 
|  | restored := fromJsonClassLoaderContext(ctx, jsonCLC) | 
|  | android.AssertIntEquals(t, "The size of the maps should be the same.", len(m), len(restored)) | 
|  | for k := range m { | 
|  | a, _ := m[k] | 
|  | b, ok := restored[k] | 
|  | android.AssertBoolEquals(t, "The both maps should have the same keys.", ok, true) | 
|  | android.AssertIntEquals(t, "The size of the elements should be the same.", len(a), len(b)) | 
|  | for i, elemA := range a { | 
|  | before := fmt.Sprintf("%v", *elemA) | 
|  | after := fmt.Sprintf("%v", *b[i]) | 
|  | android.AssertStringEquals(t, "The content should be the same.", before, after) | 
|  | } | 
|  | } | 
|  | } | 
|  |  | 
|  | // Test that unknown library paths cause a validation error. | 
|  | func testCLCUnknownPath(t *testing.T, whichPath string) { | 
|  | ctx := testContext() | 
|  | optional := false | 
|  |  | 
|  | m := make(ClassLoaderContextMap) | 
|  | if whichPath == "build" { | 
|  | m.AddContext(ctx, AnySdkVersion, "a", optional, nil, nil, nil) | 
|  | } else { | 
|  | m.AddContext(ctx, AnySdkVersion, "a", optional, buildPath(ctx, "a"), nil, nil) | 
|  | } | 
|  |  | 
|  | // The library should be added to <uses-library> tags by the manifest_fixer. | 
|  | t.Run("uses libs", func(t *testing.T) { | 
|  | haveUsesLibsReq, haveUsesLibsOpt := m.UsesLibs() | 
|  | wantUsesLibsReq := []string{"a"} | 
|  | wantUsesLibsOpt := []string{} | 
|  | if !reflect.DeepEqual(wantUsesLibsReq, haveUsesLibsReq) { | 
|  | t.Errorf("\nwant required uses libs: %s\nhave required uses libs: %s", wantUsesLibsReq, haveUsesLibsReq) | 
|  | } | 
|  | if !reflect.DeepEqual(wantUsesLibsOpt, haveUsesLibsOpt) { | 
|  | t.Errorf("\nwant optional uses libs: %s\nhave optional uses libs: %s", wantUsesLibsOpt, haveUsesLibsOpt) | 
|  | } | 
|  | }) | 
|  |  | 
|  | // But CLC cannot be constructed: there is a validation error. | 
|  | _, err := validateClassLoaderContext(m) | 
|  | checkError(t, err, fmt.Sprintf("invalid %s path for <uses-library> \"a\"", whichPath)) | 
|  | } | 
|  |  | 
|  | // Test that unknown build path is an error. | 
|  | func TestCLCUnknownBuildPath(t *testing.T) { | 
|  | testCLCUnknownPath(t, "build") | 
|  | } | 
|  |  | 
|  | // Test that unknown install path is an error. | 
|  | func TestCLCUnknownInstallPath(t *testing.T) { | 
|  | testCLCUnknownPath(t, "install") | 
|  | } | 
|  |  | 
|  | // An attempt to add conditional nested subcontext should fail. | 
|  | func TestCLCNestedConditional(t *testing.T) { | 
|  | ctx := testContext() | 
|  | optional := false | 
|  | m1 := make(ClassLoaderContextMap) | 
|  | m1.AddContext(ctx, 42, "a", optional, buildPath(ctx, "a"), installPath(ctx, "a"), nil) | 
|  | m := make(ClassLoaderContextMap) | 
|  | err := m.addContext(ctx, AnySdkVersion, "b", optional, buildPath(ctx, "b"), installPath(ctx, "b"), m1) | 
|  | checkError(t, err, "nested class loader context shouldn't have conditional part") | 
|  | } | 
|  |  | 
|  | func TestCLCMExcludeLibs(t *testing.T) { | 
|  | ctx := testContext() | 
|  | const optional = false | 
|  |  | 
|  | excludeLibs := func(t *testing.T, m ClassLoaderContextMap, excluded_libs ...string) ClassLoaderContextMap { | 
|  | // Dump the CLCM before creating a new copy that excludes a specific set of libraries. | 
|  | before := m.Dump() | 
|  |  | 
|  | // Create a new CLCM that excludes some libraries. | 
|  | c := m.ExcludeLibs(excluded_libs) | 
|  |  | 
|  | // Make sure that the original CLCM was not changed. | 
|  | after := m.Dump() | 
|  | android.AssertStringEquals(t, "input CLCM modified", before, after) | 
|  |  | 
|  | return c | 
|  | } | 
|  |  | 
|  | t.Run("exclude nothing", func(t *testing.T) { | 
|  | m := make(ClassLoaderContextMap) | 
|  | m.AddContext(ctx, 28, "a", optional, buildPath(ctx, "a"), installPath(ctx, "a"), nil) | 
|  |  | 
|  | a := excludeLibs(t, m) | 
|  |  | 
|  | android.AssertStringEquals(t, "output CLCM ", `{ | 
|  | "28": [ | 
|  | { | 
|  | "Name": "a", | 
|  | "Optional": false, | 
|  | "Host": "out/soong/a.jar", | 
|  | "Device": "/system/a.jar", | 
|  | "Subcontexts": [] | 
|  | } | 
|  | ] | 
|  | }`, a.Dump()) | 
|  | }) | 
|  |  | 
|  | t.Run("one item from list", func(t *testing.T) { | 
|  | m := make(ClassLoaderContextMap) | 
|  | m.AddContext(ctx, 28, "a", optional, buildPath(ctx, "a"), installPath(ctx, "a"), nil) | 
|  | m.AddContext(ctx, 28, "b", optional, buildPath(ctx, "b"), installPath(ctx, "b"), nil) | 
|  |  | 
|  | a := excludeLibs(t, m, "a") | 
|  |  | 
|  | expected := `{ | 
|  | "28": [ | 
|  | { | 
|  | "Name": "b", | 
|  | "Optional": false, | 
|  | "Host": "out/soong/b.jar", | 
|  | "Device": "/system/b.jar", | 
|  | "Subcontexts": [] | 
|  | } | 
|  | ] | 
|  | }` | 
|  | android.AssertStringEquals(t, "output CLCM ", expected, a.Dump()) | 
|  | }) | 
|  |  | 
|  | t.Run("all items from a list", func(t *testing.T) { | 
|  | m := make(ClassLoaderContextMap) | 
|  | m.AddContext(ctx, 28, "a", optional, buildPath(ctx, "a"), installPath(ctx, "a"), nil) | 
|  | m.AddContext(ctx, 28, "b", optional, buildPath(ctx, "b"), installPath(ctx, "b"), nil) | 
|  |  | 
|  | a := excludeLibs(t, m, "a", "b") | 
|  |  | 
|  | android.AssertStringEquals(t, "output CLCM ", `{}`, a.Dump()) | 
|  | }) | 
|  |  | 
|  | t.Run("items from a subcontext", func(t *testing.T) { | 
|  | s := make(ClassLoaderContextMap) | 
|  | s.AddContext(ctx, AnySdkVersion, "b", optional, buildPath(ctx, "b"), installPath(ctx, "b"), nil) | 
|  | s.AddContext(ctx, AnySdkVersion, "c", optional, buildPath(ctx, "c"), installPath(ctx, "c"), nil) | 
|  |  | 
|  | m := make(ClassLoaderContextMap) | 
|  | m.AddContext(ctx, 28, "a", optional, buildPath(ctx, "a"), installPath(ctx, "a"), s) | 
|  |  | 
|  | a := excludeLibs(t, m, "b") | 
|  |  | 
|  | android.AssertStringEquals(t, "output CLCM ", `{ | 
|  | "28": [ | 
|  | { | 
|  | "Name": "a", | 
|  | "Optional": false, | 
|  | "Host": "out/soong/a.jar", | 
|  | "Device": "/system/a.jar", | 
|  | "Subcontexts": [ | 
|  | { | 
|  | "Name": "c", | 
|  | "Optional": false, | 
|  | "Host": "out/soong/c.jar", | 
|  | "Device": "/system/c.jar", | 
|  | "Subcontexts": [] | 
|  | } | 
|  | ] | 
|  | } | 
|  | ] | 
|  | }`, a.Dump()) | 
|  | }) | 
|  | } | 
|  |  | 
|  | // Test that CLC is correctly serialized to JSON. | 
|  | func TestCLCtoJSON(t *testing.T) { | 
|  | ctx := testContext() | 
|  | optional := false | 
|  | m := make(ClassLoaderContextMap) | 
|  | m.AddContext(ctx, 28, "a", optional, buildPath(ctx, "a"), installPath(ctx, "a"), nil) | 
|  | m.AddContext(ctx, AnySdkVersion, "b", optional, buildPath(ctx, "b"), installPath(ctx, "b"), nil) | 
|  | android.AssertStringEquals(t, "output CLCM ", `{ | 
|  | "28": [ | 
|  | { | 
|  | "Name": "a", | 
|  | "Optional": false, | 
|  | "Host": "out/soong/a.jar", | 
|  | "Device": "/system/a.jar", | 
|  | "Subcontexts": [] | 
|  | } | 
|  | ], | 
|  | "any": [ | 
|  | { | 
|  | "Name": "b", | 
|  | "Optional": false, | 
|  | "Host": "out/soong/b.jar", | 
|  | "Device": "/system/b.jar", | 
|  | "Subcontexts": [] | 
|  | } | 
|  | ] | 
|  | }`, m.Dump()) | 
|  | } | 
|  |  | 
|  | func checkError(t *testing.T, have error, want string) { | 
|  | if have == nil { | 
|  | t.Errorf("\nwant error: '%s'\nhave: none", want) | 
|  | } else if msg := have.Error(); !strings.HasPrefix(msg, want) { | 
|  | t.Errorf("\nwant error: '%s'\nhave error: '%s'\n", want, msg) | 
|  | } | 
|  | } | 
|  |  | 
|  | func testContext() android.ModuleInstallPathContext { | 
|  | config := android.TestConfig("out", nil, "", nil) | 
|  | return android.ModuleInstallPathContextForTesting(config) | 
|  | } | 
|  |  | 
|  | func buildPath(ctx android.PathContext, lib string) android.Path { | 
|  | return android.PathForOutput(ctx, lib+".jar") | 
|  | } | 
|  |  | 
|  | func installPath(ctx android.ModuleInstallPathContext, lib string) android.InstallPath { | 
|  | return android.PathForModuleInstall(ctx, lib+".jar") | 
|  | } |