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