diff --git a/needle/driver.py b/needle/driver.py index 322a4e0..dc95c1b 100644 --- a/needle/driver.py +++ b/needle/driver.py @@ -65,14 +65,9 @@ def get_computed_property(self, prop): """, self, prop) -class NeedleWebDriver(WebDriver): +class NeedleWebDriverMixin(object): """ - A browser window that Selenium has opened. - - It is a Selenium :py:class:`~selenium.webdriver.remote.webdriver.WebDriver` - object with some extra methods for testing CSS. See - `Selenium's documentation `_ - for more information. + Selenium WebDriver mixin with some extra methods for testing CSS. """ def load_html(self, html): """ @@ -107,3 +102,13 @@ def _get_js_path(self): def create_web_element(self, *args, **kwargs): return NeedleWebElement(self, *args, **kwargs) + +class NeedleWebDriver(NeedleWebDriverMixin, WebDriver): + """ + A browser window that Selenium has opened. + + It is a Selenium :py:class:`~selenium.webdriver.remote.webdriver.WebDriver` + object with some extra methods for testing CSS. See + `Selenium's documentation `_ + for more information. + """