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_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 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'])
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'])