def test_get_data_set(self, mock_request):
mock_request.__name__ = 'request'
api = AdminAPI('http://admin.api', 'token')
api.get_data_set('group', 'type')
mock_request.assert_called_with(
'GET',
'http://admin.api/data-sets?data-group=group&data-type=type',
headers=match_equality(has_entries({
'Accept': 'application/json',
'Authorization': 'Bearer token'
})),
data=None,
)
def test_get_data_set(self, mock_request):
mock_request.__name__ = 'request'
api = AdminAPI('http://admin.api', 'token')
api.get_data_set('group', 'type')
mock_request.assert_called_with(
method='GET',
url='http://admin.api/data-sets',
headers=match_equality(
has_entries({
'Accept': 'application/json',
'Authorization': 'Bearer token'
})),
data=None,
params={
'data-group': 'group',
'data-type': 'type'
},
)
def test_get_data_set_should_return_None_on_404(self, mock_request):
response = Response()
response.status_code = 404
mock_request.return_value = response
mock_request.__name__ = 'get'
api = AdminAPI('http://admin.api', 'token')
data_set = api.get_data_set('foo', 'bar')
eq_(data_set, None)
def test_get_data_set_should_return_None_on_404(self, mock_request):
response = Response()
response.status_code = 404
mock_request.return_value = response
mock_request.__name__ = 'get'
api = AdminAPI('http://admin.api', 'token')
data_set = api.get_data_set('foo', 'bar')
eq_(data_set, None)
def test_get_data_set_should_return_None_if_no_match(self, mock_request):
response = Response()
response.status_code = 200
response._content = b'[]'
mock_request.return_value = response
mock_request.__name__ = 'get'
api = AdminAPI('http://admin.api', 'token')
data_set = api.get_data_set('foo', 'type')
eq_(data_set, None)
def test_get_data_set_should_only_return_one(self, mock_request):
response = Response()
response.status_code = 200
response._content = b'[{"data-type":"type"}]'
mock_request.return_value = response
mock_request.__name__ = 'get'
api = AdminAPI('http://admin.api', 'token')
data_set = api.get_data_set('foo', 'type')
eq_(data_set['data-type'], 'type')
def test_get_data_set_should_return_None_if_no_match(self, mock_request):
response = Response()
response.status_code = 200
response._content = b'[]'
mock_request.return_value = response
mock_request.__name__ = 'get'
api = AdminAPI('http://admin.api', 'token')
data_set = api.get_data_set('foo', 'type')
eq_(data_set, None)
def test_get_data_set_should_only_return_one(self, mock_request):
response = Response()
response.status_code = 200
response._content = b'[{"data-type":"type"}]'
mock_request.return_value = response
mock_request.__name__ = 'get'
api = AdminAPI('http://admin.api', 'token')
data_set = api.get_data_set('foo', 'type')
eq_(data_set['data-type'], 'type')