Merge pull request #101 from ktp420/zombie_browsers

Cleanup browser and other process if setup fails
2017-05-16 16:53:22 -04:00 · 2017-05-16 16:53:22 -04:00 · e06454faca
parent 78a86c5832 5365b37170
commit e06454faca
1 changed files with 16 additions and 3 deletions
--- a/seleniumbase/plugins/pytest_plugin.py
+++ b/seleniumbase/plugins/pytest_plugin.py
@ -128,8 +128,21 @@ def pytest_runtest_setup():
    pass


-def pytest_runtest_teardown():
-    """ This runs after every test with pytest """
+def pytest_runtest_teardown(item):
+    try:
+        self = item._testcase
+        try:
+            if hasattr(self, 'driver') and self.driver:
+                self.driver.quit()
+        except:
+            pass
+        try:
+            if hasattr(self, 'headless') and self.headless:
+                if self.headless_active:
+                    self.display.stop()
+        except:
+            pass
+    except:
        pass