def test_basic_auth(self): response = yield from self.get('/basic-auth/treq/treq', auth=('treq', 'treq')) self.assertEqual(response.code, 200) yield from print_response(response) json = yield from aiorequests.json_content(response) self.assertTrue(json['authenticated']) self.assertEqual(json['user'], 'treq')
def test_multipart_post(self): class FileLikeObject(StringIO): def __init__(self, val): StringIO.__init__(self, val) self.name = "david.png" def read(*args, **kwargs): return StringIO.read(*args, **kwargs) response = yield from self.post( '/post', data={"a": "b"}, files={"file1": FileLikeObject("file")}) self.assertEqual(response.code, 200) body = yield from aiorequests.json_content(response) self.assertEqual('b', body['form']['a']) self.assertEqual('file', body['files']['file1']) yield from print_response(response)
def assert_sent_header(self, response, header, expected_value): body = yield from aiorequests.json_content(response) self.assertIn(header, body['headers']) self.assertEqual(body['headers'][header], expected_value)
def assert_data(self, response, expected_data): body = yield from aiorequests.json_content(response) self.assertIn('data', body) self.assertEqual(body['data'], expected_data)
def test_cookie(self): response = yield from self.get('/cookies', cookies={'hello': 'there'}) self.assertEqual(response.code, 200) yield from print_response(response) json = yield from aiorequests.json_content(response) self.assertEqual(json['cookies']['hello'], 'there')
def test_gzip(self): response = yield from self.get('/gzip') self.assertEqual(response.code, 200) yield from print_response(response) json = yield from aiorequests.json_content(response) self.assertTrue(json['gzipped'])