Fix tests

a-maliarov · May 14, 2023 · 151e92c · 151e92c
1 parent e8dd857
commit 151e92c
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/test_main.py b/tests/test_main.py
@@ -64,10 +64,10 @@ def test_content_type_error(self):
         self.assertTrue('is not supported as a Content-Type' in str(context.exception))
 
     def test_fromdriver(self):
-        capabilities = webdriver.ChromeCapabilities()
-        capabilities.add_argument('--headless')
-        capabilities.add_argument('--no-sandbox')
-        driver = webdriver.ChromeDriver(ChromeDriverManager().install(), desired_capabilities = capabilities.desired)
+        options = webdriver.ChromeOptions()
+        options.add_argument('no-sandbox')
+        options.add_argument('headless')
+        driver = webdriver.Chrome(ChromeDriverManager().install(), options=options, keep_alive=False)
 
         solutions = list()
         for i in range(5):