updated security level helper and window handle management

securedrop/tests/functional/functional_test.py

@@ -101,10 +101,9 @@ def set_tbb_securitylevel(self, level):
         time.sleep(1)
         ActionChains(driver).send_keys(Keys.TAB).\
             send_keys(Keys.RETURN).perform()
-        alert = self.wait_for(lambda: driver.switch_to.alert)
-        alert.send_keys(str(level))
+        ActionChains(driver).send_keys(str(level)).\
+            send_keys(Keys.RETURN).perform()
         time.sleep(1)
-        self.wait_for(lambda: alert.accept())
 
     def create_torbrowser_driver(self):
         logging.info("Creating TorBrowserDriver")

securedrop/tests/functional/test_source.py

@@ -69,24 +69,26 @@ def test_duplicate_generate_pages(self):
 
         # Generate a codename in Tab A
         assert len(self.driver.window_handles) == 1
-        tab_a = self.driver.current_window_handle
         self._source_visits_source_homepage()
         self._source_chooses_to_submit_documents()
         codename_a = self.get_codename_generate()
 
         # Generate a different codename in Tab B
-        self.driver.execute_script("window.open()")
+        self.driver.execute_script("window.open('about:blank', '_blank')")
         tab_b = self.driver.window_handles[1]
+        assert len(self.driver.window_handles) == 2
         self.driver.switch_to.window(tab_b)
         assert self.driver.current_window_handle == tab_b
         self._source_visits_source_homepage()
         self._source_chooses_to_submit_documents()
         codename_b = self.get_codename_generate()
 
+        tab_a = self.driver.window_handles[0]
         assert tab_a != tab_b
         assert codename_a != codename_b
 
         # Proceed to submit documents in Tab A
+        assert len(self.driver.window_handles) == 2
         self.driver.switch_to.window(tab_a)
         assert self.driver.current_window_handle == tab_a
         self._source_continues_to_submit_page()
@@ -108,13 +110,12 @@ def test_duplicate_generate_pages(self):
         assert codename_lookup_b == codename_a
         self._source_submits_a_message()
 
-    def test_duplicate_generate_pages_with_refresh(self):
+    def test_refreshed_duplicate_generate_pages(self):
         # Test generation of multiple codenames in different browser tabs, including behavior
         # of refreshing the codemae in each tab. Ref. issue 4458.
 
         # Generate a codename in Tab A
         assert len(self.driver.window_handles) == 1
-        tab_a = self.driver.current_window_handle
         self._source_visits_source_homepage()
         self._source_chooses_to_submit_documents()
         codename_a1 = self.get_codename_generate()
@@ -124,7 +125,8 @@ def test_duplicate_generate_pages_with_refresh(self):
         assert codename_a1 != codename_a2
 
         # Generate a different codename in Tab B
-        self.driver.execute_script("window.open()")
+        self.driver.execute_script("window.open('about:blank', '_blank')")
+        tab_a = self.driver.window_handles[0]
         tab_b = self.driver.window_handles[1]
         self.driver.switch_to.window(tab_b)
         assert self.driver.current_window_handle == tab_b