suite.py: make main() argv as an argument

Instead of using sys.argv implicitly, which is inconvenient for testing.

Signed-off-by: Loic Dachary <loic@dachary.org>

diff --git a/scripts/suite.py b/scripts/suite.py
index 013650c1aacca9145238161e7ec617a0abec4ac7..1984e57c023f680728836d53582ef3467a4ed296 100644 (file)
--- a/scripts/suite.py
+++ b/scripts/suite.py
@@ -1,4 +1,5 @@
 import docopt
+import sys
 
 import teuthology.suite
 from teuthology.config import config
@@ -81,6 +82,6 @@ Scheduler arguments:
            default_results_timeout=config.results_timeout)
 
 
-def main():
-    args = docopt.docopt(doc)
+def main(argv=sys.argv[1:]):
+    args = docopt.docopt(doc, argv=argv)
     teuthology.suite.main(args)