def testUseLiveSitesFlagUnset(self):
run_state = shared_page_state.SharedPageState(DummyTest(),
self.options,
story_module.StorySet())
self.assertTrue(run_state.platform.network_controller.is_open)
self.assertEquals(run_state.platform.network_controller.wpr_mode,
wpr_modes.WPR_REPLAY)
def testBrowserStartupURLSetCorrectly(self):
story_set = story_module.StorySet()
google_page = page.Page('http://www.google.com',
startup_url='http://www.google.com',
page_set=story_set,
name='google')
example_page = page.Page('https://www.example.com',
startup_url='https://www.example.com',
page_set=story_set,
name='example')
gmail_page = page.Page('https://www.gmail.com',
startup_url='https://www.gmail.com',
page_set=story_set,
name='gmail')
for p in (google_page, example_page, gmail_page):
story_set.AddStory(p)
shared_state = shared_page_state.SharedPageState(
DummyTest(), self.options, story_set)
for p in (google_page, example_page, gmail_page):
shared_state.WillRunStory(p)
self.assertEquals(p.startup_url,
self.options.browser_options.startup_url)
def testConstructorCallsSetOptions(self):
test = DummyTest()
run_state = shared_page_state.SharedPageState(
test, self.options, story_module.StorySet(), self.possible_browser)
try:
self.assertEqual(test.options, self.options)
finally:
run_state.TearDownState()
def testWPRRecordEnable(self):
self.options.browser_options.wpr_mode = wpr_modes.WPR_RECORD
run_state = shared_page_state.SharedPageState(
DummyTest(), self.options, story_module.StorySet())
self.assertTrue(run_state.platform.network_controller.is_open)
self.assertEquals(run_state.platform.network_controller.wpr_mode,
wpr_modes.WPR_RECORD)
self.assertFalse(run_state.platform.network_controller.use_live_traffic)
def testUseLiveSitesFlagSet(self):
self.options.use_live_sites = True
run_state = shared_page_state.SharedPageState(
DummyTest(), self.options, story_module.StorySet())
self.assertTrue(run_state.platform.network_controller.is_open)
self.assertEquals(run_state.platform.network_controller.wpr_mode,
wpr_modes.WPR_OFF)
self.assertTrue(run_state.platform.network_controller.use_live_traffic)
def TestUseLiveSitesFlag(self, expected_wpr_mode):
with tempfile.NamedTemporaryFile() as f:
run_state = shared_page_state.SharedPageState(
DummyTest(), self.options, page_set.PageSet())
fake_network_controller = FakeNetworkController()
run_state._PrepareWpr(fake_network_controller, f.name, None)
self.assertEquals(fake_network_controller.wpr_mode, expected_wpr_mode)
self.assertEquals(fake_network_controller.archive_path, f.name)
def testUseLiveSitesFlagUnset(self):
test = DummyTest()
run_state = shared_page_state.SharedPageState(
test, self.options, story_module.StorySet(), self.possible_browser)
try:
self.assertTrue(run_state.platform.network_controller.is_open)
self.assertEquals(run_state.platform.network_controller.wpr_mode,
wpr_modes.WPR_REPLAY)
self.assertFalse(run_state.platform.network_controller.use_live_traffic)
finally:
run_state.TearDownState()
def testBrowserStartupURLAndExtraBrowserArgsSetCorrectly(self):
story_set = story_module.StorySet()
google_page = page.Page('http://www.google.com',
startup_url='http://www.google.com',
page_set=story_set,
name='google',
extra_browser_args=['--test_arg1'])
example_page = page.Page('https://www.example.com',
startup_url='https://www.example.com',
page_set=story_set,
name='example',
extra_browser_args=['--test_arg2'])
gmail_page = page.Page('https://www.gmail.com',
startup_url='https://www.gmail.com',
page_set=story_set,
name='gmail',
extra_browser_args=['--test_arg3'])
for p in (google_page, example_page, gmail_page):
story_set.AddStory(p)
shared_state = shared_page_state.SharedPageState(
DummyTest(), self.options, story_set)
try:
for p in (google_page, example_page, gmail_page):
shared_state.WillRunStory(p)
# Fake possible browser saves the finder_options passed into it.
page_level_options = shared_state._possible_browser.finder_options
self.assertEqual(
p.startup_url,
page_level_options.browser_options.startup_url)
self.assertNotEqual(
p.startup_url,
shared_state._finder_options.browser_options.startup_url,
'Make sure the startup_url isn\'t polluting the shared page state.'
)
for page_level_argument in p.extra_browser_args:
self.assertIn(
page_level_argument,
page_level_options.browser_options.extra_browser_args)
self.assertNotIn(
page_level_argument, shared_state._finder_options.
browser_options.extra_browser_args,
'Make sure the extra browser arguments aren\'t polluting the '
'shared page state')
finally:
shared_state.TearDownState()
def testConstructorCallsSetOptions(self):
test = DummyTest()
shared_page_state.SharedPageState(test, self.options,
story_module.StorySet())
self.assertEqual(test.options, self.options)
