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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
|
# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
# file, You can obtain one at http://mozilla.org/MPL/2.0/.
import os
from marionette_driver.by import By
from marionette_driver.errors import JavascriptException
from marionette_harness import (
MarionetteTestCase,
skip_if_chrome,
skip_if_mobile,
WindowManagerMixin,
)
class TestImportScriptContent(WindowManagerMixin, MarionetteTestCase):
contexts = set(["chrome", "content"])
script_file = os.path.abspath(
os.path.join(__file__, os.path.pardir, "importscript.js"))
another_script_file = os.path.abspath(
os.path.join(__file__, os.path.pardir, "importanotherscript.js"))
def setUp(self):
super(TestImportScriptContent, self).setUp()
for context in self.contexts:
with self.marionette.using_context(context):
self.marionette.clear_imported_scripts()
self.reset_context()
def tearDown(self):
self.close_all_windows()
super(TestImportScriptContent, self).tearDown()
def reset_context(self):
self.marionette.set_context("content")
@property
def current_context(self):
return self.marionette._send_message("getContext", key="value")
@property
def other_context(self):
return self.contexts.copy().difference([self.current_context]).pop()
def is_defined(self, symbol):
return self.marionette.execute_script(
"return typeof {} != 'undefined'".format(symbol))
def assert_defined(self, symbol, msg=None):
if msg is None:
msg = "Expected symbol {} to be defined".format(symbol)
self.assertTrue(self.is_defined(symbol), msg)
def assert_undefined(self, symbol, msg=None):
if msg is None:
msg = "Expected symbol {} to be undefined".format(symbol)
self.assertFalse(self.is_defined(symbol), msg)
def assert_scripts_cleared(self):
self.marionette.import_script(self.script_file)
self.assert_defined("testFunc")
self.marionette.clear_imported_scripts()
self.assert_undefined("testFunc")
def test_import_script(self):
self.marionette.import_script(self.script_file)
self.assertEqual(
"i'm a test function!", self.marionette.execute_script("return testFunc();"))
self.assertEqual("i'm a test function!", self.marionette.execute_async_script(
"marionetteScriptFinished(testFunc());"))
def test_import_script_twice(self):
self.marionette.import_script(self.script_file)
self.assert_defined("testFunc")
# TODO(ato): Note that the WebDriver command primitives
# does not allow us to check what scripts have been imported.
# I suspect we must to do this through an xpcshell test.
self.marionette.import_script(self.script_file)
self.assert_defined("testFunc")
def test_import_script_and_clear(self):
self.marionette.import_script(self.script_file)
self.assert_defined("testFunc")
self.marionette.clear_imported_scripts()
self.assert_scripts_cleared()
self.assert_undefined("testFunc")
with self.assertRaises(JavascriptException):
self.marionette.execute_script("return testFunc()")
with self.assertRaises(JavascriptException):
self.marionette.execute_async_script(
"marionetteScriptFinished(testFunc())")
def test_clear_scripts_in_other_context(self):
self.marionette.import_script(self.script_file)
self.assert_defined("testFunc")
# clearing other context's script file should not affect ours
with self.marionette.using_context(self.other_context):
self.marionette.clear_imported_scripts()
self.assert_undefined("testFunc")
self.assert_defined("testFunc")
def test_multiple_imports(self):
self.marionette.import_script(self.script_file)
self.marionette.import_script(self.another_script_file)
self.assert_defined("testFunc")
self.assert_defined("testAnotherFunc")
@skip_if_chrome("Needs content scope")
@skip_if_mobile("New windows not supported in Fennec")
def test_imports_apply_globally(self):
self.marionette.navigate(
self.marionette.absolute_url("test_windows.html"))
def open_window_with_link():
self.marionette.find_element(By.LINK_TEXT, "Open new window").click()
new_window = self.open_window(trigger=open_window_with_link)
self.marionette.switch_to_window(new_window)
self.marionette.import_script(self.script_file)
self.marionette.close_chrome_window()
self.marionette.switch_to_window(self.start_window)
self.assert_defined("testFunc")
class TestImportScriptChrome(TestImportScriptContent):
def reset_context(self):
self.marionette.set_context("chrome")
|
