OLD | NEW |
1 # Copyright 2016 The Chromium Authors. All rights reserved. | 1 # Copyright 2016 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 unittest | 6 import unittest |
7 import urlparse | 7 import urlparse |
8 | 8 |
9 import sandwich_misc | 9 import sandwich_misc |
10 | 10 |
11 | 11 |
12 LOADING_DIR = os.path.dirname(__file__) | 12 LOADING_DIR = os.path.dirname(__file__) |
13 TEST_DATA_DIR = os.path.join(LOADING_DIR, 'testdata') | 13 TEST_DATA_DIR = os.path.join(LOADING_DIR, 'testdata') |
14 | 14 |
15 | 15 |
16 class SandwichMiscTest(unittest.TestCase): | 16 class SandwichMiscTest(unittest.TestCase): |
17 _TRACE_PATH = os.path.join(TEST_DATA_DIR, 'scanner_vs_parser.trace') | 17 _TRACE_PATH = os.path.join(TEST_DATA_DIR, 'scanner_vs_parser.trace') |
18 | 18 |
19 def GetResourceUrl(self, path): | 19 def GetResourceUrl(self, path): |
20 return urlparse.urljoin('http://l/', path) | 20 return urlparse.urljoin('http://l/', path) |
21 | 21 |
| 22 def testNoDiscovererWhitelisting(self): |
| 23 url_set = sandwich_misc.ExtractDiscoverableUrls( |
| 24 self._TRACE_PATH, sandwich_misc.EMPTY_CACHE_DISCOVERER) |
| 25 self.assertEquals(set(), url_set) |
| 26 |
| 27 def testFullCacheWhitelisting(self): |
| 28 reference_url_set = set([self.GetResourceUrl('./'), |
| 29 self.GetResourceUrl('0.png'), |
| 30 self.GetResourceUrl('1.png'), |
| 31 self.GetResourceUrl('favicon.ico')]) |
| 32 url_set = sandwich_misc.ExtractDiscoverableUrls( |
| 33 self._TRACE_PATH, sandwich_misc.FULL_CACHE_DISCOVERER) |
| 34 self.assertEquals(reference_url_set, url_set) |
| 35 |
22 def testRedirectedMainWhitelisting(self): | 36 def testRedirectedMainWhitelisting(self): |
23 urls_set_ref = set([self.GetResourceUrl('./')]) | 37 reference_url_set = set([self.GetResourceUrl('./')]) |
24 urls_set = sandwich_misc.ExtractDiscoverableUrls( | 38 url_set = sandwich_misc.ExtractDiscoverableUrls( |
25 self._TRACE_PATH, sandwich_misc.REDIRECTED_MAIN_DISCOVERER) | 39 self._TRACE_PATH, sandwich_misc.REDIRECTED_MAIN_DISCOVERER) |
26 self.assertEquals(urls_set_ref, urls_set) | 40 self.assertEquals(reference_url_set, url_set) |
27 | 41 |
28 def testParserDiscoverableWhitelisting(self): | 42 def testParserDiscoverableWhitelisting(self): |
29 urls_set_ref = set([self.GetResourceUrl('./'), | 43 reference_url_set = set([self.GetResourceUrl('./'), |
30 self.GetResourceUrl('0.png'), | 44 self.GetResourceUrl('0.png'), |
31 self.GetResourceUrl('1.png')]) | 45 self.GetResourceUrl('1.png')]) |
32 urls_set = sandwich_misc.ExtractDiscoverableUrls( | 46 url_set = sandwich_misc.ExtractDiscoverableUrls( |
33 self._TRACE_PATH, sandwich_misc.PARSER_DISCOVERER) | 47 self._TRACE_PATH, sandwich_misc.PARSER_DISCOVERER) |
34 self.assertEquals(urls_set_ref, urls_set) | 48 self.assertEquals(reference_url_set, url_set) |
35 | 49 |
36 def testHTMLPreloadScannerWhitelisting(self): | 50 def testHTMLPreloadScannerWhitelisting(self): |
37 urls_set_ref = set([self.GetResourceUrl('./'), | 51 reference_url_set = set([self.GetResourceUrl('./'), |
38 self.GetResourceUrl('0.png')]) | 52 self.GetResourceUrl('0.png')]) |
39 urls_set = sandwich_misc.ExtractDiscoverableUrls( | 53 url_set = sandwich_misc.ExtractDiscoverableUrls( |
40 self._TRACE_PATH, sandwich_misc.HTML_PRELOAD_SCANNER_DISCOVERER) | 54 self._TRACE_PATH, sandwich_misc.HTML_PRELOAD_SCANNER_DISCOVERER) |
41 self.assertEquals(urls_set_ref, urls_set) | 55 self.assertEquals(reference_url_set, url_set) |
42 | 56 |
43 | 57 |
44 if __name__ == '__main__': | 58 if __name__ == '__main__': |
45 unittest.main() | 59 unittest.main() |
OLD | NEW |