Base test class for chrome tests

client/bin/site_chrome_test.py

Index: client/bin/site_chrome_test.py
diff --git a/client/bin/site_chrome_test.py b/client/bin/site_chrome_test.py
new file mode 100644
index 0000000000000000000000000000000000000000..4c84d698b95e9ced37ea97c02d4a4d6082c79a5e
--- /dev/null
+++ b/client/bin/site_chrome_test.py
@@ -0,0 +1,48 @@
+# Copyright (c) 2010 The Chromium OS Authors. All rights reserved.
+# Use of this source code is governed by a BSD-style license that can be
+# found in the LICENSE file.
+
+import logging, os, shutil, tempfile, utils
+from autotest_lib.client.bin import test
+from autotest_lib.client.common_lib import error, site_ui
+
+class ChromeTestBase(test.test):
+    home_dir = None
+
+    def setup(self):
+        self.job.setup_dep(['chrome_test'])
+        # create a empty srcdir to prevent the error that checks .version file
+        if not os.path.exists(self.srcdir):
+            os.mkdir(self.srcdir)
+
+
+    def run_chrome_test(self, test_to_run, extra_params = ''):
+        dep = 'chrome_test'
+        dep_dir = os.path.join(self.autodir, 'deps', dep)
+        self.job.install_pkg(dep, 'dep', dep_dir)
+
+        chrome_dir = '/opt/google/chrome'
+        cr_source_dir = '%s/test_src' % dep_dir
+        test_binary_dir = '%s/test_src/out/Release' % dep_dir
+        self.home_dir = tempfile.mkdtemp()
+
+        try:
+            setup_cmd = '%s/%s' % (test_binary_dir,
+                                  'setup_test_links.sh')
+            utils.system(setup_cmd)
+
+            cmd = '%s/%s %s' % (test_binary_dir, test_to_run, extra_params)
+            cmd = 'HOME=%s CR_SOURCE_ROOT=%s %s' % (self.home_dir,
+                                                    cr_source_dir,
+                                                    site_ui.xcommand(cmd))
+            logging.info("Running %s" % cmd)
+            utils.system(cmd)
+        except error.CmdError, e:
+            logging.debug(e)
+            raise error.TestFail('%s failed!' % test_to_run)
+
+
+    def cleanup(self):
+        if self.home_dir:
+            shutil.rmtree(self.home_dir, ignore_errors=True)
+        test.test.cleanup(self)