diff options
| author | Aldo Cortesi <aldo@corte.si> | 2018-02-25 23:00:23 +1300 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2018-02-25 23:00:23 +1300 |
| commit | 154309b81113695c43e8cc157fe1ec25bc44823f (patch) | |
| tree | a1029fdbb83a1061586d9eef5b96d50a490b556f | |
| parent | 236d88165499a3927caf193b884ac5e3bff16e67 (diff) | |
| parent | 9fed4fa40aba7320b228d6f362c68caff8c893bd (diff) | |
| download | mitmproxy-154309b81113695c43e8cc157fe1ec25bc44823f.tar.gz mitmproxy-154309b81113695c43e8cc157fe1ec25bc44823f.tar.bz2 mitmproxy-154309b81113695c43e8cc157fe1ec25bc44823f.zip | |
Merge pull request #2909 from fristonio/Fix-2867
Fix test fails in test_xss_scanner when running tox without internet connection.
| -rw-r--r-- | test/examples/test_xss_scanner.py | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/test/examples/test_xss_scanner.py b/test/examples/test_xss_scanner.py index 8cf06a2a..610bdd72 100644 --- a/test/examples/test_xss_scanner.py +++ b/test/examples/test_xss_scanner.py @@ -296,6 +296,14 @@ class TestXSSScanner(): assert xss_info == expected_xss_info assert sqli_info is None + def mocked_socket_gethostbyname(domain): + claimed_domains = ["google.com"] + if domain not in claimed_domains: + from socket import gaierror + raise gaierror("[Errno -2] Name or service not known") + else: + return '216.58.221.46' + @pytest.fixture def logger(self): class Logger(): @@ -309,6 +317,7 @@ class TestXSSScanner(): def test_find_unclaimed_URLs(self, monkeypatch, logger): logger.args = [] monkeypatch.setattr("mitmproxy.ctx.log", logger) + monkeypatch.setattr("socket.gethostbyname", self.mocked_socket_gethostbyname) xss.find_unclaimed_URLs("<html><script src=\"http://google.com\"></script></html>", "https://example.com") assert logger.args == [] |