def setup_basic_post_test(self, user, with_local_site, local_site_name,
post_valid_data):
if post_valid_data:
post_data = self.token_data.copy()
else:
post_data = {}
return (get_api_token_list_url(user, local_site_name),
api_token_item_mimetype, post_data, [local_site_name])
def setup_basic_post_test(self, user, with_local_site, local_site_name,
post_valid_data):
if post_valid_data:
post_data = self.token_data
else:
post_data = {}
return (get_api_token_list_url(user, local_site_name),
api_token_item_mimetype,
post_data,
[local_site_name])
def setup_basic_post_test(self, user, with_local_site, local_site_name,
post_valid_data):
if post_valid_data:
post_data = {
'note': 'This is my new token.',
'policy': '{"perms": "ro"}',
}
else:
post_data = {}
return (get_api_token_list_url(user, local_site_name),
api_token_item_mimetype,
post_data,
[local_site_name])
def setup_basic_get_test(self, user, with_local_site, local_site_name,
populate_items):
if populate_items:
if not with_local_site:
LocalSite.objects.create(name=self.local_site_name)
# Due to running this test with a WebAPIToken, we may
# already have one in the database we need to include.
items = list(user.webapi_tokens.all())
items.append(self.create_webapi_token(
user, note='Token 1', with_local_site=with_local_site))
self.create_webapi_token(user, note='Token 2',
with_local_site=not with_local_site)
else:
items = []
return (get_api_token_list_url(user, local_site_name),
api_token_list_mimetype,
items)
def setup_basic_get_test(self, user, with_local_site, local_site_name,
populate_items):
if populate_items:
if not with_local_site:
LocalSite.objects.create(name=self.local_site_name)
# Due to running this test with a WebAPIToken, we may
# already have one in the database we need to include.
items = list(user.webapi_tokens.all())
items.append(self.create_webapi_token(
user, note='Token 1', with_local_site=with_local_site))
self.create_webapi_token(user, note='Token 2',
with_local_site=not with_local_site)
else:
items = []
return (get_api_token_list_url(user, local_site_name),
api_token_list_mimetype,
items)
def test_post_with_generation_error(self):
"""Testing the POST users/<username>/api-tokens/ API
with Token Generation Failed error"""
def _generate_token(self, user, **kwargs):
orig_generate_token(user, max_attempts=0, **kwargs)
orig_generate_token = WebAPIToken.objects.generate_token
self.spy_on(WebAPIToken.objects.generate_token,
call_fake=_generate_token)
rsp = self.api_post(get_api_token_list_url(self.user),
self.token_data,
expected_status=500)
self.assertEqual(rsp['stat'], 'fail')
self.assertEqual(rsp['err']['code'], TOKEN_GENERATION_FAILED.code)
self.assertEqual(rsp['err']['msg'],
'Could not create a unique API token. '
'Please try again.')
def test_post_with_generation_error(self):
"""Testing the POST users/<username>/api-tokens/ API
with Token Generation Failed error"""
def _generate_token(self, user, **kwargs):
orig_generate_token(user, max_attempts=0, **kwargs)
orig_generate_token = WebAPIToken.objects.generate_token
self.spy_on(WebAPIToken.objects.generate_token,
call_fake=_generate_token)
rsp = self.api_post(get_api_token_list_url(self.user),
self.token_data,
expected_status=500)
self.assertEqual(rsp['stat'], 'fail')
self.assertEqual(rsp['err']['code'], TOKEN_GENERATION_FAILED.code)
self.assertEqual(rsp['err']['msg'],
'Could not create a unique API token. '
'Please try again.')
