1// Copyright 2020 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
17func init() {
18	RegisterSingletonType("testsuites", testSuiteFilesFactory)
19}
20
21func testSuiteFilesFactory() Singleton {
22	return &testSuiteFiles{}
23}
24
25type testSuiteFiles struct {
26	robolectric WritablePath
27}
28
29type TestSuiteModule interface {
30	Module
31	TestSuites() []string
32}
33
34func (t *testSuiteFiles) GenerateBuildActions(ctx SingletonContext) {
35	files := make(map[string]map[string]InstallPaths)
36
37	ctx.VisitAllModules(func(m Module) {
38		if tsm, ok := m.(TestSuiteModule); ok {
39			for _, testSuite := range tsm.TestSuites() {
40				if files[testSuite] == nil {
41					files[testSuite] = make(map[string]InstallPaths)
42				}
43				name := ctx.ModuleName(m)
44				files[testSuite][name] = append(files[testSuite][name], tsm.FilesToInstall()...)
45			}
46		}
47	})
48
49	t.robolectric = robolectricTestSuite(ctx, files["robolectric-tests"])
50
51	ctx.Phony("robolectric-tests", t.robolectric)
52}
53
54func (t *testSuiteFiles) MakeVars(ctx MakeVarsContext) {
55	ctx.DistForGoal("robolectric-tests", t.robolectric)
56}
57
58func robolectricTestSuite(ctx SingletonContext, files map[string]InstallPaths) WritablePath {
59	var installedPaths InstallPaths
60	for _, module := range SortedStringKeys(files) {
61		installedPaths = append(installedPaths, files[module]...)
62	}
63	testCasesDir := pathForInstall(ctx, BuildOs, X86, "testcases", false).ToMakePath()
64
65	outputFile := PathForOutput(ctx, "packaging", "robolectric-tests.zip")
66	rule := NewRuleBuilder(pctx, ctx)
67	rule.Command().BuiltTool("soong_zip").
68		FlagWithOutput("-o ", outputFile).
69		FlagWithArg("-P ", "host/testcases").
70		FlagWithArg("-C ", testCasesDir.String()).
71		FlagWithRspFileInputList("-r ", outputFile.ReplaceExtension(ctx, "rsp"), installedPaths.Paths())
72	rule.Build("robolectric_tests_zip", "robolectric-tests.zip")
73
74	return outputFile
75}
76