def test_response_processing_list(self):
def add_one(val):
return val + 1
client = CollectionClient(self.http, 'foo',
{'show': {
'process': {
'foo.bar': add_one
}
}}, {})
@all_requests
def mock_response(url, request):
return response(
200, {'data': [{
'foo': {
'bar': 1
}
}, {
'foo': {
'bar': 5
}
}]}, {'content-type': 'application/json'})
with HTTMock(mock_response):
res = client.show(123)
self.assertEqual(res[0]['foo']['bar'], 2)
self.assertEqual(res[1]['foo']['bar'], 6)
def test_parsing(self):
client = CollectionClient(
self.http,
'foo',
{
'show': {
'process': {
'foo': date_parse,
'bar': date_parse
}
}
},
{}
)
client.http.get = make_mock_response_obj({
'data': {
'foo': '1982-09-06',
'bar': '1982-09-06T08:34:00Z'
}
})
res = client.show(123)
self.assertEqual(res['foo'], datetime(1982, 9, 6, 0, 0, 0, 0))
self.assertEqual(res['bar'], datetime(1982, 9, 6, 8, 34, 0, 0, tzinfo=tzutc()))
def test_error_missing_required_arg(self):
client = CollectionClient(self.http, 'foo',
{'list': {
'required': ['bar']
}}, {})
with self.assertRaises(Exception):
client.list()
def test_update(self):
client = CollectionClient(self.http, 'foo',
{'update': {
'optional': ['bar', 'baz']
}}, {})
client.http.put = make_mock_response_obj()
client.update('foo#1', baz='hello')
client.http.put.assert_called_with(path='foo/foo#1',
data={'baz': 'hello'})
def test_sub_collections(self):
client = CollectionClient(self.http, 'foo', {},
{'bar': {
'methods': {
'list': {}
}
}})
client.http.get = make_mock_response_obj()
client.bar('foo#1').list()
client.http.get.assert_called_with(path='foo/foo#1/bar/', data={})
def test_error(self):
error_response = MockHTTPResponse(status_code=401, data={'message': "You aren't allowed to do that!"})
client = CollectionClient(self.http, 'foo', {'show':{}}, {})
client.http.session.get = Mock(return_value=error_response)
with self.assertRaises(Error) as context:
client.show(123)
exc = context.exception
self.assertEqual(exc.status_code, 401)
self.assertEqual(exc.message, "You aren't allowed to do that!")
def test_create(self):
client = CollectionClient(self.http, 'foo',
{'create': {
'required': ['bar', 'baz']
}}, {})
client.http.post = make_mock_response_obj()
client.create(bar=123, baz='hello')
client.http.post.assert_called_with(path='foo/',
data={
'bar': 123,
'baz': 'hello'
})
def test_error(self):
error_response = MockHTTPResponse(
status_code=401,
data={'message': "You aren't allowed to do that!"})
client = CollectionClient(self.http, 'foo', {'show': {}}, {})
client.http.session.get = Mock(return_value=error_response)
with self.assertRaises(Error) as context:
client.show(123)
exc = context.exception
self.assertEqual(exc.status_code, 401)
self.assertEqual(exc.message, "You aren't allowed to do that!")
def test_serialization(self):
client = CollectionClient(
self.http,
'foo',
{
'show': {
'optional': ['date']
}
},
{}
)
client.http.get = make_mock_response_obj()
res = client.show(123, date=datetime(1982, 9, 6, 8, 34, 0, 0, tzinfo=tzutc()))
client.http.get.assert_called_with(path='foo/123', data={'date':'1982-09-06T08:34:00+0000'})
def test_response_processing_ignore_missing(self):
def noop(val):
return val
client = CollectionClient(self.http, 'foo',
{'show': {
'process': {
'foo.bar': noop
}
}}, {})
@all_requests
def mock_response(url, request):
return response(200, {'data': {
'notbar': 666
}}, {'content-type': 'application/json'})
with HTTMock(mock_response):
res = client.show(123)
self.assertEqual(res['notbar'], 666)
def test_show(self):
client = CollectionClient(self.http, 'foo',
{'show': {
'optional': ['bar']
}}, {})
client.http.get = make_mock_response_obj()
client.show('foo#1')
client.http.get.assert_called_with(path='foo/foo#1', data={})
client.show('foo#1', bar=123)
client.http.get.assert_called_with(path='foo/foo#1', data={'bar': 123})
def test_list(self):
client = CollectionClient(self.http, 'foo',
{'list': {
'optional': ['bar']
}}, {})
client.http.get = make_mock_response_obj()
client.list()
client.http.get.assert_called_with(path='foo/', data={})
client.list(bar=123)
client.http.get.assert_called_with(path='foo/', data={'bar': 123})
def test_delete(self):
client = CollectionClient(self.http, 'foo', {'delete': {}}, {})
client.http.delete = make_mock_response_obj()
client.delete('foo#1')
client.http.delete.assert_called_with(path='foo/foo#1', data={})
def test_remove_prefixed_underscores(self):
client = CollectionClient(self.http, 'foo', {'list': {}}, {})
client.http.get = make_mock_response_obj()
client.list(___from='from')
client.http.get.assert_called_with(path='foo/', data={'from': 'from'})
def test_error_calling_missing_method(self):
client = CollectionClient(self.http, 'foo', {}, {})
with self.assertRaises(Exception):
client.list()