summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorLorenzo Colitti <lorenzo@google.com>2020-05-10 03:56:00 +0000
committerAndroid (Google) Code Review <android-gerrit@google.com>2020-05-10 03:56:00 +0000
commit2369fc3481c7111da241b8fbd33374a7b2ec10e6 (patch)
tree3166e8d91c0e0a933714f51c8078b45ca4ae4ca4
parent031757712c52f0c33bb33fbbfcfd51dea55b5a4e (diff)
parent84106d2eef20feaf8a6183462d8bbec3a93ecf2f (diff)
downloadtests-2369fc3481c7111da241b8fbd33374a7b2ec10e6.tar.gz
Merge "Support running only one test." into rvc-dev
-rwxr-xr-xnet/test/all_tests.py9
1 files changed, 8 insertions, 1 deletions
diff --git a/net/test/all_tests.py b/net/test/all_tests.py
index bbef3ac..38c4eac 100755
--- a/net/test/all_tests.py
+++ b/net/test/all_tests.py
@@ -54,7 +54,14 @@ if __name__ == '__main__':
sys.modules[name].InjectTests()
loader = unittest.defaultTestLoader
- test_suite = loader.loadTestsFromNames(test_modules)
+ if len(sys.argv) > 1:
+ test_suite = loader.loadTestsFromNames(sys.argv[1:])
+ else:
+ test_suite = loader.loadTestsFromNames(test_modules)
+
+ assert test_suite.countTestCases() > 0, (
+ "Inconceivable: no tests found! Command line: %s" % " ".join(sys.argv))
+
runner = unittest.TextTestRunner(verbosity=2)
result = runner.run(test_suite)
sys.exit(not result.wasSuccessful())