1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
|
#!/usr/bin/env python
#
# Copyright (C) 2019 The Android Open Source Project
#
# 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.
#
"""Unit tests for test_config_fixer.py."""
import io
import sys
import unittest
from xml.dom import minidom
import test_config_fixer
from manifest import write_xml
sys.dont_write_bytecode = True
class OverwritePackageNameTest(unittest.TestCase):
""" Unit tests for overwrite_package_name function """
manifest = (
'<?xml version="1.0" encoding="utf-8"?>\n'
'<manifest xmlns:android="http://schemas.android.com/apk/res/android"\n'
' package="com.android.foo">\n'
' <application>\n'
' </application>\n'
'</manifest>\n')
test_config = (
'<?xml version="1.0" encoding="utf-8"?>\n'
'<configuration description="Runs some tests.">\n'
' <option name="test-suite-tag" value="apct"/>\n'
' <target_preparer class="com.android.tradefed.targetprep.suite.SuiteApkInstaller">\n'
' <option name="package" value="%s"/>\n'
' </target_preparer>\n'
' <test class="com.android.tradefed.testtype.AndroidJUnitTest">\n'
' <option name="package" value="%s"/>\n'
' <option name="runtime-hint" value="20s"/>\n'
' </test>\n'
' <test class="com.android.tradefed.testtype.AndroidJUnitTest">\n'
' <option name="package" value="%s"/>\n'
' <option name="runtime-hint" value="15s"/>\n'
' </test>\n'
'</configuration>\n')
def test_all(self):
doc = minidom.parseString(self.test_config % ("com.android.foo", "com.android.foo", "com.android.bar"))
manifest = minidom.parseString(self.manifest)
test_config_fixer.overwrite_package_name(doc, manifest, "com.soong.foo")
output = io.StringIO()
test_config_fixer.write_xml(output, doc)
# Only the matching package name in a test node should be updated.
expected = self.test_config % ("com.android.foo", "com.soong.foo", "com.android.bar")
self.assertEqual(expected, output.getvalue())
class OverwriteTestFileNameTest(unittest.TestCase):
""" Unit tests for overwrite_test_file_name function """
test_config_test_app_install_setup = (
'<?xml version="1.0" encoding="utf-8"?>\n'
'<configuration description="Runs some tests.">\n'
' <target_preparer class="com.android.tradefed.targetprep.TestAppInstallSetup">\n'
' <option name="test-file-name" value="%s"/>\n'
' </target_preparer>\n'
' <test class="com.android.tradefed.testtype.AndroidJUnitTest">\n'
' <option name="package" value="com.android.foo"/>\n'
' <option name="runtime-hint" value="20s"/>\n'
' </test>\n'
'</configuration>\n')
test_config_suite_apk_installer = (
'<?xml version="1.0" encoding="utf-8"?>\n'
'<configuration description="Runs some tests.">\n'
' <target_preparer class="com.android.tradefed.targetprep.suite.SuiteApkInstaller">\n'
' <option name="test-file-name" value="%s"/>\n'
' </target_preparer>\n'
' <test class="com.android.tradefed.testtype.AndroidJUnitTest">\n'
' <option name="package" value="com.android.foo"/>\n'
' <option name="runtime-hint" value="20s"/>\n'
' </test>\n'
'</configuration>\n')
def test_testappinstallsetup(self):
doc = minidom.parseString(self.test_config_test_app_install_setup % ("foo.apk"))
test_config_fixer.overwrite_test_file_name(doc, "bar.apk")
output = io.StringIO()
test_config_fixer.write_xml(output, doc)
# Only the matching package name in a test node should be updated.
expected = self.test_config_test_app_install_setup % ("bar.apk")
self.assertEqual(expected, output.getvalue())
def test_suiteapkinstaller(self):
doc = minidom.parseString(self.test_config_suite_apk_installer % ("foo.apk"))
test_config_fixer.overwrite_test_file_name(doc, "bar.apk")
output = io.StringIO()
test_config_fixer.write_xml(output, doc)
# Only the matching package name in a test node should be updated.
expected = self.test_config_suite_apk_installer % ("bar.apk")
self.assertEqual(expected, output.getvalue())
class OverwriteMainlineModulePackageNameTest(unittest.TestCase):
""" Unit tests for overwrite_mainline_module_package_name function """
test_config = (
'<?xml version="1.0" encoding="utf-8"?>\n'
'<configuration description="Runs some tests.">\n'
' <target_preparer class="com.android.tradefed.targetprep.TestAppInstallSetup">\n'
' <option name="test-file-name" value="foo.apk"/>\n'
' </target_preparer>\n'
' <test class="com.android.tradefed.testtype.AndroidJUnitTest">\n'
' <option name="package" value="com.android.foo"/>\n'
' <option name="runtime-hint" value="20s"/>\n'
' </test>\n'
' <object type="module_controller" class="com.android.tradefed.testtype.suite.module.MainlineTestModuleController">\n'
' <option name="enable" value="true"/>\n'
' <option name="mainline-module-package-name" value="%s"/>\n'
' </object>\n'
'</configuration>\n')
def test_testappinstallsetup(self):
doc = minidom.parseString(self.test_config % ("com.android.old.package.name"))
test_config_fixer.overwrite_mainline_module_package_name(doc, "com.android.new.package.name")
output = io.StringIO()
test_config_fixer.write_xml(output, doc)
# Only the mainline module package name should be updated. Format the xml
# with minidom first to avoid mismatches due to trivial reformatting.
expected = io.StringIO()
write_xml(expected, minidom.parseString(self.test_config % ("com.android.new.package.name")))
self.maxDiff = None
self.assertEqual(expected.getvalue(), output.getvalue())
if __name__ == '__main__':
unittest.main(verbosity=2)
|
