Grasscutters/mitmproxy
3
0
Fork 0
mirror of https://github.com/Grasscutters/mitmproxy.git synced 2024-11-26 10:16:27 +00:00
Code Issues Projects Releases Wiki Activity

Merge pull request #2909 from fristonio/Fix-2867

Browse Source 
Fix test fails in test_xss_scanner when running tox without internet connection.
This commit is contained in:
Aldo Cortesi 2018-02-25 23:00:23 +13:00 committed by GitHub
commit 154309b811
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
test/examples/test_xss_scanner.py
View File

@ -296,6 +296,14 @@ class TestXSSScanner():
assert xss_info == expected_xss_info assert xss_info == expected_xss_info
assert sqli_info is None 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 @pytest.fixture
def logger(self): def logger(self):
class Logger(): class Logger():
@ -309,6 +317,7 @@ class TestXSSScanner():
def test_find_unclaimed_URLs(self, monkeypatch, logger): def test_find_unclaimed_URLs(self, monkeypatch, logger):
logger.args = [] logger.args = []
monkeypatch.setattr("mitmproxy.ctx.log", logger) 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>", xss.find_unclaimed_URLs("<html><script src=\"http://google.com\"></script></html>",
"https://example.com") "https://example.com")
assert logger.args == [] assert logger.args == []