Skip to content
Snippets Groups Projects
Commit 3498f237 authored by Erwan Rouchet's avatar Erwan Rouchet Committed by Bastien Abadie
Browse files

Create token on user creation

parent 5355d6d2
No related branches found
No related tags found
No related merge requests found
default_app_config = 'arkindex.users.apps.UsersConfig'
......@@ -2,4 +2,7 @@ from django.apps import AppConfig
class UsersConfig(AppConfig):
name = 'users'
name = 'arkindex.users'
def ready(self):
from arkindex.users import signals # noqa
from django.conf import settings
from django.db.models.signals import post_save
from django.dispatch import receiver
from rest_framework.authtoken.models import Token
@receiver(post_save, sender=settings.AUTH_USER_MODEL)
def create_auth_token(sender, instance=None, created=False, **kwargs):
if created:
Token.objects.create(user=instance)
......@@ -3,6 +3,7 @@ from django.urls import reverse
from django.core import mail
from django.contrib import auth
from arkindex.users.models import User
from rest_framework.authtoken.models import Token
class TestUsers(TestCase):
......@@ -41,3 +42,9 @@ class TestUsers(TestCase):
self.assertListEqual(response.redirect_chain, [(reverse('password_reset_done'), 302)])
self.assertEqual(len(mail.outbox), 1)
self.assertEqual(mail.outbox[0].to, ['email@address.com'])
def test_auto_token(self):
"""
Check creating a user automatically creates a token
"""
self.assertEqual(Token.objects.filter(user=self.user).count(), 1)
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment