Better error-handling when stopping virtual displays

Author: mdmintz

seleniumbase/fixtures/base_case.py

@@ -3478,7 +3478,12 @@ def tearDown(self):
                 self.__quit_all_drivers()
             if self.headless:
                 if self.headless_active:
-                    self.display.stop()
+                    try:
+                        self.display.stop()
+                    except AttributeError:
+                        pass
+                    except Exception:
+                        pass
                     self.display = None
             if self.with_db_reporting:
                 if has_exception:

seleniumbase/plugins/selenium_plugin.py

@@ -290,4 +290,9 @@ def afterTest(self, test):
             pass
         if self.options.headless:
             if self.headless_active:
-                self.display.stop()
+                try:
+                    self.display.stop()
+                except AttributeError:
+                    pass
+                except Exception:
+                    pass