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