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)
