diff options
Diffstat (limited to 'Tools/Scripts/webkitpy/common/net/web_mock.py')
| -rw-r--r-- | Tools/Scripts/webkitpy/common/net/web_mock.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/Tools/Scripts/webkitpy/common/net/web_mock.py b/Tools/Scripts/webkitpy/common/net/web_mock.py index 423573c60..b53cb66ee 100644 --- a/Tools/Scripts/webkitpy/common/net/web_mock.py +++ b/Tools/Scripts/webkitpy/common/net/web_mock.py @@ -30,11 +30,14 @@ import StringIO class MockWeb(object): - def __init__(self): + def __init__(self, urls=None): + self.urls = urls or {} self.urls_fetched = [] def get_binary(self, url, convert_404_to_None=False): self.urls_fetched.append(url) + if url in self.urls: + return self.urls[url] return "MOCK Web result, convert 404 to None=%s" % convert_404_to_None |