def test_runner_reuses_sessions_when_it_can(self):
check = Check(
job_id = self.job.id,
url = 'http://foo.bar.com',
browser_name = 'firefox',
version = '15',
platform = 'ANY',
try_count = 1,
running = True,
)
db.session.add(check)
db.session.commit()
runner_main([self.check, check], 'http://foo.bar.com', self.tempdir)
self.assertEqual(1, self.webdriver.Remote.call_count)
self.assertEqual(2, self.driver.get.call_count)
self.assertEqual(1, self.driver.quit.call_count)
self.assertFalse(check.running)
self.assertFalse(self.check.running)
def test_runner_driver_connect_exception_handled(self):
self.webdriver.Remote.side_effect = ValueError
runner_main([self.check], 'http://foo.bar.com', self.tempdir)
self.assertFalse(self.check.running)
self.assertFalse(self.driver.quit.called)
def test_run_check_saves_screenshot(self):
runner_main([self.check], 'http://foo.bar.com', self.tempdir)
with open(path.join(self.tempdir, self.check.filename)) as f:
self.assertEqual('pic', f.read())
