test_case.py

import logging

from selenium.firefox.webdriver import WebDriver

from django.conf import settings
from django.test import TestCase

djangoassets_logger = logging.getLogger('assets.cssutils')
djangoassets_logger.setLevel(logging.WARNING)

logging.basicConfig(level=logging.WARNING)

class SeleniumTestCase(TestCase):
    selenium = True
    selenium_take_ss = True
    selenium_driver_attr = 'driver'

    def setUp(self, *args, **kwargs):
        selenium_profile = getattr(settings, 'SELENIUM_FF_PROFILE', None)
        self.driver = WebDriver(profile=selenium_profile)

        root_logger = logging.getLogger('root')
        root_logger.setLevel(logging.WARNING)

    def tearDown(self):
        # Need to disable the onbeforeunload event or stuff is going to break
        # if we had a failed test that now won't close
        self.driver.execute_script('window.onbeforeunload = function(){};')
        # Nose handles the driver.quit()