Added mypy fixes for functional tests

freedomofpress · Jan 5, 2024 · 9fdb49b · 9fdb49b
1 parent 5bc45e4
commit 9fdb49b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/securedrop/tests/functional/app_navigators/journalist_app_nav.py b/securedrop/tests/functional/app_navigators/journalist_app_nav.py
@@ -119,6 +119,7 @@ def cookie_string_from_selenium_cookies(
             return result
 
         cks = cookie_string_from_selenium_cookies(self.driver.get_cookies())
+        assert isinstance(file_url, str)
         raw_content = self._download_content_at_url(file_url, cks)
 
         decryption_result = utils.decrypt_as_journalist(raw_content)

diff --git a/securedrop/tests/functional/conftest.py b/securedrop/tests/functional/conftest.py
@@ -22,14 +22,14 @@
 
 # Function-scoped so that tests can be run in parallel if needed
 @pytest.fixture()
-def firefox_web_driver() -> WebDriver:
+def firefox_web_driver() -> WebDriver:  # type: ignore
     with get_web_driver(web_driver_type=WebDriverTypeEnum.FIREFOX) as web_driver:
         yield web_driver
 
 
 # Function-scoped so that tests can be run in parallel if needed
 @pytest.fixture()
-def tor_browser_web_driver() -> WebDriver:
+def tor_browser_web_driver() -> WebDriver:  # type: ignore
     with get_web_driver(web_driver_type=WebDriverTypeEnum.TOR_BROWSER) as web_driver:
         yield web_driver