setup: Add tests --redhat-url option

crobinso · crobinso · commit ceda72316d5f · 2015-05-22T10:13:13.000-04:00
For testing against an arbitrary redhat bugzilla URL. Useful for doing
internal pre-release testing
diff --git a/setup.py b/setup.py
@@ -30,12 +30,15 @@ class TestCommand(Command):
          "This will also be very slow."),
         ("only=", None,
          "Run only tests whose name contains the passed string"),
+        ("redhat-url=", None,
+         "Redhat bugzilla URL to use for ro/rw_functional tests"),
     ]
 
     def initialize_options(self):
         self.ro_functional = False
         self.rw_functional = False
         self.only = None
+        self.redhat_url = None
 
     def finalize_options(self):
         pass
@@ -76,6 +79,9 @@ def run(self):
             except:
                 print("installHandler hack failed")
 
+        import tests as testsmodule
+        testsmodule.REDHAT_URL = self.redhat_url
+
         tests = unittest.TestLoader().loadTestsFromNames(testfiles)
         if self.only:
             newtests = []
diff --git a/tests/__init__.py b/tests/__init__.py
@@ -31,6 +31,9 @@ def _cleanup_cb():
 atexit.register(_cleanup_cb)
 bugzillascript = _import("bugzillascript", "bin/bugzilla")
 
+# This is overwritten by python setup.py test --redhat-url, and then
+# used in ro/rw tests
+REDHAT_URL = None
 
 
 def diff(orig, new):
diff --git a/tests/ro_functional.py b/tests/ro_functional.py
@@ -156,7 +156,7 @@ class BZFDO(BaseTest):
 
 
 class RHTest(BaseTest):
-    url = "https://bugzilla.redhat.com/xmlrpc.cgi"
+    url = tests.REDHAT_URL or "https://bugzilla.redhat.com/xmlrpc.cgi"
     bzclass = bugzilla.RHBugzilla
 
     test0 = BaseTest._testBZClass
diff --git a/tests/rw_functional.py b/tests/rw_functional.py
@@ -65,8 +65,7 @@ class RHPartnerTest(BaseTest):
     # Despite its name, this instance is simply for bugzilla testing,
     # doesn't send out emails and is blown away occasionally. The front
     # page has some info.
-    url = "https://partner-bugzilla.redhat.com/xmlrpc.cgi"
-    # url = "https://bzweb01-devel.app.eng.rdu.redhat.com/"
+    url = tests.REDHAT_URL or "https://partner-bugzilla.redhat.com/xmlrpc.cgi"
     bzclass = bugzilla.RHBugzilla
 
 
