| Index: telemetry/third_party/web-page-replay/httpclient_test.py
|
| diff --git a/telemetry/third_party/web-page-replay/httpclient_test.py b/telemetry/third_party/web-page-replay/httpclient_test.py
|
| index 336284e99889b2fe1d74eb6766105b4babe86db0..30b92095d4607ed124efb79b68e471c03e1c4ded 100644
|
| --- a/telemetry/third_party/web-page-replay/httpclient_test.py
|
| +++ b/telemetry/third_party/web-page-replay/httpclient_test.py
|
| @@ -21,6 +21,7 @@ import dnsproxy
|
| import httparchive
|
| import httpclient
|
| import platformsettings
|
| +import script_injector
|
| import test_utils
|
|
|
|
|
| @@ -211,15 +212,18 @@ class ActualNetworkFetchTest(test_utils.RealNetworkFetchTest):
|
|
|
| class HttpArchiveFetchTest(unittest.TestCase):
|
|
|
| + TEST_REQUEST_TIME = datetime.datetime(2016, 11, 17, 1, 2, 3, 456)
|
| +
|
| def createTestResponse(self):
|
| return httparchive.ArchivedHttpResponse(
|
| 11, 200, 'OK', [('content-type', 'text/html')],
|
| - '<body>test</body>', request_time=datetime.datetime(2016, 11, 17))
|
| + ['<body>test</body>'],
|
| + request_time=HttpArchiveFetchTest.TEST_REQUEST_TIME)
|
|
|
| def checkTestResponse(self, actual_response, archive, request):
|
| self.assertEqual(actual_response, archive[request])
|
| - self.assertEqual('<body>test</body>', actual_response.response_data)
|
| - self.assertEqual(datetime.datetime(2016, 11, 17),
|
| + self.assertEqual(['<body>test</body>'], actual_response.response_data)
|
| + self.assertEqual(HttpArchiveFetchTest.TEST_REQUEST_TIME,
|
| actual_response.request_time)
|
|
|
| @staticmethod
|
| @@ -253,6 +257,26 @@ class ReplayHttpArchiveFetchTest(HttpArchiveFetchTest):
|
| archive, None, self.dummy_injector)
|
| self.checkTestResponse(fetch(request), archive, request)
|
|
|
| + @mock.patch('script_injector.util.resource_string')
|
| + @mock.patch('script_injector.util.resource_exists')
|
| + @mock.patch('script_injector.os.path.exists')
|
| + def testInjectedDate(self, os_path, util_exists, util_resource_string):
|
| + os_path.return_value = False
|
| + util_exists.return_value = True
|
| + util_resource_string.return_value = \
|
| + ["""var time_seed={}""".format(script_injector.TIME_SEED_MARKER)]
|
| + request = httparchive.ArchivedHttpRequest(
|
| + 'GET', 'www.test.com', '/', None, {})
|
| + response = self.createTestResponse()
|
| + archive = httparchive.HttpArchive()
|
| + archive[request] = response
|
| +
|
| + fetch = httpclient.ReplayHttpArchiveFetch(
|
| + archive, None, script_injector.GetScriptInjector("time_script.js"))
|
| + self.assertEqual(
|
| + ['<script>var time_seed=1479344523000</script><body>test</body>'],
|
| + fetch(request).response_data)
|
| +
|
|
|
| if __name__ == '__main__':
|
| unittest.main()
|
|
|
Chromium Code Reviews
Issue 2555943002:
Update web-page-replay to the latest commit (Closed)
