blob: 7440869f33570fcf218246f5c2d536b31f31b0b6 [file] [log] [blame]
Colin Crosscec81712017-07-13 14:43:27 -07001// Copyright 2017 Google Inc. All rights reserved.
2//
3// Licensed under the Apache License, Version 2.0 (the "License");
4// you may not use this file except in compliance with the License.
5// You may obtain a copy of the License at
6//
7// http://www.apache.org/licenses/LICENSE-2.0
8//
9// Unless required by applicable law or agreed to in writing, software
10// distributed under the License is distributed on an "AS IS" BASIS,
11// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12// See the License for the specific language governing permissions and
13// limitations under the License.
14
15package android
16
17import (
Martin Stjernholm1ebef5b2022-02-10 23:34:28 +000018 "bytes"
Colin Crosscec81712017-07-13 14:43:27 -070019 "fmt"
Paul Duffin9b478b02019-12-10 13:41:51 +000020 "path/filepath"
Logan Chienee97c3e2018-03-12 16:34:26 +080021 "regexp"
Martin Stjernholm4c021242020-05-13 01:13:50 +010022 "sort"
Colin Crosscec81712017-07-13 14:43:27 -070023 "strings"
Paul Duffin281deb22021-03-06 20:29:19 +000024 "sync"
Logan Chien42039712018-03-12 16:29:17 +080025 "testing"
Colin Crosscec81712017-07-13 14:43:27 -070026
Martin Stjernholm1ebef5b2022-02-10 23:34:28 +000027 mkparser "android/soong/androidmk/parser"
28
Colin Crosscec81712017-07-13 14:43:27 -070029 "github.com/google/blueprint"
Paul Duffin25259e92021-03-07 15:45:56 +000030 "github.com/google/blueprint/proptools"
Colin Crosscec81712017-07-13 14:43:27 -070031)
32
Paul Duffin3f7bf9f2022-11-08 12:21:15 +000033func newTestContextForFixture(config Config) *TestContext {
Jeff Gastonb274ed32017-12-01 17:10:33 -080034 ctx := &TestContext{
Paul Duffin3f7bf9f2022-11-08 12:21:15 +000035 Context: &Context{blueprint.NewContext(), config},
Jeff Gastonb274ed32017-12-01 17:10:33 -080036 }
37
Colin Cross1b488422019-03-04 22:33:56 -080038 ctx.postDeps = append(ctx.postDeps, registerPathDepsMutator)
39
Colin Crossae8600b2020-10-29 17:09:13 -070040 ctx.SetFs(ctx.config.fs)
41 if ctx.config.mockBpList != "" {
42 ctx.SetModuleListFile(ctx.config.mockBpList)
43 }
44
Jeff Gaston088e29e2017-11-29 16:47:17 -080045 return ctx
Colin Crosscec81712017-07-13 14:43:27 -070046}
47
Paul Duffin3f7bf9f2022-11-08 12:21:15 +000048func NewTestContext(config Config) *TestContext {
49 ctx := newTestContextForFixture(config)
50
51 nameResolver := NewNameResolver(config)
52 ctx.NameResolver = nameResolver
53 ctx.SetNameInterface(nameResolver)
54
55 return ctx
56}
57
Paul Duffina560d5a2021-02-28 01:38:51 +000058var PrepareForTestWithArchMutator = GroupFixturePreparers(
Paul Duffin35816122021-02-24 01:49:52 +000059 // Configure architecture targets in the fixture config.
60 FixtureModifyConfig(modifyTestConfigToSupportArchMutator),
61
62 // Add the arch mutator to the context.
63 FixtureRegisterWithContext(func(ctx RegistrationContext) {
64 ctx.PreDepsMutators(registerArchMutator)
65 }),
66)
67
68var PrepareForTestWithDefaults = FixtureRegisterWithContext(func(ctx RegistrationContext) {
69 ctx.PreArchMutators(RegisterDefaultsPreArchMutators)
70})
71
72var PrepareForTestWithComponentsMutator = FixtureRegisterWithContext(func(ctx RegistrationContext) {
73 ctx.PreArchMutators(RegisterComponentsMutator)
74})
75
76var PrepareForTestWithPrebuilts = FixtureRegisterWithContext(RegisterPrebuiltMutators)
77
78var PrepareForTestWithOverrides = FixtureRegisterWithContext(func(ctx RegistrationContext) {
79 ctx.PostDepsMutators(RegisterOverridePostDepsMutators)
80})
81
Paul Duffine96108d2021-05-06 16:39:27 +010082var PrepareForTestWithLicenses = GroupFixturePreparers(
83 FixtureRegisterWithContext(RegisterLicenseKindBuildComponents),
84 FixtureRegisterWithContext(RegisterLicenseBuildComponents),
85 FixtureRegisterWithContext(registerLicenseMutators),
86)
87
Bob Badour05079212022-05-20 16:41:39 -070088var PrepareForTestWithGenNotice = FixtureRegisterWithContext(RegisterGenNoticeBuildComponents)
89
Paul Duffine96108d2021-05-06 16:39:27 +010090func registerLicenseMutators(ctx RegistrationContext) {
91 ctx.PreArchMutators(RegisterLicensesPackageMapper)
92 ctx.PreArchMutators(RegisterLicensesPropertyGatherer)
93 ctx.PostDepsMutators(RegisterLicensesDependencyChecker)
94}
95
96var PrepareForTestWithLicenseDefaultModules = GroupFixturePreparers(
97 FixtureAddTextFile("build/soong/licenses/Android.bp", `
98 license {
99 name: "Android-Apache-2.0",
100 package_name: "Android",
101 license_kinds: ["SPDX-license-identifier-Apache-2.0"],
102 copyright_notice: "Copyright (C) The Android Open Source Project",
103 license_text: ["LICENSE"],
104 }
105
106 license_kind {
107 name: "SPDX-license-identifier-Apache-2.0",
108 conditions: ["notice"],
109 url: "https://spdx.org/licenses/Apache-2.0.html",
110 }
111
112 license_kind {
113 name: "legacy_unencumbered",
114 conditions: ["unencumbered"],
115 }
116 `),
117 FixtureAddFile("build/soong/licenses/LICENSE", nil),
118)
119
Paul Duffin4fbfb592021-07-09 16:47:38 +0100120var PrepareForTestWithNamespace = FixtureRegisterWithContext(func(ctx RegistrationContext) {
121 registerNamespaceBuildComponents(ctx)
122 ctx.PreArchMutators(RegisterNamespaceMutator)
123})
124
Martin Stjernholm1ebef5b2022-02-10 23:34:28 +0000125var PrepareForTestWithMakevars = FixtureRegisterWithContext(func(ctx RegistrationContext) {
126 ctx.RegisterSingletonType("makevars", makeVarsSingletonFunc)
127})
128
Kiyoung Kimfaf6af32024-08-12 11:15:19 +0900129var PrepareForTestVintfFragmentModules = FixtureRegisterWithContext(func(ctx RegistrationContext) {
130 registerVintfFragmentComponents(ctx)
131})
132
Paul Duffinec3292b2021-03-09 01:01:31 +0000133// Test fixture preparer that will register most java build components.
134//
135// Singletons and mutators should only be added here if they are needed for a majority of java
136// module types, otherwise they should be added under a separate preparer to allow them to be
137// selected only when needed to reduce test execution time.
138//
139// Module types do not have much of an overhead unless they are used so this should include as many
140// module types as possible. The exceptions are those module types that require mutators and/or
141// singletons in order to function in which case they should be kept together in a separate
142// preparer.
143//
144// The mutators in this group were chosen because they are needed by the vast majority of tests.
145var PrepareForTestWithAndroidBuildComponents = GroupFixturePreparers(
Paul Duffin530483c2021-03-07 13:20:38 +0000146 // Sorted alphabetically as the actual order does not matter as tests automatically enforce the
147 // correct order.
Paul Duffin35816122021-02-24 01:49:52 +0000148 PrepareForTestWithArchMutator,
Paul Duffin35816122021-02-24 01:49:52 +0000149 PrepareForTestWithComponentsMutator,
Paul Duffin530483c2021-03-07 13:20:38 +0000150 PrepareForTestWithDefaults,
Paul Duffin35816122021-02-24 01:49:52 +0000151 PrepareForTestWithFilegroup,
Paul Duffin530483c2021-03-07 13:20:38 +0000152 PrepareForTestWithOverrides,
153 PrepareForTestWithPackageModule,
154 PrepareForTestWithPrebuilts,
155 PrepareForTestWithVisibility,
Kiyoung Kimfaf6af32024-08-12 11:15:19 +0900156 PrepareForTestVintfFragmentModules,
Paul Duffin35816122021-02-24 01:49:52 +0000157)
158
Paul Duffinec3292b2021-03-09 01:01:31 +0000159// Prepares an integration test with all build components from the android package.
160//
161// This should only be used by tests that want to run with as much of the build enabled as possible.
162var PrepareForIntegrationTestWithAndroid = GroupFixturePreparers(
163 PrepareForTestWithAndroidBuildComponents,
164)
165
Paul Duffin25259e92021-03-07 15:45:56 +0000166// Prepares a test that may be missing dependencies by setting allow_missing_dependencies to
167// true.
168var PrepareForTestWithAllowMissingDependencies = GroupFixturePreparers(
169 FixtureModifyProductVariables(func(variables FixtureProductVariables) {
170 variables.Allow_missing_dependencies = proptools.BoolPtr(true)
171 }),
172 FixtureModifyContext(func(ctx *TestContext) {
173 ctx.SetAllowMissingDependencies(true)
174 }),
175)
176
Paul Duffin76e5c8a2021-03-20 14:19:46 +0000177// Prepares a test that disallows non-existent paths.
178var PrepareForTestDisallowNonExistentPaths = FixtureModifyConfig(func(config Config) {
179 config.TestAllowNonExistentPaths = false
180})
181
Colin Crossa66b4632024-08-08 15:50:47 -0700182// PrepareForTestWithBuildFlag returns a FixturePreparer that sets the given flag to the given value.
183func PrepareForTestWithBuildFlag(flag, value string) FixturePreparer {
184 return FixtureModifyProductVariables(func(variables FixtureProductVariables) {
185 if variables.BuildFlags == nil {
186 variables.BuildFlags = make(map[string]string)
187 }
188 variables.BuildFlags[flag] = value
189 })
190}
191
Colin Crossae8600b2020-10-29 17:09:13 -0700192func NewTestArchContext(config Config) *TestContext {
193 ctx := NewTestContext(config)
Colin Crossae4c6182017-09-15 17:33:55 -0700194 ctx.preDeps = append(ctx.preDeps, registerArchMutator)
195 return ctx
196}
197
Colin Crosscec81712017-07-13 14:43:27 -0700198type TestContext struct {
Colin Cross4c83e5c2019-02-25 14:54:28 -0800199 *Context
Colin Crossf22fe412024-10-01 14:02:12 -0700200 preArch, preDeps, postDeps, postApex, finalDeps []RegisterMutatorFunc
201 NameResolver *NameResolver
Paul Duffin281deb22021-03-06 20:29:19 +0000202
Cole Faustae6cda62023-11-01 15:32:40 -0700203 // The list of singletons registered for the test.
204 singletons sortableComponents
Paul Duffind182fb32021-03-07 12:24:44 +0000205
Cole Faustae6cda62023-11-01 15:32:40 -0700206 // The order in which the mutators and singletons will be run in this test
Paul Duffin41d77c72021-03-07 12:23:48 +0000207 // context; for debugging.
Cole Faustae6cda62023-11-01 15:32:40 -0700208 mutatorOrder, singletonOrder []string
Colin Crosscec81712017-07-13 14:43:27 -0700209}
210
211func (ctx *TestContext) PreArchMutators(f RegisterMutatorFunc) {
212 ctx.preArch = append(ctx.preArch, f)
213}
214
Paul Duffina80ef842020-01-14 12:09:36 +0000215func (ctx *TestContext) HardCodedPreArchMutators(f RegisterMutatorFunc) {
216 // Register mutator function as normal for testing.
217 ctx.PreArchMutators(f)
218}
219
Yu Liu663e4502024-08-12 18:23:59 +0000220func (ctx *TestContext) otherModuleProvider(m blueprint.Module, p blueprint.AnyProviderKey) (any, bool) {
Liz Kammer92c72592022-10-31 14:44:28 -0400221 return ctx.Context.ModuleProvider(m, p)
222}
223
Colin Crosscec81712017-07-13 14:43:27 -0700224func (ctx *TestContext) PreDepsMutators(f RegisterMutatorFunc) {
225 ctx.preDeps = append(ctx.preDeps, f)
226}
227
228func (ctx *TestContext) PostDepsMutators(f RegisterMutatorFunc) {
229 ctx.postDeps = append(ctx.postDeps, f)
230}
231
Colin Crossf22fe412024-10-01 14:02:12 -0700232func (ctx *TestContext) PostApexMutators(f RegisterMutatorFunc) {
233 ctx.postApex = append(ctx.postApex, f)
234}
235
Martin Stjernholm710ec3a2020-01-16 15:12:04 +0000236func (ctx *TestContext) FinalDepsMutators(f RegisterMutatorFunc) {
237 ctx.finalDeps = append(ctx.finalDeps, f)
238}
239
Colin Cross3c0a83d2023-12-12 14:13:26 -0800240func (ctx *TestContext) OtherModuleProviderAdaptor() OtherModuleProviderContext {
241 return NewOtherModuleProviderAdaptor(func(module blueprint.Module, provider blueprint.AnyProviderKey) (any, bool) {
Yu Liu663e4502024-08-12 18:23:59 +0000242 return ctx.otherModuleProvider(module, provider)
Colin Cross3c0a83d2023-12-12 14:13:26 -0800243 })
244}
245
Cole Faust43ddd082024-06-17 12:32:40 -0700246func (ctx *TestContext) OtherModulePropertyErrorf(module Module, property string, fmt_ string, args ...interface{}) {
247 panic(fmt.Sprintf(fmt_, args...))
248}
249
Paul Duffin281deb22021-03-06 20:29:19 +0000250// registeredComponentOrder defines the order in which a sortableComponent type is registered at
251// runtime and provides support for reordering the components registered for a test in the same
252// way.
253type registeredComponentOrder struct {
254 // The name of the component type, used for error messages.
255 componentType string
256
257 // The names of the registered components in the order in which they were registered.
258 namesInOrder []string
259
260 // Maps from the component name to its position in the runtime ordering.
261 namesToIndex map[string]int
262
263 // A function that defines the order between two named components that can be used to sort a slice
264 // of component names into the same order as they appear in namesInOrder.
265 less func(string, string) bool
266}
267
268// registeredComponentOrderFromExistingOrder takes an existing slice of sortableComponents and
269// creates a registeredComponentOrder that contains a less function that can be used to sort a
270// subset of that list of names so it is in the same order as the original sortableComponents.
271func registeredComponentOrderFromExistingOrder(componentType string, existingOrder sortableComponents) registeredComponentOrder {
272 // Only the names from the existing order are needed for this so create a list of component names
273 // in the correct order.
274 namesInOrder := componentsToNames(existingOrder)
275
276 // Populate the map from name to position in the list.
277 nameToIndex := make(map[string]int)
278 for i, n := range namesInOrder {
279 nameToIndex[n] = i
280 }
281
282 // A function to use to map from a name to an index in the original order.
283 indexOf := func(name string) int {
284 index, ok := nameToIndex[name]
285 if !ok {
286 // Should never happen as tests that use components that are not known at runtime do not sort
287 // so should never use this function.
288 panic(fmt.Errorf("internal error: unknown %s %q should be one of %s", componentType, name, strings.Join(namesInOrder, ", ")))
289 }
290 return index
291 }
292
293 // The less function.
294 less := func(n1, n2 string) bool {
295 i1 := indexOf(n1)
296 i2 := indexOf(n2)
297 return i1 < i2
298 }
299
300 return registeredComponentOrder{
301 componentType: componentType,
302 namesInOrder: namesInOrder,
303 namesToIndex: nameToIndex,
304 less: less,
305 }
306}
307
308// componentsToNames maps from the slice of components to a slice of their names.
309func componentsToNames(components sortableComponents) []string {
310 names := make([]string, len(components))
311 for i, c := range components {
312 names[i] = c.componentName()
313 }
314 return names
315}
316
317// enforceOrdering enforces the supplied components are in the same order as is defined in this
318// object.
319//
320// If the supplied components contains any components that are not registered at runtime, i.e. test
321// specific components, then it is impossible to sort them into an order that both matches the
322// runtime and also preserves the implicit ordering defined in the test. In that case it will not
323// sort the components, instead it will just check that the components are in the correct order.
324//
325// Otherwise, this will sort the supplied components in place.
326func (o *registeredComponentOrder) enforceOrdering(components sortableComponents) {
327 // Check to see if the list of components contains any components that are
328 // not registered at runtime.
329 var unknownComponents []string
330 testOrder := componentsToNames(components)
331 for _, name := range testOrder {
332 if _, ok := o.namesToIndex[name]; !ok {
333 unknownComponents = append(unknownComponents, name)
334 break
335 }
336 }
337
338 // If the slice contains some unknown components then it is not possible to
339 // sort them into an order that matches the runtime while also preserving the
340 // order expected from the test, so in that case don't sort just check that
341 // the order of the known mutators does match.
342 if len(unknownComponents) > 0 {
343 // Check order.
344 o.checkTestOrder(testOrder, unknownComponents)
345 } else {
346 // Sort the components.
347 sort.Slice(components, func(i, j int) bool {
348 n1 := components[i].componentName()
349 n2 := components[j].componentName()
350 return o.less(n1, n2)
351 })
352 }
353}
354
355// checkTestOrder checks that the supplied testOrder matches the one defined by this object,
356// panicking if it does not.
357func (o *registeredComponentOrder) checkTestOrder(testOrder []string, unknownComponents []string) {
358 lastMatchingTest := -1
359 matchCount := 0
360 // Take a copy of the runtime order as it is modified during the comparison.
361 runtimeOrder := append([]string(nil), o.namesInOrder...)
362 componentType := o.componentType
363 for i, j := 0, 0; i < len(testOrder) && j < len(runtimeOrder); {
364 test := testOrder[i]
365 runtime := runtimeOrder[j]
366
367 if test == runtime {
368 testOrder[i] = test + fmt.Sprintf(" <-- matched with runtime %s %d", componentType, j)
369 runtimeOrder[j] = runtime + fmt.Sprintf(" <-- matched with test %s %d", componentType, i)
370 lastMatchingTest = i
371 i += 1
372 j += 1
373 matchCount += 1
374 } else if _, ok := o.namesToIndex[test]; !ok {
375 // The test component is not registered globally so assume it is the correct place, treat it
376 // as having matched and skip it.
377 i += 1
378 matchCount += 1
379 } else {
380 // Assume that the test list is in the same order as the runtime list but the runtime list
381 // contains some components that are not present in the tests. So, skip the runtime component
382 // to try and find the next one that matches the current test component.
383 j += 1
384 }
385 }
386
387 // If every item in the test order was either test specific or matched one in the runtime then
388 // it is in the correct order. Otherwise, it was not so fail.
389 if matchCount != len(testOrder) {
390 // The test component names were not all matched with a runtime component name so there must
391 // either be a component present in the test that is not present in the runtime or they must be
392 // in the wrong order.
393 testOrder[lastMatchingTest+1] = testOrder[lastMatchingTest+1] + " <--- unmatched"
394 panic(fmt.Errorf("the tests uses test specific components %q and so cannot be automatically sorted."+
395 " Unfortunately it uses %s components in the wrong order.\n"+
396 "test order:\n %s\n"+
397 "runtime order\n %s\n",
398 SortedUniqueStrings(unknownComponents),
399 componentType,
400 strings.Join(testOrder, "\n "),
401 strings.Join(runtimeOrder, "\n ")))
402 }
403}
404
405// registrationSorter encapsulates the information needed to ensure that the test mutators are
406// registered, and thereby executed, in the same order as they are at runtime.
407//
408// It MUST be populated lazily AFTER all package initialization has been done otherwise it will
409// only define the order for a subset of all the registered build components that are available for
410// the packages being tested.
411//
412// e.g if this is initialized during say the cc package initialization then any tests run in the
413// java package will not sort build components registered by the java package's init() functions.
414type registrationSorter struct {
415 // Used to ensure that this is only created once.
416 once sync.Once
417
418 // The order of mutators
419 mutatorOrder registeredComponentOrder
Paul Duffin41d77c72021-03-07 12:23:48 +0000420
421 // The order of singletons
422 singletonOrder registeredComponentOrder
Paul Duffin281deb22021-03-06 20:29:19 +0000423}
424
425// populate initializes this structure from globally registered build components.
426//
427// Only the first call has any effect.
428func (s *registrationSorter) populate() {
429 s.once.Do(func() {
430 // Created an ordering from the globally registered mutators.
431 globallyRegisteredMutators := collateGloballyRegisteredMutators()
432 s.mutatorOrder = registeredComponentOrderFromExistingOrder("mutator", globallyRegisteredMutators)
Paul Duffin41d77c72021-03-07 12:23:48 +0000433
434 // Create an ordering from the globally registered singletons.
435 globallyRegisteredSingletons := collateGloballyRegisteredSingletons()
436 s.singletonOrder = registeredComponentOrderFromExistingOrder("singleton", globallyRegisteredSingletons)
Paul Duffin281deb22021-03-06 20:29:19 +0000437 })
438}
439
440// Provides support for enforcing the same order in which build components are registered globally
441// to the order in which they are registered during tests.
442//
443// MUST only be accessed via the globallyRegisteredComponentsOrder func.
444var globalRegistrationSorter registrationSorter
445
446// globallyRegisteredComponentsOrder returns the globalRegistrationSorter after ensuring it is
447// correctly populated.
448func globallyRegisteredComponentsOrder() *registrationSorter {
449 globalRegistrationSorter.populate()
450 return &globalRegistrationSorter
451}
452
Colin Crossae8600b2020-10-29 17:09:13 -0700453func (ctx *TestContext) Register() {
Paul Duffin281deb22021-03-06 20:29:19 +0000454 globalOrder := globallyRegisteredComponentsOrder()
455
Colin Crossf22fe412024-10-01 14:02:12 -0700456 mutators := collateRegisteredMutators(ctx.preArch, ctx.preDeps, ctx.postDeps, ctx.postApex, ctx.finalDeps)
Paul Duffin281deb22021-03-06 20:29:19 +0000457 // Ensure that the mutators used in the test are in the same order as they are used at runtime.
458 globalOrder.mutatorOrder.enforceOrdering(mutators)
Paul Duffinc05b0342021-03-06 13:28:13 +0000459 mutators.registerAll(ctx.Context)
Colin Crosscec81712017-07-13 14:43:27 -0700460
Paul Duffin41d77c72021-03-07 12:23:48 +0000461 // Ensure that the singletons used in the test are in the same order as they are used at runtime.
462 globalOrder.singletonOrder.enforceOrdering(ctx.singletons)
Paul Duffind182fb32021-03-07 12:24:44 +0000463 ctx.singletons.registerAll(ctx.Context)
464
Paul Duffin41d77c72021-03-07 12:23:48 +0000465 // Save the sorted components order away to make them easy to access while debugging.
Paul Duffinf5de6682021-03-08 23:42:10 +0000466 ctx.mutatorOrder = componentsToNames(mutators)
467 ctx.singletonOrder = componentsToNames(singletons)
Colin Cross31a738b2019-12-30 18:45:15 -0800468}
469
470func (ctx *TestContext) ParseFileList(rootDir string, filePaths []string) (deps []string, errs []error) {
471 // This function adapts the old style ParseFileList calls that are spread throughout the tests
472 // to the new style that takes a config.
473 return ctx.Context.ParseFileList(rootDir, filePaths, ctx.config)
474}
475
476func (ctx *TestContext) ParseBlueprintsFiles(rootDir string) (deps []string, errs []error) {
477 // This function adapts the old style ParseBlueprintsFiles calls that are spread throughout the
478 // tests to the new style that takes a config.
479 return ctx.Context.ParseBlueprintsFiles(rootDir, ctx.config)
Colin Cross4b49b762019-11-22 15:25:03 -0800480}
481
482func (ctx *TestContext) RegisterModuleType(name string, factory ModuleFactory) {
483 ctx.Context.RegisterModuleType(name, ModuleFactoryAdaptor(factory))
484}
485
Colin Cross9aed5bc2020-12-28 15:15:34 -0800486func (ctx *TestContext) RegisterSingletonModuleType(name string, factory SingletonModuleFactory) {
487 s, m := SingletonModuleFactoryAdaptor(name, factory)
488 ctx.RegisterSingletonType(name, s)
489 ctx.RegisterModuleType(name, m)
490}
491
LaMont Jonese59c0db2023-05-15 21:50:29 +0000492func (ctx *TestContext) RegisterParallelSingletonModuleType(name string, factory SingletonModuleFactory) {
493 s, m := SingletonModuleFactoryAdaptor(name, factory)
494 ctx.RegisterParallelSingletonType(name, s)
495 ctx.RegisterModuleType(name, m)
496}
497
Colin Cross4b49b762019-11-22 15:25:03 -0800498func (ctx *TestContext) RegisterSingletonType(name string, factory SingletonFactory) {
LaMont Jonese59c0db2023-05-15 21:50:29 +0000499 ctx.singletons = append(ctx.singletons, newSingleton(name, factory, false))
500}
501
502func (ctx *TestContext) RegisterParallelSingletonType(name string, factory SingletonFactory) {
503 ctx.singletons = append(ctx.singletons, newSingleton(name, factory, true))
Colin Crosscec81712017-07-13 14:43:27 -0700504}
505
Martin Stjernholm14cdd712021-09-10 22:39:59 +0100506// ModuleVariantForTests selects a specific variant of the module with the given
507// name by matching the variations map against the variations of each module
508// variant. A module variant matches the map if every variation that exists in
509// both have the same value. Both the module and the map are allowed to have
510// extra variations that the other doesn't have. Panics if not exactly one
511// module variant matches.
512func (ctx *TestContext) ModuleVariantForTests(name string, matchVariations map[string]string) TestingModule {
513 modules := []Module{}
514 ctx.VisitAllModules(func(m blueprint.Module) {
515 if ctx.ModuleName(m) == name {
516 am := m.(Module)
517 amMut := am.base().commonProperties.DebugMutators
518 amVar := am.base().commonProperties.DebugVariations
519 matched := true
520 for i, mut := range amMut {
521 if wantedVar, found := matchVariations[mut]; found && amVar[i] != wantedVar {
522 matched = false
523 break
524 }
525 }
526 if matched {
527 modules = append(modules, am)
528 }
529 }
530 })
531
532 if len(modules) == 0 {
533 // Show all the modules or module variants that do exist.
534 var allModuleNames []string
535 var allVariants []string
536 ctx.VisitAllModules(func(m blueprint.Module) {
537 allModuleNames = append(allModuleNames, ctx.ModuleName(m))
538 if ctx.ModuleName(m) == name {
539 allVariants = append(allVariants, m.(Module).String())
540 }
541 })
542
543 if len(allVariants) == 0 {
544 panic(fmt.Errorf("failed to find module %q. All modules:\n %s",
545 name, strings.Join(SortedUniqueStrings(allModuleNames), "\n ")))
546 } else {
547 sort.Strings(allVariants)
548 panic(fmt.Errorf("failed to find module %q matching %v. All variants:\n %s",
549 name, matchVariations, strings.Join(allVariants, "\n ")))
550 }
551 }
552
553 if len(modules) > 1 {
554 moduleStrings := []string{}
555 for _, m := range modules {
556 moduleStrings = append(moduleStrings, m.String())
557 }
558 sort.Strings(moduleStrings)
559 panic(fmt.Errorf("module %q has more than one variant that match %v:\n %s",
560 name, matchVariations, strings.Join(moduleStrings, "\n ")))
561 }
562
563 return newTestingModule(ctx.config, modules[0])
564}
565
Colin Crosscec81712017-07-13 14:43:27 -0700566func (ctx *TestContext) ModuleForTests(name, variant string) TestingModule {
567 var module Module
568 ctx.VisitAllModules(func(m blueprint.Module) {
569 if ctx.ModuleName(m) == name && ctx.ModuleSubDir(m) == variant {
570 module = m.(Module)
571 }
572 })
573
574 if module == nil {
Jeff Gaston294356f2017-09-27 17:05:30 -0700575 // find all the modules that do exist
Colin Crossbeae6ec2020-08-11 12:02:11 -0700576 var allModuleNames []string
577 var allVariants []string
Jeff Gaston294356f2017-09-27 17:05:30 -0700578 ctx.VisitAllModules(func(m blueprint.Module) {
Colin Crossbeae6ec2020-08-11 12:02:11 -0700579 allModuleNames = append(allModuleNames, ctx.ModuleName(m))
580 if ctx.ModuleName(m) == name {
581 allVariants = append(allVariants, ctx.ModuleSubDir(m))
582 }
Jeff Gaston294356f2017-09-27 17:05:30 -0700583 })
Colin Crossbeae6ec2020-08-11 12:02:11 -0700584 sort.Strings(allVariants)
Jeff Gaston294356f2017-09-27 17:05:30 -0700585
Colin Crossbeae6ec2020-08-11 12:02:11 -0700586 if len(allVariants) == 0 {
587 panic(fmt.Errorf("failed to find module %q. All modules:\n %s",
Martin Stjernholm98e0d882021-09-09 21:34:02 +0100588 name, strings.Join(SortedUniqueStrings(allModuleNames), "\n ")))
Colin Crossbeae6ec2020-08-11 12:02:11 -0700589 } else {
590 panic(fmt.Errorf("failed to find module %q variant %q. All variants:\n %s",
591 name, variant, strings.Join(allVariants, "\n ")))
592 }
Colin Crosscec81712017-07-13 14:43:27 -0700593 }
594
Paul Duffin709e0e32021-03-22 10:09:02 +0000595 return newTestingModule(ctx.config, module)
Colin Crosscec81712017-07-13 14:43:27 -0700596}
597
Jiyong Park37b25202018-07-11 10:49:27 +0900598func (ctx *TestContext) ModuleVariantsForTests(name string) []string {
599 var variants []string
600 ctx.VisitAllModules(func(m blueprint.Module) {
601 if ctx.ModuleName(m) == name {
602 variants = append(variants, ctx.ModuleSubDir(m))
603 }
604 })
605 return variants
606}
607
Colin Cross4c83e5c2019-02-25 14:54:28 -0800608// SingletonForTests returns a TestingSingleton for the singleton registered with the given name.
609func (ctx *TestContext) SingletonForTests(name string) TestingSingleton {
610 allSingletonNames := []string{}
611 for _, s := range ctx.Singletons() {
612 n := ctx.SingletonName(s)
613 if n == name {
614 return TestingSingleton{
Paul Duffin709e0e32021-03-22 10:09:02 +0000615 baseTestingComponent: newBaseTestingComponent(ctx.config, s.(testBuildProvider)),
Paul Duffin31a22882021-03-22 09:29:00 +0000616 singleton: s.(*singletonAdaptor).Singleton,
Colin Cross4c83e5c2019-02-25 14:54:28 -0800617 }
618 }
619 allSingletonNames = append(allSingletonNames, n)
620 }
621
622 panic(fmt.Errorf("failed to find singleton %q."+
623 "\nall singletons: %v", name, allSingletonNames))
624}
625
Martin Stjernholm1ebef5b2022-02-10 23:34:28 +0000626type InstallMakeRule struct {
627 Target string
628 Deps []string
629 OrderOnlyDeps []string
630}
631
632func parseMkRules(t *testing.T, config Config, nodes []mkparser.Node) []InstallMakeRule {
633 var rules []InstallMakeRule
634 for _, node := range nodes {
635 if mkParserRule, ok := node.(*mkparser.Rule); ok {
636 var rule InstallMakeRule
637
638 if targets := mkParserRule.Target.Words(); len(targets) == 0 {
639 t.Fatalf("no targets for rule %s", mkParserRule.Dump())
640 } else if len(targets) > 1 {
641 t.Fatalf("unsupported multiple targets for rule %s", mkParserRule.Dump())
642 } else if !targets[0].Const() {
643 t.Fatalf("unsupported non-const target for rule %s", mkParserRule.Dump())
644 } else {
645 rule.Target = normalizeStringRelativeToTop(config, targets[0].Value(nil))
646 }
647
648 prereqList := &rule.Deps
649 for _, prereq := range mkParserRule.Prerequisites.Words() {
650 if !prereq.Const() {
651 t.Fatalf("unsupported non-const prerequisite for rule %s", mkParserRule.Dump())
652 }
653
654 if prereq.Value(nil) == "|" {
655 prereqList = &rule.OrderOnlyDeps
656 continue
657 }
658
659 *prereqList = append(*prereqList, normalizeStringRelativeToTop(config, prereq.Value(nil)))
660 }
661
662 rules = append(rules, rule)
663 }
664 }
665
666 return rules
667}
668
669func (ctx *TestContext) InstallMakeRulesForTesting(t *testing.T) []InstallMakeRule {
670 installs := ctx.SingletonForTests("makevars").Singleton().(*makeVarsSingleton).installsForTesting
671 buf := bytes.NewBuffer(append([]byte(nil), installs...))
672 parser := mkparser.NewParser("makevars", buf)
673
674 nodes, errs := parser.Parse()
675 if len(errs) > 0 {
676 t.Fatalf("error parsing install rules: %s", errs[0])
677 }
678
679 return parseMkRules(t, ctx.config, nodes)
680}
681
Paul Duffin8eb45732022-10-04 19:03:31 +0100682// MakeVarVariable provides access to make vars that will be written by the makeVarsSingleton
683type MakeVarVariable interface {
684 // Name is the name of the variable.
685 Name() string
686
687 // Value is the value of the variable.
688 Value() string
689}
690
691func (v makeVarsVariable) Name() string {
692 return v.name
693}
694
695func (v makeVarsVariable) Value() string {
696 return v.value
697}
698
699// PrepareForTestAccessingMakeVars sets up the test so that MakeVarsForTesting will work.
700var PrepareForTestAccessingMakeVars = GroupFixturePreparers(
701 PrepareForTestWithAndroidMk,
702 PrepareForTestWithMakevars,
703)
704
705// MakeVarsForTesting returns a filtered list of MakeVarVariable objects that represent the
706// variables that will be written out.
707//
708// It is necessary to use PrepareForTestAccessingMakeVars in tests that want to call this function.
709// Along with any other preparers needed to add the make vars.
710func (ctx *TestContext) MakeVarsForTesting(filter func(variable MakeVarVariable) bool) []MakeVarVariable {
711 vars := ctx.SingletonForTests("makevars").Singleton().(*makeVarsSingleton).varsForTesting
712 result := make([]MakeVarVariable, 0, len(vars))
713 for _, v := range vars {
714 if filter(v) {
715 result = append(result, v)
716 }
717 }
718
719 return result
720}
721
Colin Crossaa255532020-07-03 13:18:24 -0700722func (ctx *TestContext) Config() Config {
723 return ctx.config
724}
725
Colin Cross4c83e5c2019-02-25 14:54:28 -0800726type testBuildProvider interface {
727 BuildParamsForTests() []BuildParams
728 RuleParamsForTests() map[blueprint.Rule]blueprint.RuleParams
729}
730
731type TestingBuildParams struct {
732 BuildParams
733 RuleParams blueprint.RuleParams
Paul Duffin709e0e32021-03-22 10:09:02 +0000734
735 config Config
736}
737
738// RelativeToTop creates a new instance of this which has had any usages of the current test's
739// temporary and test specific build directory replaced with a path relative to the notional top.
740//
741// The parts of this structure which are changed are:
742// * BuildParams
Colin Crossd079e0b2022-08-16 10:27:33 -0700743// - Args
744// - All Path, Paths, WritablePath and WritablePaths fields.
Paul Duffin709e0e32021-03-22 10:09:02 +0000745//
746// * RuleParams
Colin Crossd079e0b2022-08-16 10:27:33 -0700747// - Command
748// - Depfile
749// - Rspfile
750// - RspfileContent
Colin Crossd079e0b2022-08-16 10:27:33 -0700751// - CommandDeps
752// - CommandOrderOnly
Paul Duffin709e0e32021-03-22 10:09:02 +0000753//
754// See PathRelativeToTop for more details.
Paul Duffina71a67a2021-03-29 00:42:57 +0100755//
756// deprecated: this is no longer needed as TestingBuildParams are created in this form.
Paul Duffin709e0e32021-03-22 10:09:02 +0000757func (p TestingBuildParams) RelativeToTop() TestingBuildParams {
758 // If this is not a valid params then just return it back. That will make it easy to use with the
759 // Maybe...() methods.
760 if p.Rule == nil {
761 return p
762 }
763 if p.config.config == nil {
Paul Duffine8366da2021-03-24 10:40:38 +0000764 return p
Paul Duffin709e0e32021-03-22 10:09:02 +0000765 }
766 // Take a copy of the build params and replace any args that contains test specific temporary
767 // paths with paths relative to the top.
768 bparams := p.BuildParams
Paul Duffinbbb0f8f2021-03-24 10:34:52 +0000769 bparams.Depfile = normalizeWritablePathRelativeToTop(bparams.Depfile)
770 bparams.Output = normalizeWritablePathRelativeToTop(bparams.Output)
771 bparams.Outputs = bparams.Outputs.RelativeToTop()
Paul Duffinbbb0f8f2021-03-24 10:34:52 +0000772 bparams.ImplicitOutput = normalizeWritablePathRelativeToTop(bparams.ImplicitOutput)
773 bparams.ImplicitOutputs = bparams.ImplicitOutputs.RelativeToTop()
774 bparams.Input = normalizePathRelativeToTop(bparams.Input)
775 bparams.Inputs = bparams.Inputs.RelativeToTop()
776 bparams.Implicit = normalizePathRelativeToTop(bparams.Implicit)
777 bparams.Implicits = bparams.Implicits.RelativeToTop()
778 bparams.OrderOnly = bparams.OrderOnly.RelativeToTop()
779 bparams.Validation = normalizePathRelativeToTop(bparams.Validation)
780 bparams.Validations = bparams.Validations.RelativeToTop()
Paul Duffin709e0e32021-03-22 10:09:02 +0000781 bparams.Args = normalizeStringMapRelativeToTop(p.config, bparams.Args)
782
783 // Ditto for any fields in the RuleParams.
784 rparams := p.RuleParams
785 rparams.Command = normalizeStringRelativeToTop(p.config, rparams.Command)
786 rparams.Depfile = normalizeStringRelativeToTop(p.config, rparams.Depfile)
787 rparams.Rspfile = normalizeStringRelativeToTop(p.config, rparams.Rspfile)
788 rparams.RspfileContent = normalizeStringRelativeToTop(p.config, rparams.RspfileContent)
Paul Duffin709e0e32021-03-22 10:09:02 +0000789 rparams.CommandDeps = normalizeStringArrayRelativeToTop(p.config, rparams.CommandDeps)
790 rparams.CommandOrderOnly = normalizeStringArrayRelativeToTop(p.config, rparams.CommandOrderOnly)
791
792 return TestingBuildParams{
793 BuildParams: bparams,
794 RuleParams: rparams,
795 }
Colin Cross4c83e5c2019-02-25 14:54:28 -0800796}
797
Paul Duffinbbb0f8f2021-03-24 10:34:52 +0000798func normalizeWritablePathRelativeToTop(path WritablePath) WritablePath {
799 if path == nil {
800 return nil
801 }
802 return path.RelativeToTop().(WritablePath)
803}
804
805func normalizePathRelativeToTop(path Path) Path {
806 if path == nil {
807 return nil
808 }
809 return path.RelativeToTop()
810}
811
Jiakai Zhangcf61e3c2023-05-08 16:28:38 +0000812func allOutputs(p BuildParams) []string {
813 outputs := append(WritablePaths(nil), p.Outputs...)
814 outputs = append(outputs, p.ImplicitOutputs...)
815 if p.Output != nil {
816 outputs = append(outputs, p.Output)
817 }
818 return outputs.Strings()
819}
820
821// AllOutputs returns all 'BuildParams.Output's and 'BuildParams.Outputs's in their full path string forms.
822func (p TestingBuildParams) AllOutputs() []string {
823 return allOutputs(p.BuildParams)
824}
825
Paul Duffin0eda26b92021-03-22 09:34:29 +0000826// baseTestingComponent provides functionality common to both TestingModule and TestingSingleton.
827type baseTestingComponent struct {
Paul Duffin709e0e32021-03-22 10:09:02 +0000828 config Config
Paul Duffin0eda26b92021-03-22 09:34:29 +0000829 provider testBuildProvider
830}
831
Paul Duffin709e0e32021-03-22 10:09:02 +0000832func newBaseTestingComponent(config Config, provider testBuildProvider) baseTestingComponent {
833 return baseTestingComponent{config, provider}
834}
835
836// A function that will normalize a string containing paths, e.g. ninja command, by replacing
837// any references to the test specific temporary build directory that changes with each run to a
838// fixed path relative to a notional top directory.
839//
840// This is similar to StringPathRelativeToTop except that assumes the string is a single path
841// containing at most one instance of the temporary build directory at the start of the path while
842// this assumes that there can be any number at any position.
843func normalizeStringRelativeToTop(config Config, s string) string {
Colin Cross3b1c6842024-07-26 11:52:57 -0700844 // The outDir usually looks something like: /tmp/testFoo2345/001
Paul Duffin709e0e32021-03-22 10:09:02 +0000845 //
Colin Cross3b1c6842024-07-26 11:52:57 -0700846 // Replace any usage of the outDir with out/soong, e.g. replace "/tmp/testFoo2345/001" with
Paul Duffin709e0e32021-03-22 10:09:02 +0000847 // "out/soong".
Lukacs T. Berki9f6c24a2021-08-26 15:07:24 +0200848 outSoongDir := filepath.Clean(config.soongOutDir)
Paul Duffin709e0e32021-03-22 10:09:02 +0000849 re := regexp.MustCompile(`\Q` + outSoongDir + `\E\b`)
850 s = re.ReplaceAllString(s, "out/soong")
851
Colin Cross3b1c6842024-07-26 11:52:57 -0700852 // Replace any usage of the outDir/.. with out, e.g. replace "/tmp/testFoo2345" with
Paul Duffin709e0e32021-03-22 10:09:02 +0000853 // "out". This must come after the previous replacement otherwise this would replace
854 // "/tmp/testFoo2345/001" with "out/001" instead of "out/soong".
855 outDir := filepath.Dir(outSoongDir)
856 re = regexp.MustCompile(`\Q` + outDir + `\E\b`)
857 s = re.ReplaceAllString(s, "out")
858
859 return s
860}
861
862// normalizeStringArrayRelativeToTop creates a new slice constructed by applying
863// normalizeStringRelativeToTop to each item in the slice.
864func normalizeStringArrayRelativeToTop(config Config, slice []string) []string {
865 newSlice := make([]string, len(slice))
866 for i, s := range slice {
867 newSlice[i] = normalizeStringRelativeToTop(config, s)
868 }
869 return newSlice
870}
871
872// normalizeStringMapRelativeToTop creates a new map constructed by applying
873// normalizeStringRelativeToTop to each value in the map.
874func normalizeStringMapRelativeToTop(config Config, m map[string]string) map[string]string {
875 newMap := map[string]string{}
876 for k, v := range m {
877 newMap[k] = normalizeStringRelativeToTop(config, v)
878 }
879 return newMap
Paul Duffin0eda26b92021-03-22 09:34:29 +0000880}
881
882func (b baseTestingComponent) newTestingBuildParams(bparams BuildParams) TestingBuildParams {
Colin Cross4c83e5c2019-02-25 14:54:28 -0800883 return TestingBuildParams{
Paul Duffin709e0e32021-03-22 10:09:02 +0000884 config: b.config,
Colin Cross4c83e5c2019-02-25 14:54:28 -0800885 BuildParams: bparams,
Paul Duffin0eda26b92021-03-22 09:34:29 +0000886 RuleParams: b.provider.RuleParamsForTests()[bparams.Rule],
Paul Duffine8366da2021-03-24 10:40:38 +0000887 }.RelativeToTop()
Colin Cross4c83e5c2019-02-25 14:54:28 -0800888}
889
Paul Duffin0eda26b92021-03-22 09:34:29 +0000890func (b baseTestingComponent) maybeBuildParamsFromRule(rule string) (TestingBuildParams, []string) {
Thiébaud Weksteen3600b802020-08-27 15:50:24 +0200891 var searchedRules []string
Paul Duffin4dbf6cf2021-06-08 10:06:37 +0100892 buildParams := b.provider.BuildParamsForTests()
893 for _, p := range buildParams {
894 ruleAsString := p.Rule.String()
895 searchedRules = append(searchedRules, ruleAsString)
896 if strings.Contains(ruleAsString, rule) {
Paul Duffin0eda26b92021-03-22 09:34:29 +0000897 return b.newTestingBuildParams(p), searchedRules
Colin Cross4c83e5c2019-02-25 14:54:28 -0800898 }
899 }
Thiébaud Weksteen3600b802020-08-27 15:50:24 +0200900 return TestingBuildParams{}, searchedRules
Colin Cross4c83e5c2019-02-25 14:54:28 -0800901}
902
Paul Duffin0eda26b92021-03-22 09:34:29 +0000903func (b baseTestingComponent) buildParamsFromRule(rule string) TestingBuildParams {
904 p, searchRules := b.maybeBuildParamsFromRule(rule)
Colin Cross4c83e5c2019-02-25 14:54:28 -0800905 if p.Rule == nil {
Paul Duffin4dbf6cf2021-06-08 10:06:37 +0100906 panic(fmt.Errorf("couldn't find rule %q.\nall rules:\n%s", rule, strings.Join(searchRules, "\n")))
Colin Cross4c83e5c2019-02-25 14:54:28 -0800907 }
908 return p
909}
910
Martin Stjernholm827ba622022-02-03 00:20:11 +0000911func (b baseTestingComponent) maybeBuildParamsFromDescription(desc string) (TestingBuildParams, []string) {
912 var searchedDescriptions []string
Paul Duffin0eda26b92021-03-22 09:34:29 +0000913 for _, p := range b.provider.BuildParamsForTests() {
Martin Stjernholm827ba622022-02-03 00:20:11 +0000914 searchedDescriptions = append(searchedDescriptions, p.Description)
Colin Crossb88b3c52019-06-10 15:15:17 -0700915 if strings.Contains(p.Description, desc) {
Martin Stjernholm827ba622022-02-03 00:20:11 +0000916 return b.newTestingBuildParams(p), searchedDescriptions
Colin Cross4c83e5c2019-02-25 14:54:28 -0800917 }
918 }
Martin Stjernholm827ba622022-02-03 00:20:11 +0000919 return TestingBuildParams{}, searchedDescriptions
Colin Cross4c83e5c2019-02-25 14:54:28 -0800920}
921
Paul Duffin0eda26b92021-03-22 09:34:29 +0000922func (b baseTestingComponent) buildParamsFromDescription(desc string) TestingBuildParams {
Martin Stjernholm827ba622022-02-03 00:20:11 +0000923 p, searchedDescriptions := b.maybeBuildParamsFromDescription(desc)
Colin Cross4c83e5c2019-02-25 14:54:28 -0800924 if p.Rule == nil {
Martin Stjernholm827ba622022-02-03 00:20:11 +0000925 panic(fmt.Errorf("couldn't find description %q\nall descriptions:\n%s", desc, strings.Join(searchedDescriptions, "\n")))
Colin Cross4c83e5c2019-02-25 14:54:28 -0800926 }
927 return p
928}
929
Paul Duffin0eda26b92021-03-22 09:34:29 +0000930func (b baseTestingComponent) maybeBuildParamsFromOutput(file string) (TestingBuildParams, []string) {
Martin Stjernholma4aaa472021-09-17 02:51:48 +0100931 searchedOutputs := WritablePaths(nil)
Paul Duffin0eda26b92021-03-22 09:34:29 +0000932 for _, p := range b.provider.BuildParamsForTests() {
Colin Cross4c83e5c2019-02-25 14:54:28 -0800933 outputs := append(WritablePaths(nil), p.Outputs...)
Colin Cross1d2cf042019-03-29 15:33:06 -0700934 outputs = append(outputs, p.ImplicitOutputs...)
Colin Cross4c83e5c2019-02-25 14:54:28 -0800935 if p.Output != nil {
936 outputs = append(outputs, p.Output)
937 }
938 for _, f := range outputs {
Paul Duffin4e6e35c2021-03-22 11:34:57 +0000939 if f.String() == file || f.Rel() == file || PathRelativeToTop(f) == file {
Paul Duffin0eda26b92021-03-22 09:34:29 +0000940 return b.newTestingBuildParams(p), nil
Colin Cross4c83e5c2019-02-25 14:54:28 -0800941 }
Martin Stjernholma4aaa472021-09-17 02:51:48 +0100942 searchedOutputs = append(searchedOutputs, f)
Colin Cross4c83e5c2019-02-25 14:54:28 -0800943 }
944 }
Martin Stjernholma4aaa472021-09-17 02:51:48 +0100945
946 formattedOutputs := []string{}
947 for _, f := range searchedOutputs {
948 formattedOutputs = append(formattedOutputs,
949 fmt.Sprintf("%s (rel=%s)", PathRelativeToTop(f), f.Rel()))
950 }
951
952 return TestingBuildParams{}, formattedOutputs
Colin Cross4c83e5c2019-02-25 14:54:28 -0800953}
954
Paul Duffin0eda26b92021-03-22 09:34:29 +0000955func (b baseTestingComponent) buildParamsFromOutput(file string) TestingBuildParams {
956 p, searchedOutputs := b.maybeBuildParamsFromOutput(file)
Colin Cross4c83e5c2019-02-25 14:54:28 -0800957 if p.Rule == nil {
Paul Duffin4e6e35c2021-03-22 11:34:57 +0000958 panic(fmt.Errorf("couldn't find output %q.\nall outputs:\n %s\n",
959 file, strings.Join(searchedOutputs, "\n ")))
Colin Cross4c83e5c2019-02-25 14:54:28 -0800960 }
961 return p
962}
963
Paul Duffin0eda26b92021-03-22 09:34:29 +0000964func (b baseTestingComponent) allOutputs() []string {
Colin Cross4c83e5c2019-02-25 14:54:28 -0800965 var outputFullPaths []string
Paul Duffin0eda26b92021-03-22 09:34:29 +0000966 for _, p := range b.provider.BuildParamsForTests() {
Jiakai Zhangcf61e3c2023-05-08 16:28:38 +0000967 outputFullPaths = append(outputFullPaths, allOutputs(p)...)
Colin Cross4c83e5c2019-02-25 14:54:28 -0800968 }
969 return outputFullPaths
970}
971
Paul Duffin31a22882021-03-22 09:29:00 +0000972// MaybeRule finds a call to ctx.Build with BuildParams.Rule set to a rule with the given name. Returns an empty
973// BuildParams if no rule is found.
974func (b baseTestingComponent) MaybeRule(rule string) TestingBuildParams {
Paul Duffin0eda26b92021-03-22 09:34:29 +0000975 r, _ := b.maybeBuildParamsFromRule(rule)
Paul Duffin31a22882021-03-22 09:29:00 +0000976 return r
977}
978
979// Rule finds a call to ctx.Build with BuildParams.Rule set to a rule with the given name. Panics if no rule is found.
980func (b baseTestingComponent) Rule(rule string) TestingBuildParams {
Paul Duffin0eda26b92021-03-22 09:34:29 +0000981 return b.buildParamsFromRule(rule)
Paul Duffin31a22882021-03-22 09:29:00 +0000982}
983
984// MaybeDescription finds a call to ctx.Build with BuildParams.Description set to a the given string. Returns an empty
985// BuildParams if no rule is found.
986func (b baseTestingComponent) MaybeDescription(desc string) TestingBuildParams {
Martin Stjernholm827ba622022-02-03 00:20:11 +0000987 p, _ := b.maybeBuildParamsFromDescription(desc)
988 return p
Paul Duffin31a22882021-03-22 09:29:00 +0000989}
990
991// Description finds a call to ctx.Build with BuildParams.Description set to a the given string. Panics if no rule is
992// found.
993func (b baseTestingComponent) Description(desc string) TestingBuildParams {
Paul Duffin0eda26b92021-03-22 09:34:29 +0000994 return b.buildParamsFromDescription(desc)
Paul Duffin31a22882021-03-22 09:29:00 +0000995}
996
997// MaybeOutput finds a call to ctx.Build with a BuildParams.Output or BuildParams.Outputs whose String() or Rel()
998// value matches the provided string. Returns an empty BuildParams if no rule is found.
999func (b baseTestingComponent) MaybeOutput(file string) TestingBuildParams {
Paul Duffin0eda26b92021-03-22 09:34:29 +00001000 p, _ := b.maybeBuildParamsFromOutput(file)
Paul Duffin31a22882021-03-22 09:29:00 +00001001 return p
1002}
1003
1004// Output finds a call to ctx.Build with a BuildParams.Output or BuildParams.Outputs whose String() or Rel()
1005// value matches the provided string. Panics if no rule is found.
1006func (b baseTestingComponent) Output(file string) TestingBuildParams {
Paul Duffin0eda26b92021-03-22 09:34:29 +00001007 return b.buildParamsFromOutput(file)
Paul Duffin31a22882021-03-22 09:29:00 +00001008}
1009
1010// AllOutputs returns all 'BuildParams.Output's and 'BuildParams.Outputs's in their full path string forms.
1011func (b baseTestingComponent) AllOutputs() []string {
Paul Duffin0eda26b92021-03-22 09:34:29 +00001012 return b.allOutputs()
Paul Duffin31a22882021-03-22 09:29:00 +00001013}
1014
Colin Crossb77ffc42019-01-05 22:09:19 -08001015// TestingModule is wrapper around an android.Module that provides methods to find information about individual
1016// ctx.Build parameters for verification in tests.
Colin Crosscec81712017-07-13 14:43:27 -07001017type TestingModule struct {
Paul Duffin31a22882021-03-22 09:29:00 +00001018 baseTestingComponent
Colin Crosscec81712017-07-13 14:43:27 -07001019 module Module
1020}
1021
Paul Duffin709e0e32021-03-22 10:09:02 +00001022func newTestingModule(config Config, module Module) TestingModule {
Paul Duffin31a22882021-03-22 09:29:00 +00001023 return TestingModule{
Paul Duffin709e0e32021-03-22 10:09:02 +00001024 newBaseTestingComponent(config, module),
Paul Duffin31a22882021-03-22 09:29:00 +00001025 module,
1026 }
1027}
1028
Colin Crossb77ffc42019-01-05 22:09:19 -08001029// Module returns the Module wrapped by the TestingModule.
Colin Crosscec81712017-07-13 14:43:27 -07001030func (m TestingModule) Module() Module {
1031 return m.module
1032}
1033
Paul Duffin97d8b402021-03-22 16:04:50 +00001034// VariablesForTestsRelativeToTop returns a copy of the Module.VariablesForTests() with every value
1035// having any temporary build dir usages replaced with paths relative to a notional top.
1036func (m TestingModule) VariablesForTestsRelativeToTop() map[string]string {
1037 return normalizeStringMapRelativeToTop(m.config, m.module.VariablesForTests())
1038}
1039
mrziwangd38e63d2024-07-15 13:43:37 -07001040// OutputFiles checks if module base outputFiles property has any output
mrziwange81e77a2024-06-13 17:02:59 -07001041// files can be used to return.
mrziwangd38e63d2024-07-15 13:43:37 -07001042// Exits the test immediately if there is an error and
mrziwange81e77a2024-06-13 17:02:59 -07001043// otherwise returns the result of calling Paths.RelativeToTop
Paul Duffin962783a2021-03-29 00:00:17 +01001044// on the returned Paths.
Yu Liu51c22312024-08-20 23:56:15 +00001045func (m TestingModule) OutputFiles(ctx *TestContext, t *testing.T, tag string) Paths {
1046 outputFiles := OtherModuleProviderOrDefault(ctx.OtherModuleProviderAdaptor(), m.Module(), OutputFilesProvider)
mrziwangabdb2932024-06-18 12:43:41 -07001047 if tag == "" && outputFiles.DefaultOutputFiles != nil {
1048 return outputFiles.DefaultOutputFiles.RelativeToTop()
1049 } else if taggedOutputFiles, hasTag := outputFiles.TaggedOutputFiles[tag]; hasTag {
mrziwangd38e63d2024-07-15 13:43:37 -07001050 return taggedOutputFiles.RelativeToTop()
mrziwange81e77a2024-06-13 17:02:59 -07001051 }
1052
mrziwangd38e63d2024-07-15 13:43:37 -07001053 t.Fatal(fmt.Errorf("No test output file has been set for tag %q", tag))
1054 return nil
Paul Duffin962783a2021-03-29 00:00:17 +01001055}
1056
Colin Cross4c83e5c2019-02-25 14:54:28 -08001057// TestingSingleton is wrapper around an android.Singleton that provides methods to find information about individual
1058// ctx.Build parameters for verification in tests.
1059type TestingSingleton struct {
Paul Duffin31a22882021-03-22 09:29:00 +00001060 baseTestingComponent
Colin Cross4c83e5c2019-02-25 14:54:28 -08001061 singleton Singleton
Colin Cross4c83e5c2019-02-25 14:54:28 -08001062}
1063
1064// Singleton returns the Singleton wrapped by the TestingSingleton.
1065func (s TestingSingleton) Singleton() Singleton {
1066 return s.singleton
1067}
1068
Logan Chien42039712018-03-12 16:29:17 +08001069func FailIfErrored(t *testing.T, errs []error) {
1070 t.Helper()
1071 if len(errs) > 0 {
1072 for _, err := range errs {
1073 t.Error(err)
1074 }
1075 t.FailNow()
1076 }
1077}
Logan Chienee97c3e2018-03-12 16:34:26 +08001078
Paul Duffinea8a3862021-03-04 17:58:33 +00001079// Fail if no errors that matched the regular expression were found.
1080//
1081// Returns true if a matching error was found, false otherwise.
1082func FailIfNoMatchingErrors(t *testing.T, pattern string, errs []error) bool {
Logan Chienee97c3e2018-03-12 16:34:26 +08001083 t.Helper()
1084
1085 matcher, err := regexp.Compile(pattern)
1086 if err != nil {
Paul Duffinea8a3862021-03-04 17:58:33 +00001087 t.Fatalf("failed to compile regular expression %q because %s", pattern, err)
Logan Chienee97c3e2018-03-12 16:34:26 +08001088 }
1089
1090 found := false
1091 for _, err := range errs {
1092 if matcher.FindStringIndex(err.Error()) != nil {
1093 found = true
1094 break
1095 }
1096 }
1097 if !found {
Steven Moreland082e2062022-08-30 01:11:11 +00001098 t.Errorf("could not match the expected error regex %q (checked %d error(s))", pattern, len(errs))
Logan Chienee97c3e2018-03-12 16:34:26 +08001099 for i, err := range errs {
Colin Crossaede88c2020-08-11 12:17:01 -07001100 t.Errorf("errs[%d] = %q", i, err)
Logan Chienee97c3e2018-03-12 16:34:26 +08001101 }
1102 }
Paul Duffinea8a3862021-03-04 17:58:33 +00001103
1104 return found
Logan Chienee97c3e2018-03-12 16:34:26 +08001105}
Jaewoong Jung9aa3ab12019-04-03 15:47:29 -07001106
Paul Duffin91e38192019-08-05 15:07:57 +01001107func CheckErrorsAgainstExpectations(t *testing.T, errs []error, expectedErrorPatterns []string) {
1108 t.Helper()
1109
1110 if expectedErrorPatterns == nil {
1111 FailIfErrored(t, errs)
1112 } else {
1113 for _, expectedError := range expectedErrorPatterns {
1114 FailIfNoMatchingErrors(t, expectedError, errs)
1115 }
1116 if len(errs) > len(expectedErrorPatterns) {
1117 t.Errorf("additional errors found, expected %d, found %d",
1118 len(expectedErrorPatterns), len(errs))
1119 for i, expectedError := range expectedErrorPatterns {
1120 t.Errorf("expectedErrors[%d] = %s", i, expectedError)
1121 }
1122 for i, err := range errs {
1123 t.Errorf("errs[%d] = %s", i, err)
1124 }
Paul Duffinea8a3862021-03-04 17:58:33 +00001125 t.FailNow()
Paul Duffin91e38192019-08-05 15:07:57 +01001126 }
1127 }
Paul Duffin91e38192019-08-05 15:07:57 +01001128}
1129
Jingwen Chencda22c92020-11-23 00:22:30 -05001130func SetKatiEnabledForTests(config Config) {
1131 config.katiEnabled = true
Paul Duffin8c3fec42020-03-04 20:15:08 +00001132}
1133
Dennis Shend4f5d932023-01-31 20:27:21 +00001134func SetTrimmedApexEnabledForTests(config Config) {
1135 config.productVariables.TrimmedApex = new(bool)
1136 *config.productVariables.TrimmedApex = true
1137}
1138
Colin Crossaa255532020-07-03 13:18:24 -07001139func AndroidMkEntriesForTest(t *testing.T, ctx *TestContext, mod blueprint.Module) []AndroidMkEntries {
Liz Kammer6be69062022-11-04 16:06:02 -04001140 t.Helper()
Jaewoong Jung9aa3ab12019-04-03 15:47:29 -07001141 var p AndroidMkEntriesProvider
1142 var ok bool
1143 if p, ok = mod.(AndroidMkEntriesProvider); !ok {
Roland Levillaindfe75b32019-07-23 16:53:32 +01001144 t.Errorf("module does not implement AndroidMkEntriesProvider: " + mod.Name())
Jaewoong Jung9aa3ab12019-04-03 15:47:29 -07001145 }
Jiyong Park0b0e1b92019-12-03 13:24:29 +09001146
1147 entriesList := p.AndroidMkEntries()
LaMont Jonesb5099382024-01-10 23:42:36 +00001148 aconfigUpdateAndroidMkEntries(ctx, mod.(Module), &entriesList)
Jihoon Kanga3a05462024-04-05 00:36:44 +00001149 for i := range entriesList {
Colin Crossaa255532020-07-03 13:18:24 -07001150 entriesList[i].fillInEntries(ctx, mod)
Jiyong Park0b0e1b92019-12-03 13:24:29 +09001151 }
1152 return entriesList
Jaewoong Jung9aa3ab12019-04-03 15:47:29 -07001153}
Jooyung Han12df5fb2019-07-11 16:18:47 +09001154
Colin Crossaa255532020-07-03 13:18:24 -07001155func AndroidMkDataForTest(t *testing.T, ctx *TestContext, mod blueprint.Module) AndroidMkData {
Liz Kammer6be69062022-11-04 16:06:02 -04001156 t.Helper()
Jooyung Han12df5fb2019-07-11 16:18:47 +09001157 var p AndroidMkDataProvider
1158 var ok bool
1159 if p, ok = mod.(AndroidMkDataProvider); !ok {
Sam Delmerico4e115cc2023-01-19 15:36:52 -05001160 t.Fatalf("module does not implement AndroidMkDataProvider: " + mod.Name())
Jooyung Han12df5fb2019-07-11 16:18:47 +09001161 }
1162 data := p.AndroidMk()
Colin Crossaa255532020-07-03 13:18:24 -07001163 data.fillInData(ctx, mod)
LaMont Jonesb5099382024-01-10 23:42:36 +00001164 aconfigUpdateAndroidMkData(ctx, mod.(Module), &data)
Jooyung Han12df5fb2019-07-11 16:18:47 +09001165 return data
1166}
Paul Duffin9b478b02019-12-10 13:41:51 +00001167
1168// Normalize the path for testing.
1169//
1170// If the path is relative to the build directory then return the relative path
1171// to avoid tests having to deal with the dynamically generated build directory.
1172//
1173// Otherwise, return the supplied path as it is almost certainly a source path
1174// that is relative to the root of the source tree.
1175//
1176// The build and source paths should be distinguishable based on their contents.
Paul Duffin567465d2021-03-16 01:21:34 +00001177//
1178// deprecated: use PathRelativeToTop instead as it handles make install paths and differentiates
1179// between output and source properly.
Paul Duffin9b478b02019-12-10 13:41:51 +00001180func NormalizePathForTesting(path Path) string {
Paul Duffin064b70c2020-11-02 17:32:38 +00001181 if path == nil {
1182 return "<nil path>"
1183 }
Paul Duffin9b478b02019-12-10 13:41:51 +00001184 p := path.String()
1185 if w, ok := path.(WritablePath); ok {
Lukacs T. Berki9f6c24a2021-08-26 15:07:24 +02001186 rel, err := filepath.Rel(w.getSoongOutDir(), p)
Paul Duffin9b478b02019-12-10 13:41:51 +00001187 if err != nil {
1188 panic(err)
1189 }
1190 return rel
1191 }
1192 return p
1193}
1194
Paul Duffin567465d2021-03-16 01:21:34 +00001195// NormalizePathsForTesting creates a slice of strings where each string is the result of applying
1196// NormalizePathForTesting to the corresponding Path in the input slice.
1197//
1198// deprecated: use PathsRelativeToTop instead as it handles make install paths and differentiates
1199// between output and source properly.
Paul Duffin9b478b02019-12-10 13:41:51 +00001200func NormalizePathsForTesting(paths Paths) []string {
1201 var result []string
1202 for _, path := range paths {
1203 relative := NormalizePathForTesting(path)
1204 result = append(result, relative)
1205 }
1206 return result
1207}
Paul Duffin567465d2021-03-16 01:21:34 +00001208
1209// PathRelativeToTop returns a string representation of the path relative to a notional top
1210// directory.
1211//
Paul Duffin85d8f0d2021-03-24 10:18:18 +00001212// It return "<nil path>" if the supplied path is nil, otherwise it returns the result of calling
1213// Path.RelativeToTop to obtain a relative Path and then calling Path.String on that to get the
1214// string representation.
Paul Duffin567465d2021-03-16 01:21:34 +00001215func PathRelativeToTop(path Path) string {
1216 if path == nil {
1217 return "<nil path>"
1218 }
Paul Duffin85d8f0d2021-03-24 10:18:18 +00001219 return path.RelativeToTop().String()
Paul Duffin567465d2021-03-16 01:21:34 +00001220}
1221
1222// PathsRelativeToTop creates a slice of strings where each string is the result of applying
1223// PathRelativeToTop to the corresponding Path in the input slice.
1224func PathsRelativeToTop(paths Paths) []string {
1225 var result []string
1226 for _, path := range paths {
1227 relative := PathRelativeToTop(path)
1228 result = append(result, relative)
1229 }
1230 return result
1231}
1232
1233// StringPathRelativeToTop returns a string representation of the path relative to a notional top
1234// directory.
1235//
Paul Duffin85d8f0d2021-03-24 10:18:18 +00001236// See Path.RelativeToTop for more details as to what `relative to top` means.
Paul Duffin567465d2021-03-16 01:21:34 +00001237//
1238// This is provided for processing paths that have already been converted into a string, e.g. paths
1239// in AndroidMkEntries structures. As a result it needs to be supplied the soong output dir against
1240// which it can try and relativize paths. PathRelativeToTop must be used for process Path objects.
1241func StringPathRelativeToTop(soongOutDir string, path string) string {
Paul Duffin85d8f0d2021-03-24 10:18:18 +00001242 ensureTestOnly()
Paul Duffin567465d2021-03-16 01:21:34 +00001243
1244 // A relative path must be a source path so leave it as it is.
1245 if !filepath.IsAbs(path) {
1246 return path
1247 }
1248
1249 // Check to see if the path is relative to the soong out dir.
1250 rel, isRel, err := maybeRelErr(soongOutDir, path)
1251 if err != nil {
1252 panic(err)
1253 }
1254
1255 if isRel {
Colin Cross3b1c6842024-07-26 11:52:57 -07001256 if strings.HasSuffix(soongOutDir, testOutSoongSubDir) {
1257 // The path is in the soong out dir so indicate that in the relative path.
1258 return filepath.Join(TestOutSoongDir, rel)
1259 } else {
1260 // Handle the PathForArbitraryOutput case
1261 return filepath.Join(testOutDir, rel)
1262
1263 }
Paul Duffin567465d2021-03-16 01:21:34 +00001264 }
1265
1266 // Check to see if the path is relative to the top level out dir.
1267 outDir := filepath.Dir(soongOutDir)
1268 rel, isRel, err = maybeRelErr(outDir, path)
1269 if err != nil {
1270 panic(err)
1271 }
1272
1273 if isRel {
1274 // The path is in the out dir so indicate that in the relative path.
1275 return filepath.Join("out", rel)
1276 }
1277
1278 // This should never happen.
1279 panic(fmt.Errorf("internal error: absolute path %s is not relative to the out dir %s", path, outDir))
1280}
1281
1282// StringPathsRelativeToTop creates a slice of strings where each string is the result of applying
1283// StringPathRelativeToTop to the corresponding string path in the input slice.
1284//
1285// This is provided for processing paths that have already been converted into a string, e.g. paths
1286// in AndroidMkEntries structures. As a result it needs to be supplied the soong output dir against
1287// which it can try and relativize paths. PathsRelativeToTop must be used for process Paths objects.
1288func StringPathsRelativeToTop(soongOutDir string, paths []string) []string {
1289 var result []string
1290 for _, path := range paths {
1291 relative := StringPathRelativeToTop(soongOutDir, path)
1292 result = append(result, relative)
1293 }
1294 return result
1295}
Paul Duffinf53555d2021-03-29 00:21:00 +01001296
1297// StringRelativeToTop will normalize a string containing paths, e.g. ninja command, by replacing
1298// any references to the test specific temporary build directory that changes with each run to a
1299// fixed path relative to a notional top directory.
1300//
1301// This is similar to StringPathRelativeToTop except that assumes the string is a single path
1302// containing at most one instance of the temporary build directory at the start of the path while
1303// this assumes that there can be any number at any position.
1304func StringRelativeToTop(config Config, command string) string {
1305 return normalizeStringRelativeToTop(config, command)
1306}
Paul Duffin0aafcbf2021-03-29 00:56:32 +01001307
1308// StringsRelativeToTop will return a new slice such that each item in the new slice is the result
1309// of calling StringRelativeToTop on the corresponding item in the input slice.
1310func StringsRelativeToTop(config Config, command []string) []string {
1311 return normalizeStringArrayRelativeToTop(config, command)
1312}
Yu Liueae7b362023-11-16 17:05:47 -08001313
1314func EnsureListContainsSuffix(t *testing.T, result []string, expected string) {
1315 t.Helper()
1316 if !SuffixInList(result, expected) {
1317 t.Errorf("%q is not found in %v", expected, result)
1318 }
1319}
Cole Fausta963b942024-04-11 17:43:00 -07001320
1321type panickingConfigAndErrorContext struct {
1322 ctx *TestContext
1323}
1324
1325func (ctx *panickingConfigAndErrorContext) OtherModulePropertyErrorf(module Module, property, fmt string, args ...interface{}) {
1326 panic(ctx.ctx.PropertyErrorf(module, property, fmt, args...).Error())
1327}
1328
1329func (ctx *panickingConfigAndErrorContext) Config() Config {
1330 return ctx.ctx.Config()
1331}
1332
Cole Faust4e2bf9f2024-09-11 13:26:20 -07001333func (ctx *panickingConfigAndErrorContext) HasMutatorFinished(mutatorName string) bool {
1334 return ctx.ctx.HasMutatorFinished(mutatorName)
1335}
1336
Cole Faust55b56fe2024-08-23 12:06:11 -07001337func (ctx *panickingConfigAndErrorContext) otherModuleProvider(m blueprint.Module, p blueprint.AnyProviderKey) (any, bool) {
1338 return ctx.ctx.otherModuleProvider(m, p)
1339}
1340
Cole Fauste8a87832024-09-11 11:35:46 -07001341func PanickingConfigAndErrorContext(ctx *TestContext) ConfigurableEvaluatorContext {
Cole Fausta963b942024-04-11 17:43:00 -07001342 return &panickingConfigAndErrorContext{
1343 ctx: ctx,
1344 }
1345}