moved headless and bidi option to a separate setter methods

Author: sandeepsuryaprasad

py/conftest.py

@@ -149,6 +149,8 @@ def __init__(self, driver_class, request):
         self._driver = None
         self._service = None
         self.options = driver_class
+        self.headless = driver_class
+        self.bidi = driver_class
 
     @property
     def supported_drivers(self):
@@ -196,15 +198,27 @@ def driver_path(self):
 
     @property
     def headless(self):
-        if self._request.config.option.headless:
-            return True
-        return False
+        return self._headless
+
+    @headless.setter
+    def headless(self, cls_name):
+        self._headless = self._request.config.option.headless
+        if self._headless:
+            if cls_name.lower() == "chrome" or cls_name.lower() == "edge":
+                self._options.add_argument("--headless=new")
+            if cls_name.lower() == "firefox":
+                self._options.add_argument("-headless")
 
     @property
     def bidi(self):
-        if self._request.config.option.bidi:
-            return True
-        return False
+        return self._bidi
+
+    @bidi.setter
+    def bidi(self, cls_name):
+        self._bidi = self._request.config.option.bidi
+        if self._bidi:
+            self._options.web_socket_url = True
+            self._options.unhandled_prompt_behavior = "ignore"
 
     @property
     def options(self):
@@ -238,16 +252,6 @@ def options(self, cls_name):
                 for arg in self.browser_args.split():
                     self._options.add_argument(arg)
 
-        if self.headless:
-            if cls_name.lower() == "chrome" or cls_name.lower() == "edge":
-                self._options.add_argument("--headless=new")
-            if cls_name.lower() == "firefox":
-                self._options.add_argument("-headless")
-
-        if self.bidi:
-            self._options.web_socket_url = True
-            self._options.unhandled_prompt_behavior = "ignore"
-
     @property
     def service(self):
         executable = self.driver_path