dexpreopt/class_loader_context_test.go - SHIFTPHONES/android_build_soong - Gitiles

 // 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 (
 	"reflect"
 	"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
 	//     ├── b
 	//     ├── c
 	//     ├── d
 	//     │   ├── a2
 	//     │   ├── b2
 	//     │   └── c2
 	//     │       ├── a1
 	//     │       └── b1
 	//     ├── f
 	//     ├── a3
 	//     └── b3
 	//
 	ctx := testContext()

 	m := make(ClassLoaderContextMap)

 	m.AddContext(ctx, "a", buildPath(ctx, "a"), installPath(ctx, "a"))
 	m.AddContext(ctx, "b", buildPath(ctx, "b"), installPath(ctx, "b"))

 	// "Maybe" variant in the good case: add as usual.
 	c := "c"
 	m.MaybeAddContext(ctx, &c, buildPath(ctx, "c"), installPath(ctx, "c"))

 	// "Maybe" variant in the bad case: don't add library with unknown name, keep going.
 	m.MaybeAddContext(ctx, nil, nil, nil)

 	// Add some libraries with nested subcontexts.

 	m1 := make(ClassLoaderContextMap)
 	m1.AddContext(ctx, "a1", buildPath(ctx, "a1"), installPath(ctx, "a1"))
 	m1.AddContext(ctx, "b1", buildPath(ctx, "b1"), installPath(ctx, "b1"))

 	m2 := make(ClassLoaderContextMap)
 	m2.AddContext(ctx, "a2", buildPath(ctx, "a2"), installPath(ctx, "a2"))
 	m2.AddContext(ctx, "b2", buildPath(ctx, "b2"), installPath(ctx, "b2"))
 	m2.AddContextForSdk(ctx, AnySdkVersion, "c2", buildPath(ctx, "c2"), installPath(ctx, "c2"), m1)

 	m3 := make(ClassLoaderContextMap)
 	m3.AddContext(ctx, "a3", buildPath(ctx, "a3"), installPath(ctx, "a3"))
 	m3.AddContext(ctx, "b3", buildPath(ctx, "b3"), installPath(ctx, "b3"))

 	m.AddContextForSdk(ctx, AnySdkVersion, "d", 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.AddContextForSdk(ctx, 42, "f", buildPath(ctx, "f"), installPath(ctx, "f"), nil)
 	m.AddContextForSdk(ctx, AnySdkVersion, "f", 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.AddContextForSdk(ctx, 29, AndroidHidlManager, buildPath(ctx, AndroidHidlManager), nil, nil)
 	m.AddContextForSdk(ctx, 29, AndroidHidlBase, 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.AddContextForSdk(ctx, 30, AndroidTestMock, buildPath(ctx, AndroidTestMock), nil, nil)

 	valid, validationError := validateClassLoaderContext(m)

 	fixClassLoaderContext(m)

 	var haveStr string
 	var havePaths android.Paths
 	var haveUsesLibs []string
 	if valid && validationError == nil {
 		haveStr, havePaths = ComputeClassLoaderContext(m)
 		haveUsesLibs = 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 class loader context structure is correct.
 	t.Run("string", func(t *testing.T) {
 		wantStr := " --host-context-for-sdk 29 " +
 			"PCL[out/" + AndroidHidlManager + ".jar]#" +
 			"PCL[out/" + AndroidHidlBase + ".jar]" +
 			" --target-context-for-sdk 29 " +
 			"PCL[/system/framework/" + AndroidHidlManager + ".jar]#" +
 			"PCL[/system/framework/" + AndroidHidlBase + ".jar]" +
 			" --host-context-for-sdk any " +
 			"PCL[out/a.jar]#PCL[out/b.jar]#PCL[out/c.jar]#PCL[out/d.jar]" +
 			"{PCL[out/a2.jar]#PCL[out/b2.jar]#PCL[out/c2.jar]" +
 			"{PCL[out/a1.jar]#PCL[out/b1.jar]}}#" +
 			"PCL[out/f.jar]#PCL[out/a3.jar]#PCL[out/b3.jar]" +
 			" --target-context-for-sdk any " +
 			"PCL[/system/a.jar]#PCL[/system/b.jar]#PCL[/system/c.jar]#PCL[/system/d.jar]" +
 			"{PCL[/system/a2.jar]#PCL[/system/b2.jar]#PCL[/system/c2.jar]" +
 			"{PCL[/system/a1.jar]#PCL[/system/b1.jar]}}#" +
 			"PCL[/system/f.jar]#PCL[/system/a3.jar]#PCL[/system/b3.jar]"
 		if wantStr != haveStr {
 			t.Errorf("\nwant class loader context: %s\nhave class loader context: %s", wantStr, haveStr)
 		}
 	})

 	// Test that all expected build paths are gathered.
 	t.Run("paths", func(t *testing.T) {
 		wantPaths := []string{
 			"out/android.hidl.manager-V1.0-java.jar", "out/android.hidl.base-V1.0-java.jar",
 			"out/a.jar", "out/b.jar", "out/c.jar", "out/d.jar",
 			"out/a2.jar", "out/b2.jar", "out/c2.jar",
 			"out/a1.jar", "out/b1.jar",
 			"out/f.jar", "out/a3.jar", "out/b3.jar",
 		}
 		if !reflect.DeepEqual(wantPaths, havePaths.Strings()) {
 			t.Errorf("\nwant paths: %s\nhave paths: %s", wantPaths, havePaths)
 		}
 	})

 	// Test for libraries that are added by the manifest_fixer.
 	t.Run("uses libs", func(t *testing.T) {
 		wantUsesLibs := []string{"a", "b", "c", "d", "a2", "b2", "c2", "a1", "b1", "f", "a3", "b3"}
 		if !reflect.DeepEqual(wantUsesLibs, haveUsesLibs) {
 			t.Errorf("\nwant uses libs: %s\nhave uses libs: %s", wantUsesLibs, haveUsesLibs)
 		}
 	})
 }

 // Test that an unexpected unknown build path causes immediate error.
 func TestCLCUnknownBuildPath(t *testing.T) {
 	ctx := testContext()
 	m := make(ClassLoaderContextMap)
 	err := m.addContext(ctx, AnySdkVersion, "a", nil, nil, true, nil)
 	checkError(t, err, "unknown build path to <uses-library> \"a\"")
 }

 // Test that an unexpected unknown install path causes immediate error.
 func TestCLCUnknownInstallPath(t *testing.T) {
 	ctx := testContext()
 	m := make(ClassLoaderContextMap)
 	err := m.addContext(ctx, AnySdkVersion, "a", buildPath(ctx, "a"), nil, true, nil)
 	checkError(t, err, "unknown install path to <uses-library> \"a\"")
 }

 func TestCLCMaybeAdd(t *testing.T) {
 	ctx := testContext()

 	m := make(ClassLoaderContextMap)
 	a := "a"
 	m.MaybeAddContext(ctx, &a, nil, nil)

 	// The library should be added to <uses-library> tags by the manifest_fixer.
 	t.Run("maybe add", func(t *testing.T) {
 		haveUsesLibs := m.UsesLibs()
 		wantUsesLibs := []string{"a"}
 		if !reflect.DeepEqual(wantUsesLibs, haveUsesLibs) {
 			t.Errorf("\nwant uses libs: %s\nhave uses libs: %s", wantUsesLibs, haveUsesLibs)
 		}
 	})

 	// But class loader context in such cases should raise an error on validation.
 	t.Run("validate", func(t *testing.T) {
 		_, err := validateClassLoaderContext(m)
 		checkError(t, err, "invalid path for <uses-library> \"a\"")
 	})
 }

 // An attempt to add conditional nested subcontext should fail.
 func TestCLCNestedConditional(t *testing.T) {
 	ctx := testContext()
 	m1 := make(ClassLoaderContextMap)
 	m1.AddContextForSdk(ctx, 42, "a", buildPath(ctx, "a"), installPath(ctx, "a"), nil)
 	m := make(ClassLoaderContextMap)
 	err := m.addContext(ctx, AnySdkVersion, "b", buildPath(ctx, "b"), installPath(ctx, "b"), true, m1)
 	checkError(t, err, "nested class loader context shouldn't have conditional part")
 }

 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")
 }
	// 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 (
	"reflect"
	"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
	// ├── b
	// ├── c
	// ├── d
	// │ ├── a2
	// │ ├── b2
	// │ └── c2
	// │ ├── a1
	// │ └── b1
	// ├── f
	// ├── a3
	// └── b3
	//
	ctx := testContext()

	m := make(ClassLoaderContextMap)

	m.AddContext(ctx, "a", buildPath(ctx, "a"), installPath(ctx, "a"))
	m.AddContext(ctx, "b", buildPath(ctx, "b"), installPath(ctx, "b"))

	// "Maybe" variant in the good case: add as usual.
	c := "c"
	m.MaybeAddContext(ctx, &c, buildPath(ctx, "c"), installPath(ctx, "c"))

	// "Maybe" variant in the bad case: don't add library with unknown name, keep going.
	m.MaybeAddContext(ctx, nil, nil, nil)

	// Add some libraries with nested subcontexts.

	m1 := make(ClassLoaderContextMap)
	m1.AddContext(ctx, "a1", buildPath(ctx, "a1"), installPath(ctx, "a1"))
	m1.AddContext(ctx, "b1", buildPath(ctx, "b1"), installPath(ctx, "b1"))

	m2 := make(ClassLoaderContextMap)
	m2.AddContext(ctx, "a2", buildPath(ctx, "a2"), installPath(ctx, "a2"))
	m2.AddContext(ctx, "b2", buildPath(ctx, "b2"), installPath(ctx, "b2"))
	m2.AddContextForSdk(ctx, AnySdkVersion, "c2", buildPath(ctx, "c2"), installPath(ctx, "c2"), m1)

	m3 := make(ClassLoaderContextMap)
	m3.AddContext(ctx, "a3", buildPath(ctx, "a3"), installPath(ctx, "a3"))
	m3.AddContext(ctx, "b3", buildPath(ctx, "b3"), installPath(ctx, "b3"))

	m.AddContextForSdk(ctx, AnySdkVersion, "d", 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.AddContextForSdk(ctx, 42, "f", buildPath(ctx, "f"), installPath(ctx, "f"), nil)
	m.AddContextForSdk(ctx, AnySdkVersion, "f", 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.AddContextForSdk(ctx, 29, AndroidHidlManager, buildPath(ctx, AndroidHidlManager), nil, nil)
	m.AddContextForSdk(ctx, 29, AndroidHidlBase, 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.AddContextForSdk(ctx, 30, AndroidTestMock, buildPath(ctx, AndroidTestMock), nil, nil)

	valid, validationError := validateClassLoaderContext(m)

	fixClassLoaderContext(m)

	var haveStr string
	var havePaths android.Paths
	var haveUsesLibs []string
	if valid && validationError == nil {
	haveStr, havePaths = ComputeClassLoaderContext(m)
	haveUsesLibs = 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 class loader context structure is correct.
	t.Run("string", func(t *testing.T) {
	wantStr := " --host-context-for-sdk 29 " +
	"PCL[out/" + AndroidHidlManager + ".jar]#" +
	"PCL[out/" + AndroidHidlBase + ".jar]" +
	" --target-context-for-sdk 29 " +
	"PCL[/system/framework/" + AndroidHidlManager + ".jar]#" +
	"PCL[/system/framework/" + AndroidHidlBase + ".jar]" +
	" --host-context-for-sdk any " +
	"PCL[out/a.jar]#PCL[out/b.jar]#PCL[out/c.jar]#PCL[out/d.jar]" +
	"{PCL[out/a2.jar]#PCL[out/b2.jar]#PCL[out/c2.jar]" +
	"{PCL[out/a1.jar]#PCL[out/b1.jar]}}#" +
	"PCL[out/f.jar]#PCL[out/a3.jar]#PCL[out/b3.jar]" +
	" --target-context-for-sdk any " +
	"PCL[/system/a.jar]#PCL[/system/b.jar]#PCL[/system/c.jar]#PCL[/system/d.jar]" +
	"{PCL[/system/a2.jar]#PCL[/system/b2.jar]#PCL[/system/c2.jar]" +
	"{PCL[/system/a1.jar]#PCL[/system/b1.jar]}}#" +
	"PCL[/system/f.jar]#PCL[/system/a3.jar]#PCL[/system/b3.jar]"
	if wantStr != haveStr {
	t.Errorf("\nwant class loader context: %s\nhave class loader context: %s", wantStr, haveStr)
	}
	})

	// Test that all expected build paths are gathered.
	t.Run("paths", func(t *testing.T) {
	wantPaths := []string{
	"out/android.hidl.manager-V1.0-java.jar", "out/android.hidl.base-V1.0-java.jar",
	"out/a.jar", "out/b.jar", "out/c.jar", "out/d.jar",
	"out/a2.jar", "out/b2.jar", "out/c2.jar",
	"out/a1.jar", "out/b1.jar",
	"out/f.jar", "out/a3.jar", "out/b3.jar",
	}
	if !reflect.DeepEqual(wantPaths, havePaths.Strings()) {
	t.Errorf("\nwant paths: %s\nhave paths: %s", wantPaths, havePaths)
	}
	})

	// Test for libraries that are added by the manifest_fixer.
	t.Run("uses libs", func(t *testing.T) {
	wantUsesLibs := []string{"a", "b", "c", "d", "a2", "b2", "c2", "a1", "b1", "f", "a3", "b3"}
	if !reflect.DeepEqual(wantUsesLibs, haveUsesLibs) {
	t.Errorf("\nwant uses libs: %s\nhave uses libs: %s", wantUsesLibs, haveUsesLibs)
	}
	})
	}

	// Test that an unexpected unknown build path causes immediate error.
	func TestCLCUnknownBuildPath(t *testing.T) {
	ctx := testContext()
	m := make(ClassLoaderContextMap)
	err := m.addContext(ctx, AnySdkVersion, "a", nil, nil, true, nil)
	checkError(t, err, "unknown build path to <uses-library> \"a\"")
	}

	// Test that an unexpected unknown install path causes immediate error.
	func TestCLCUnknownInstallPath(t *testing.T) {
	ctx := testContext()
	m := make(ClassLoaderContextMap)
	err := m.addContext(ctx, AnySdkVersion, "a", buildPath(ctx, "a"), nil, true, nil)
	checkError(t, err, "unknown install path to <uses-library> \"a\"")
	}

	func TestCLCMaybeAdd(t *testing.T) {
	ctx := testContext()

	m := make(ClassLoaderContextMap)
	a := "a"
	m.MaybeAddContext(ctx, &a, nil, nil)

	// The library should be added to <uses-library> tags by the manifest_fixer.
	t.Run("maybe add", func(t *testing.T) {
	haveUsesLibs := m.UsesLibs()
	wantUsesLibs := []string{"a"}
	if !reflect.DeepEqual(wantUsesLibs, haveUsesLibs) {
	t.Errorf("\nwant uses libs: %s\nhave uses libs: %s", wantUsesLibs, haveUsesLibs)
	}
	})

	// But class loader context in such cases should raise an error on validation.
	t.Run("validate", func(t *testing.T) {
	_, err := validateClassLoaderContext(m)
	checkError(t, err, "invalid path for <uses-library> \"a\"")
	})
	}

	// An attempt to add conditional nested subcontext should fail.
	func TestCLCNestedConditional(t *testing.T) {
	ctx := testContext()
	m1 := make(ClassLoaderContextMap)
	m1.AddContextForSdk(ctx, 42, "a", buildPath(ctx, "a"), installPath(ctx, "a"), nil)
	m := make(ClassLoaderContextMap)
	err := m.addContext(ctx, AnySdkVersion, "b", buildPath(ctx, "b"), installPath(ctx, "b"), true, m1)
	checkError(t, err, "nested class loader context shouldn't have conditional part")
	}

	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")
	}