| # Copyright 2016 The Chromium Authors. All rights reserved. |
| # Use of this source code is governed by a BSD-style license that can be |
| # found in the LICENSE file. |
| |
| import unittest |
| |
| from prefetch_view import PrefetchSimulationView |
| import request_dependencies_lens |
| from request_dependencies_lens_unittest import TestRequests |
| import request_track |
| import test_utils |
| |
| |
| class PrefetchSimulationViewTestCase(unittest.TestCase): |
| def setUp(self): |
| super(PrefetchSimulationViewTestCase, self).setUp() |
| self._SetUp() |
| |
| def testExpandRedirectChains(self): |
| self.assertListEqual( |
| [TestRequests.FIRST_REDIRECT_REQUEST, |
| TestRequests.SECOND_REDIRECT_REQUEST, TestRequests.REDIRECTED_REQUEST], |
| PrefetchSimulationView._ExpandRedirectChains( |
| [TestRequests.FIRST_REDIRECT_REQUEST], self.dependencies_lens)) |
| |
| def testParserDiscoverableRequests(self): |
| first_request = TestRequests.FIRST_REDIRECT_REQUEST |
| discovered_requests = PrefetchSimulationView.ParserDiscoverableRequests( |
| first_request, self.dependencies_lens) |
| self.assertListEqual( |
| [TestRequests.FIRST_REDIRECT_REQUEST, |
| TestRequests.JS_REQUEST, TestRequests.JS_REQUEST_OTHER_FRAME, |
| TestRequests.JS_REQUEST_UNRELATED_FRAME], discovered_requests) |
| |
| def testPreloadedRequests(self): |
| first_request = TestRequests.FIRST_REDIRECT_REQUEST |
| preloaded_requests = PrefetchSimulationView.PreloadedRequests( |
| first_request, self.dependencies_lens, self.trace) |
| self.assertListEqual([first_request], preloaded_requests) |
| self._SetUp( |
| [{'args': {'data': {'url': 'http://bla.com/nyancat.js'}}, |
| 'cat': 'blink.net', 'id': '0xaf9f14fa9dd6c314', 'name': 'Resource', |
| 'ph': 'X', 'ts': 1, 'dur': 120, 'pid': 12, 'tid': 12}, |
| {'args': {'step': 'Preload'}, 'cat': 'blink.net', |
| 'id': '0xaf9f14fa9dd6c314', 'name': 'Resource', 'ph': 'T', |
| 'ts': 12, 'pid': 12, 'tid': 12}]) |
| preloaded_requests = PrefetchSimulationView.PreloadedRequests( |
| first_request, self.dependencies_lens, self.trace) |
| self.assertListEqual([TestRequests.FIRST_REDIRECT_REQUEST, |
| TestRequests.JS_REQUEST, TestRequests.JS_REQUEST_OTHER_FRAME, |
| TestRequests.JS_REQUEST_UNRELATED_FRAME], preloaded_requests) |
| |
| def testCost(self): |
| self.assertEqual(40 + 12, self.prefetch_view.Cost()) |
| |
| def testUpdateNodeCosts(self): |
| self.prefetch_view.UpdateNodeCosts(lambda _: 100) |
| self.assertEqual(500 + 40 + 12, self.prefetch_view.Cost()) |
| |
| def testUpdateNodeCostsPartial(self): |
| self.prefetch_view.UpdateNodeCosts( |
| lambda n: 100 if (n.request.request_id |
| == TestRequests.REDIRECTED_REQUEST.request_id) else 0) |
| self.assertEqual(100 + 40 + 12, self.prefetch_view.Cost()) |
| |
| def testToFromJsonDict(self): |
| self.assertEqual(40 + 12, self.prefetch_view.Cost()) |
| json_dict = self.prefetch_view.ToJsonDict() |
| new_view = PrefetchSimulationView.FromJsonDict(json_dict) |
| self.assertEqual(40 + 12, new_view.Cost()) |
| # Updated Costs. |
| self.prefetch_view.UpdateNodeCosts(lambda _: 100) |
| self.assertEqual(500 + 40 + 12, self.prefetch_view.Cost()) |
| json_dict = self.prefetch_view.ToJsonDict() |
| new_view = PrefetchSimulationView.FromJsonDict(json_dict) |
| self.assertEqual(500 + 40 + 12, new_view.Cost()) |
| |
| def _SetUp(self, added_trace_events=None): |
| trace_events = [ |
| {'ts': 5, 'ph': 'X', 'dur': 10, 'pid': 2, 'tid': 1, 'cat': 'blink.net'}] |
| if added_trace_events is not None: |
| trace_events += added_trace_events |
| self.trace = TestRequests.CreateLoadingTrace(trace_events) |
| self.dependencies_lens = request_dependencies_lens.RequestDependencyLens( |
| self.trace) |
| self.user_satisfied_lens = test_utils.MockUserSatisfiedLens(self.trace) |
| self.user_satisfied_lens._postload_msec = 12 |
| self.prefetch_view = PrefetchSimulationView( |
| self.trace, self.dependencies_lens, self.user_satisfied_lens) |
| for e in self.prefetch_view.graph.graph.Edges(): |
| e.cost = 10 |
| |
| |
| if __name__ == '__main__': |
| unittest.main() |
