Workaround pytest-dev/pytest#3775

abotalov · Aug 8, 2018 · d244ea0 · d244ea0
1 parent 6a23f86
commit d244ea0
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/test/selenium/conftest.py b/test/selenium/conftest.py
@@ -10,7 +10,8 @@
 import urlparse
 
 import pytest
-from pytest_selenium import pytest_selenium
+# Wokaround https://github.com/pytest-dev/pytest/issues/3775
+from pytest_selenium import pytest_selenium as pt_selenium
 from selenium.webdriver.remote.remote_connection import (
     LOGGER as SELENIUM_LOGGER)
 
@@ -78,7 +79,7 @@ def gather_screenshot(item, report, driver, summary, extra):
     if pytest_html is not None:
       extra.append(pytest_html.extras.image(screenshot, 'Screenshot'))
 
-  pytest_selenium._gather_screenshot = gather_screenshot
+  pt_selenium._gather_screenshot = gather_screenshot
 
 
 patch_pytest_selenium_screenshot()