commit 8063b66d25ab315c96330e6cad67211b06adcd09
author Andrew de los Reyes <adlr@chromium.org> Fri Jul 11 14:58:30 2014 -0700
committer chrome-internal-fetch <chrome-internal-fetch@google.com> Thu Jul 17 00:39:03 2014 +0000
tree 2d6a1ebda2ff007e9ca40890f4ad481c80e9e077
parent cab5b11a722e3e6be018937e1034021b14c00f0d

TestCollector: Module to guide baseline test collection

This was based on test_robot.py originally.

BUG=chromium:393347
TEST=Baseline test collection and replay works

Change-Id: I549845577b59b9e9367208a9336dbfe317dcc519
Reviewed-on: https://chromium-review.googlesource.com/207704
Reviewed-by: Dennis Kempin <denniskempin@chromium.org>
Tested-by: Andrew de los Reyes <adlr@chromium.org>
Commit-Queue: Andrew de los Reyes <adlr@chromium.org>

framework/src/main.py

diff --git a/framework/src/main.py b/framework/src/main.py
index e6616e8..0160957 100644
--- a/framework/src/main.py
+++ b/framework/src/main.py
@@ -21,6 +21,7 @@
 from test_case import TestCase
 from test_factory import TestFactory
 from test_robot import RobotTestGenerator
+from test_collector import TestCollector
 from test_runner import ParallelTestRunner as TestRunner
 
 
@@ -285,9 +286,13 @@
   parser.add_option("--robot",
                     dest="robot", default=None,
                     help="Instruct robot to generate test cases")
-  parser.add_option("--overwrite",
-                    dest="overwrite", action="store_true", default=False,
-                    help="(use with --robot) Overwrite existing tests")
+  parser.add_option("--collect_from",
+                    dest="collect_ip", default=None,
+                    help="Interactively collect tests at given device IP");
+  parser.add_option(
+    "--overwrite",
+    dest="overwrite", action="store_true", default=False,
+    help="(use with --robot or --collect_from) Overwrite existing tests")
   parser.add_option("--no-calib",
                     dest="nocalib", action="store_true", default=False,
                     help="(use with --robot) Skip calibration step.")
@@ -348,6 +353,9 @@
     Get(test_name, view)
   elif options.gdb:
     GDB(test_name)
+  elif options.collect_ip:
+    generator = TestCollector(options.collect_ip, os.environ["TESTS_DIR"])
+    generator.GenerateAll(test_name, options.overwrite)
   elif options.robot:
     generator = RobotTestGenerator(options.robot, not options.nocalib,
         options.slow, options.manual_fingertips, os.environ["TESTS_DIR"])