summaryrefslogtreecommitdiff
path: root/scripts/test_config_fixer.py
diff options
context:
space:
mode:
author Rahul Sabnis <rahulsabnis@google.com> 2022-04-20 20:29:49 +0000
committer Android (Google) Code Review <android-gerrit@google.com> 2022-04-20 20:29:49 +0000
commit3f23b770020755a70a53d9c543ee3dc01d74a26b (patch)
treed1263745184e8ec99564251f7c78a939ea884d22 /scripts/test_config_fixer.py
parented2a5918b0a40a26ba9ac9a6a3377dfb76c4afde (diff)
parentee1c1998d3a69d008ee5ba6b850155ddc25c89ce (diff)
Merge "Updates test_config_fixer to also work for SuiteApkInstaller" into tm-dev
Diffstat (limited to 'scripts/test_config_fixer.py')
-rw-r--r--scripts/test_config_fixer.py4
1 files changed, 3 insertions, 1 deletions
diff --git a/scripts/test_config_fixer.py b/scripts/test_config_fixer.py
index c150e8cfd..3dbc22ec2 100644
--- a/scripts/test_config_fixer.py
+++ b/scripts/test_config_fixer.py
@@ -28,6 +28,8 @@ from manifest import parse_manifest
from manifest import parse_test_config
from manifest import write_xml
+KNOWN_PREPARERS = ['com.android.tradefed.targetprep.TestAppInstallSetup',
+ 'com.android.tradefed.targetprep.suite.SuiteApkInstaller']
def parse_args():
"""Parse commandline arguments."""
@@ -64,7 +66,7 @@ def overwrite_test_file_name(test_config_doc, test_file_name):
tests = get_children_with_tag(test_config, 'target_preparer')
for test in tests:
- if test.getAttribute('class') == "com.android.tradefed.targetprep.TestAppInstallSetup":
+ if test.getAttribute('class') in KNOWN_PREPARERS:
options = get_children_with_tag(test, 'option')
for option in options:
if option.getAttribute('name') == "test-file-name":