diff --git a/tests/__init__.py b/tests/__init__.py index 04f77040..2ab4b138 100644 --- a/tests/__init__.py +++ b/tests/__init__.py @@ -11,7 +11,7 @@ # Mock URL access for local testing # -def mock_open_url(url, allow_local=False): +def mock_open_url(url, allow_local=False, timeout=None): """ Open local files instead of URLs. If it's a local file path, leave it alone; otherwise, diff --git a/tests/test_controllers.py b/tests/test_controllers.py index fcbecf05..75228934 100644 --- a/tests/test_controllers.py +++ b/tests/test_controllers.py @@ -164,7 +164,7 @@ def test_empty_url(self): def test_local_file(self): """Make sure we're not leaking local data.""" - response = self.get('/data?url=/etc/passwd', status=500) + response = self.get('/data?url=/etc/passwd', status=403) @patch(URL_MOCK_TARGET, new=URL_MOCK_OBJECT) def test_url(self):