diff --git a/backend/core/tests/test_auth.py b/backend/core/tests/test_auth.py index fd52b6e917992038cb5f543615db66532f61d176..99bf60d37602d43c8a8d3725ec40c6e3e53315c0 100644 --- a/backend/core/tests/test_auth.py +++ b/backend/core/tests/test_auth.py @@ -5,13 +5,17 @@ from core.models import UserAccount class AuthTests(APITestCase): @classmethod def setUpTestData(cls): - cls.user = UserAccount.objects.create(username='user') - cls.user.set_password('p') + cls.credentials = {'username': 'user', 'password': 'p'} + cls.user = UserAccount.objects.create(username=cls.credentials['username']) + cls.user.set_password(cls.credentials['password']) cls.user.save() cls.client = APIClient() def test_get_token(self): - response = self.client.post('/api-token-auth/', - {'username': 'user', - 'password': 'p'}, format='json') + response = self.client.post('/api-token-auth/', self.credentials) + self.assertContains(response, 'token') + + def test_refresh_token(self): + token = self.client.post('/api-token-auth/', self.credentials).data + response = self.client.post('/api-token-refresh/', token) self.assertContains(response, 'token')