1
0
mirror of https://github.com/netbox-community/netbox.git synced 2024-05-10 07:54:54 +00:00

Add description & expires fields to token test

This commit is contained in:
Jeremy Stretch
2023-09-01 11:43:34 -04:00
parent 7848beedce
commit c38884fa11

View File

@ -141,17 +141,25 @@ class TokenTest(
""" """
Test the provisioning of a new REST API token given a valid username and password. Test the provisioning of a new REST API token given a valid username and password.
""" """
data = { user_credentials = {
'username': 'user1', 'username': 'user1',
'password': 'abc123', 'password': 'abc123',
} }
user = User.objects.create_user(**data) user = User.objects.create_user(**user_credentials)
data = {
**user_credentials,
'description': 'My API token',
'expires': '2099-12-31T23:59:59Z',
}
url = reverse('users-api:token_provision') url = reverse('users-api:token_provision')
response = self.client.post(url, data, format='json', **self.header) response = self.client.post(url, data, format='json', **self.header)
self.assertEqual(response.status_code, 201) self.assertEqual(response.status_code, 201)
self.assertIn('key', response.data) self.assertIn('key', response.data)
self.assertEqual(len(response.data['key']), 40) self.assertEqual(len(response.data['key']), 40)
self.assertEqual(response.data['description'], data['description'])
self.assertEqual(response.data['expires'], data['expires'])
token = Token.objects.get(user=user) token = Token.objects.get(user=user)
self.assertEqual(token.key, response.data['key']) self.assertEqual(token.key, response.data['key'])