1 files changed, 111 insertions, 0 deletions
diff --git a/catapult/telemetry/telemetry/internal/browser/browser_options_unittest.py b/catapult/telemetry/telemetry/internal/browser/browser_options_unittest.py
new file mode 100644
index 00000000..08791a56
--- /dev/null
+++ b/catapult/telemetry/telemetry/internal/browser/browser_options_unittest.py
@@ -0,0 +1,111 @@
+# Copyright 2012 The Chromium 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 optparse
+import os
+import unittest
+
+from telemetry.internal.browser import browser_options
+
+
+class BrowserOptionsTest(unittest.TestCase):
+  def testDefaults(self):
+    options = browser_options.BrowserFinderOptions()
+    parser = options.CreateParser()
+    parser.add_option('-x', action='store', default=3)
+    parser.parse_args(['--browser', 'any'])
+    self.assertEquals(options.x, 3) # pylint: disable=no-member
+
+  def testDefaultsPlusOverride(self):
+    options = browser_options.BrowserFinderOptions()
+    parser = options.CreateParser()
+    parser.add_option('-x', action='store', default=3)
+    parser.parse_args(['--browser', 'any', '-x', 10])
+    self.assertEquals(options.x, 10) # pylint: disable=no-member
+
+  def testDefaultsDontClobberPresetValue(self):
+    options = browser_options.BrowserFinderOptions()
+    setattr(options, 'x', 7)
+    parser = options.CreateParser()
+    parser.add_option('-x', action='store', default=3)
+    parser.parse_args(['--browser', 'any'])
+    self.assertEquals(options.x, 7) # pylint: disable=no-member
+
+  def testCount0(self):
+    options = browser_options.BrowserFinderOptions()
+    parser = options.CreateParser()
+    parser.add_option('-x', action='count', dest='v')
+    parser.parse_args(['--browser', 'any'])
+    self.assertEquals(options.v, None) # pylint: disable=no-member
+
+  def testCount2(self):
+    options = browser_options.BrowserFinderOptions()
+    parser = options.CreateParser()
+    parser.add_option('-x', action='count', dest='v')
+    parser.parse_args(['--browser', 'any', '-xx'])
+    self.assertEquals(options.v, 2) # pylint: disable=no-member
+
+  def testOptparseMutabilityWhenSpecified(self):
+    options = browser_options.BrowserFinderOptions()
+    parser = options.CreateParser()
+    parser.add_option('-x', dest='verbosity', action='store_true')
+    options_ret, _ = parser.parse_args(['--browser', 'any', '-x'])
+    self.assertEquals(options_ret, options)
+    self.assertTrue(options.verbosity)
+
+  def testOptparseMutabilityWhenNotSpecified(self):
+    options = browser_options.BrowserFinderOptions()
+
+    parser = options.CreateParser()
+    parser.add_option('-x', dest='verbosity', action='store_true')
+    options_ret, _ = parser.parse_args(['--browser', 'any'])
+    self.assertEquals(options_ret, options)
+    self.assertFalse(options.verbosity)
+
+  def testProfileDirDefault(self):
+    options = browser_options.BrowserFinderOptions()
+    parser = options.CreateParser()
+    parser.parse_args(['--browser', 'any'])
+    self.assertEquals(options.browser_options.profile_dir, None)
+
+  def testProfileDir(self):
+    options = browser_options.BrowserFinderOptions()
+    parser = options.CreateParser()
+    # Need to use a directory that exists.
+    current_dir = os.path.dirname(__file__)
+    parser.parse_args(['--browser', 'any', '--profile-dir', current_dir])
+    self.assertEquals(options.browser_options.profile_dir, current_dir)
+
+  def testExtraBrowserArgs(self):
+    options = browser_options.BrowserFinderOptions()
+    parser = options.CreateParser()
+    parser.parse_args(['--extra-browser-args=--foo --bar'])
+
+    self.assertEquals(options.browser_options.extra_browser_args,
+                      set(['--foo', '--bar']))
+
+  def testMergeDefaultValues(self):
+    options = browser_options.BrowserFinderOptions()
+    options.already_true = True
+    options.already_false = False
+    options.override_to_true = False
+    options.override_to_false = True
+
+    parser = optparse.OptionParser()
+    parser.add_option('--already_true', action='store_true')
+    parser.add_option('--already_false', action='store_true')
+    parser.add_option('--unset', action='store_true')
+    parser.add_option('--default_true', action='store_true', default=True)
+    parser.add_option('--default_false', action='store_true', default=False)
+    parser.add_option('--override_to_true', action='store_true', default=False)
+    parser.add_option('--override_to_false', action='store_true', default=True)
+
+    options.MergeDefaultValues(parser.get_default_values())
+
+    self.assertTrue(options.already_true)
+    self.assertFalse(options.already_false)
+    self.assertTrue(options.unset is None)
+    self.assertTrue(options.default_true)
+    self.assertFalse(options.default_false)
+    self.assertFalse(options.override_to_true)
+    self.assertTrue(options.override_to_false)