OLD | NEW |
1 # Copyright 2015 The Chromium Authors. All rights reserved. | 1 # Copyright 2015 The Chromium Authors. All rights reserved. |
2 # Use of this source code is governed by a BSD-style license that can be | 2 # Use of this source code is governed by a BSD-style license that can be |
3 # found in the LICENSE file. | 3 # found in the LICENSE file. |
4 | 4 |
5 import os | 5 import os |
6 import sys | 6 import sys |
7 import unittest | 7 import unittest |
8 | 8 |
9 import dag | 9 import dag |
10 import loading_model | 10 import loading_model |
11 import loading_trace | 11 import loading_trace |
12 import request_track | 12 import request_track |
13 import request_dependencies_lens | 13 import request_dependencies_lens |
14 | 14 |
15 | 15 |
16 class SimpleLens(object): | 16 class SimpleLens(object): |
17 def __init__(self, trace): | 17 def __init__(self, trace): |
18 self._trace = trace | 18 self._trace = trace |
19 | 19 |
20 def GetRequestDependencies(self): | 20 def GetRequestDependencies(self): |
21 url_to_rq = {} | 21 url_to_rq = {} |
22 deps = [] | 22 deps = [] |
23 for rq in self._trace.request_track.GetEvents(): | 23 for rq in self._trace.request_track.GetEvents(): |
24 assert rq.url not in url_to_rq | 24 assert rq.url not in url_to_rq |
25 url_to_rq[rq.url] = rq | 25 url_to_rq[rq.url] = rq |
26 for rq in self._trace.request_track.GetEvents(): | 26 for rq in self._trace.request_track.GetEvents(): |
27 if rq.initiator in url_to_rq: | 27 if rq.initiator in url_to_rq: |
28 deps.append((rq, url_to_rq[rq.initiator], '')) | 28 deps.append(( url_to_rq[rq.initiator], rq, '')) |
29 return deps | 29 return deps |
30 | 30 |
31 | 31 |
32 class MockRequestTrack(object): | 32 class MockRequestTrack(object): |
33 def __init__(self, requests): | 33 def __init__(self, requests): |
34 self._requests = requests | 34 self._requests = requests |
35 | 35 |
36 def GetEvents(self): | 36 def GetEvents(self): |
37 return self._requests | 37 return self._requests |
38 | 38 |
(...skipping 148 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
187 | 187 |
188 self.assertTrue(loading_model.ResourceGraph._IsAdUrl( | 188 self.assertTrue(loading_model.ResourceGraph._IsAdUrl( |
189 'http://ums.adtechus.com/mapuser?providerid=1003;' | 189 'http://ums.adtechus.com/mapuser?providerid=1003;' |
190 'userid=RUmecco4z3o====')) | 190 'userid=RUmecco4z3o====')) |
191 self.assertTrue(loading_model.ResourceGraph._IsAdUrl( | 191 self.assertTrue(loading_model.ResourceGraph._IsAdUrl( |
192 'http://pagead2.googlesyndication.com/pagead/js/adsbygoogle.js')) | 192 'http://pagead2.googlesyndication.com/pagead/js/adsbygoogle.js')) |
193 | 193 |
194 | 194 |
195 if __name__ == '__main__': | 195 if __name__ == '__main__': |
196 unittest.main() | 196 unittest.main() |
OLD | NEW |