Add more custom parameters for passing into tests

Author: mdmintz

seleniumbase/fixtures/base_case.py

@@ -4509,6 +4509,9 @@ def setUp(self, masterqa_mode=False):
             test_id = self.__get_test_id()
             self.browser = sb_config.browser
             self.data = sb_config.data
+            self.var1 = sb_config.var1
+            self.var2 = sb_config.var2
+            self.var3 = sb_config.var3
             self.slow_mode = sb_config.slow_mode
             self.demo_mode = sb_config.demo_mode
             self.demo_sleep = sb_config.demo_sleep

seleniumbase/plugins/base_plugin.py

@@ -15,6 +15,9 @@ class Base(Plugin):
     This parser plugin includes the following command-line options for Nose:
     --env=ENV  (Set a test environment. Use "self.env" to use this in tests.)
     --data=DATA  (Extra data to pass to tests. Use "self.data" in tests.)
+    --var1=DATA  (Extra data to pass to tests. Use "self.var1" in tests.)
+    --var2=DATA  (Extra data to pass to tests. Use "self.var2" in tests.)
+    --var3=DATA  (Extra data to pass to tests. Use "self.var3" in tests.)
     --settings-file=FILE  (Overrides SeleniumBase settings.py values.)
     --archive-logs  (Archive old log files instead of deleting them.)
     --report  (The option to create a fancy report after tests complete.)
@@ -45,7 +48,22 @@ def options(self, parser, env):
             '--data',
             dest='data',
             default=None,
-            help='Extra data to pass from the command line.')
+            help='Extra data to pass to tests from the command line.')
+        parser.add_option(
+            '--var1',
+            dest='var1',
+            default=None,
+            help='Extra data to pass to tests from the command line.')
+        parser.add_option(
+            '--var2',
+            dest='var2',
+            default=None,
+            help='Extra data to pass to tests from the command line.')
+        parser.add_option(
+            '--var3',
+            dest='var3',
+            default=None,
+            help='Extra data to pass to tests from the command line.')
         parser.add_option(
             '--settings_file', '--settings-file', '--settings',
             action='store',
@@ -123,6 +141,9 @@ def beforeTest(self, test):
         test.test.environment = self.options.environment
         test.test.env = self.options.environment  # Add a shortened version
         test.test.data = self.options.data
+        test.test.var1 = self.options.var1
+        test.test.var2 = self.options.var2
+        test.test.var3 = self.options.var3
         test.test.settings_file = self.options.settings_file
         test.test.log_path = self.options.log_path
         test.test.args = self.options

seleniumbase/plugins/pytest_plugin.py

@@ -18,6 +18,9 @@ def pytest_addoption(parser):
     --settings-file=FILE  (Overrides SeleniumBase settings.py values.)
     --env=ENV  (Set a test environment. Use "self.env" to use this in tests.)
     --data=DATA  (Extra data to pass to tests. Use "self.data" in tests.)
+    --var1=DATA  (Extra data to pass to tests. Use "self.var1" in tests.)
+    --var2=DATA  (Extra data to pass to tests. Use "self.var2" in tests.)
+    --var3=DATA  (Extra data to pass to tests. Use "self.var3" in tests.)
     --user-data-dir=DIR  (Set the Chrome user data directory to use.)
     --server=SERVER  (The server / IP address used by the tests.)
     --port=PORT  (The port that's used by the test server.)
@@ -87,7 +90,19 @@ def pytest_addoption(parser):
     parser.addoption('--data',
                      dest='data',
                      default=None,
-                     help='Extra data to pass from the command line.')
+                     help='Extra data to pass to tests from the command line.')
+    parser.addoption('--var1',
+                     dest='var1',
+                     default=None,
+                     help='Extra data to pass to tests from the command line.')
+    parser.addoption('--var2',
+                     dest='var2',
+                     default=None,
+                     help='Extra data to pass to tests from the command line.')
+    parser.addoption('--var3',
+                     dest='var3',
+                     default=None,
+                     help='Extra data to pass to tests from the command line.')
     parser.addoption('--cap_file', '--cap-file',
                      dest='cap_file',
                      default=None,
@@ -407,6 +422,9 @@ def pytest_configure(config):
     sb_config.is_pytest = True
     sb_config.browser = config.getoption('browser')
     sb_config.data = config.getoption('data')
+    sb_config.var1 = config.getoption('var1')
+    sb_config.var2 = config.getoption('var2')
+    sb_config.var3 = config.getoption('var3')
     sb_config.environment = config.getoption('environment')
     sb_config.with_selenium = config.getoption('with_selenium')
     sb_config.user_agent = config.getoption('user_agent')