blob: 00da1f5de2ed938dff02f2ab0e881e7e815c0798 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
|
#!/usr/bin/env python
# -*- coding: UTF-8 -*-
import unittest
import pbtest
class ServiceWorkersTest(pbtest.PBSeleniumTest):
"""Verifies interaction with sites that use Service Worker caches"""
def get_tab_data_domains(self):
domains = self.js(
"let tabData = chrome.extension.getBackgroundPage().badger.tabData;"
"return (Object.keys(tabData).map(tab_id => {"
" return tabData[tab_id].frames[0].host;"
"}));"
)
return domains
def test_returning_to_sw_cached_page(self):
FIXTURE_URL = (
"https://efforg.github.io/privacybadger-test-fixtures/html/"
"service_workers.html"
)
# visit the Service Worker page to activate the worker
self.load_url(FIXTURE_URL)
# Service Workers are off by default in Firefox 60 ESR
if not self.js("return 'serviceWorker' in navigator;"):
self.skipTest("Service Workers are disabled")
# wait for the worker to initialize its cache
self.wait_for_script("return window.WORKER_READY;")
# visit a different site (doesn't matter what it is,
# just needs to be an http site with a different domain)
self.load_url("https://dnt-test.trackersimulator.org/")
# return to the SW page
self.driver.back()
# now open a new window (to avoid clearing badger.tabData)
# and verify results
self.open_window()
self.load_url(self.options_url)
domains = self.get_tab_data_domains()
self.assertIn("efforg.github.io", domains,
"SW page URL was not correctly attributed")
self.assertEqual(len(domains), 1,
"tabData contains an unexpected number of entries")
if __name__ == "__main__":
unittest.main()
|
