def setUp(self): self.sprint = Sprint("user", "public_key", "localhost", 9295, False)
class SprintTestCase(unittest.TestCase): def setUp(self): self.sprint = Sprint("user", "public_key", "localhost", 9295, False) def test_successful(self): resp = {"/login/api":'{"ok":true, "api_key":"private-key"}', "/api/1/curl/execute":'{"ok":true, "status":"queued", \ "region":"california", "job_id":"a123"}', "/api/1/jobs/a123/status": '{"_id":"a123","ok":true,\ "status":"completed","result":{"region":"california","duration":10,\ "steps":[{"connect":1,"request":{"line":"GET / HTTP/1.1",\ "method":"GET","url":"http://localhost:9295","headers":{},\ "content":"MTIzNA=="},"duration":10,\ "response":{"line":"GET / HTTP/1.1","message":"message",\ "status":200,"headers":{},"content":"MTIzNA=="}}]}}'} def callback(result): self.assertIsNotNone(result) self.assertEqual('california', result.region) self.assertEqual(10, result.duration) self.assertEqual(10, result.steps[0].duration) self.assertIsNotNone(result.steps[0].request) self.assertIsNotNone(result.steps[0].response) self.assertEqual('GET', result.steps[0].request.method) self.assertEqual(200, result.steps[0].response.status) self.assertEqual('1234', result.steps[0].request.content) options = {'steps':[{'url': 'http://example.com'}]} self.sprint.client.connection = mock.connection(resp) self.sprint.execute(options, callback) def test_fail_login(self): resp = {"/login/api":'{"error":"login", "reason":"test"}'} def callback(result): self.assertFalse(True) options = {'steps':[{'url': 'http://example.com'}]} self.sprint.client.connection = mock.connection(resp) with self.assertRaises(Error) as err: self.sprint.execute(options, callback) self.assertEqual('login', err.exception.error) def test_fail_validation_steps(self): resp = {} def callback(result): self.assertFalse(True) options = {} self.sprint.client.connection = mock.connection(resp) with self.assertRaises(ValidationError) as err: self.sprint.execute(options, callback) self.assertEqual('validation', err.exception.error) self.assertIn('steps', err.exception.fields) def test_fail_validation_url(self): resp = {} def callback(result): self.assertFalse(True) options = {'steps':[{}]} self.sprint.client.connection = mock.connection(resp) with self.assertRaises(ValidationError) as err: self.sprint.execute(options, callback) self.assertEqual('validation', err.exception.error) self.assertIn('url', err.exception.fields) def test_fail_validation_cookie(self): resp = {} def callback(result): self.assertFalse(True) options = {'steps':[{'cookies':'string'}]} self.sprint.client.connection = mock.connection(resp) with self.assertRaises(ValidationError) as err: self.sprint.execute(options, callback) self.assertEqual('validation', err.exception.error) self.assertIn('url', err.exception.fields) self.assertIn('cookies', err.exception.fields) def test_fail_queue(self): resp = {"/login/api":'{"ok":true, "api_key":"private-key"}', "/api/1/curl/execute":'{"error":"throttle", \ "reason":"Slow down please!"}'} def callback(result): self.assertIsNotNone(result) self.assertTrue('error' in result) self.assertEqual('throttle', result['error']) options = {'steps':[{'url': 'http://example.com'}]} self.sprint.client.connection = mock.connection(resp) with self.assertRaises(Error) as err: self.sprint.execute(options, callback)