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())