def testHTMLPreloadScannerWhitelisting(self):
reference_url_set = set(
[self.GetResourceUrl('./'),
self.GetResourceUrl('0.png')])
url_set = sandwich_misc.ExtractDiscoverableUrls(
self._TRACE_PATH, sandwich_misc.HTML_PRELOAD_SCANNER_DISCOVERER)
self.assertEquals(reference_url_set, url_set)
def testParserDiscoverableWhitelisting(self):
reference_url_set = set([
self.GetResourceUrl('./'),
self.GetResourceUrl('0.png'),
self.GetResourceUrl('1.png')
])
url_set = sandwich_misc.ExtractDiscoverableUrls(
self._TRACE_PATH, sandwich_misc.PARSER_DISCOVERER)
self.assertEquals(reference_url_set, url_set)
def testFullCacheWhitelisting(self):
reference_url_set = set([
self.GetResourceUrl('./'),
self.GetResourceUrl('0.png'),
self.GetResourceUrl('1.png'),
self.GetResourceUrl('favicon.ico')
])
url_set = sandwich_misc.ExtractDiscoverableUrls(
self._TRACE_PATH, sandwich_misc.FULL_CACHE_DISCOVERER)
self.assertEquals(reference_url_set, url_set)
def SetupBenchmark():
trace_path = os.path.join(self._subresources_for_urls_run_task.path, '0',
sandwich_runner.TRACE_FILENAME)
whitelisted_urls = sandwich_misc.ExtractDiscoverableUrls(
trace_path, subresource_discoverer)
urls_resources = json.load(open(self._subresources_for_urls_task.path))
# TODO(gabadie): Implement support for multiple URLs in this Task.
assert len(urls_resources) == 1
url = urls_resources.keys()[0]
url_resources = urls_resources[url]
common_util.EnsureParentDirectoryExists(SetupBenchmark.path)
with open(SetupBenchmark.path, 'w') as output:
json.dump({
'cache_whitelist': [url for url in whitelisted_urls],
'subresource_discoverer': subresource_discoverer,
'url_resources': url_resources,
}, output)
def testRedirectedMainWhitelisting(self):
reference_url_set = set([self.GetResourceUrl('./')])
url_set = sandwich_misc.ExtractDiscoverableUrls(
self._TRACE_PATH, sandwich_misc.REDIRECTED_MAIN_DISCOVERER)
self.assertEquals(reference_url_set, url_set)
def testNoDiscovererWhitelisting(self):
url_set = sandwich_misc.ExtractDiscoverableUrls(
self._TRACE_PATH, sandwich_misc.EMPTY_CACHE_DISCOVERER)
self.assertEquals(set(), url_set)