Ticket #2879: doh.diff

django/contrib/selenium/__init__.py

@@ +1 @@
+from django.contrib.selenium.testcases import SeleniumRCTestCase

django/contrib/selenium/testcases.py

@@ +1 @@
+import httplib
+import socket
+
+from django.conf import settings
+from django.test.testcases import LiveServerTestCase
+from django.utils.unittest import skipUnless, SkipTest
+
+try:
+    from selenium import selenium
+except ImportError:
+    selenium_installed = False
+else:
+    selenium_installed = True
+
+_NOT_INSTALLED_MSG = "The 'selenium' package isn't installed"
+_CANT_CONNECT_MSG = ("Can't connect to the Selenium server using address"
+                     " %s and port %s")
+
+
+class SeleniumRCTestCase(LiveServerTestCase):
+    """
+    Does basically the same as TestServerTestCase but also connects to the
+    Selenium RC server. The selenium client is then available with
+    'self.selenium'. The requirements are to have the 'selenium' installed in
+    the python path and to have the Selenium RC server running. If those
+    requirements are not filled then the tests will be skipped.
+    """
+    _selenium_server_running = None
+
+    @property
+    def selenium_server_running(self):
+        """
+        Determine if we can connect to the Selenium RC server.
+        Only evaluated once for performance reasons.
+        """
+        if SeleniumRCTestCase._selenium_server_running is not None:
+            return SeleniumRCTestCase._selenium_server_running
+
+        try:
+            conn = httplib.HTTPConnection(settings.SELENIUM_RC_HOST,
+                                          settings.SELENIUM_RC_PORT)
+            try:
+                conn.request("GET", "/selenium-server/driver/", '', {})
+            finally:
+                conn.close()
+            SeleniumRCTestCase._selenium_server_running = True
+        except socket.error:
+            SeleniumRCTestCase._selenium_server_running = False
+
+        return SeleniumRCTestCase._selenium_server_running
+
+    def setUp(self):
+        if not self.selenium_server_running:
+            message = (_CANT_CONNECT_MSG % (settings.SELENIUM_RC_HOST,
+                                            settings.SELENIUM_RC_PORT))
+            raise SkipTest(message)
+
+        super(SeleniumRCTestCase, self).setUp()
+
+        # Launch the Selenium server
+        self.selenium = selenium(
+            settings.SELENIUM_RC_HOST,
+            int(settings.SELENIUM_RC_PORT),
+            settings.SELENIUM_RC_BROWSER,
+            self.live_test_server_url)
+        self.selenium.start()
+
+    def tearDown(self):
+        super(SeleniumRCTestCase, self).tearDown()
+        self.selenium.stop()
+
+SeleniumRCTestCase = skipUnless(selenium_installed,
+    _NOT_INSTALLED_MSG)(SeleniumRCTestCase)