complete unit tests + hide tokens in admin

Author: merwok

demo/demo/accounts/admin.py

@@ -18,7 +18,7 @@ class EmailConfirmationAdmin(admin.ModelAdmin):
 
 
 class APITokenAdmin(admin.ModelAdmin):
-    list_display = ('key', 'user', 'created')
+    list_display = ('__str__', 'created')
     ordering = ('user', '-created')
     date_hierarchy = 'created'
     readonly_fields = ('key', 'created')

rest_auth_toolkit/models.py

@@ -103,7 +103,7 @@ class Meta:
         abstract = True
 
     def __str__(self):
-        return self.key
+        return 'API token for {user}'.format(user=self.user)
 
     def save(self, *args, **kwargs):
         if not self.key:

tests/unit/conftest.py

@@ -1,10 +1,27 @@
 import pytest
 
-from demo.accounts.models import User
+from demo.accounts.models import User, APIToken
 
 
 @pytest.fixture
 def user0(db):
-    return User.objects.create(
+    return User.objects.create_user(
         email='[email protected]',
+        is_active=True,
+    )
+
+
+@pytest.fixture
+def user1(db):
+    return User.objects.create_user(
+        email='[email protected]',
+        password='unitpass123',
+        is_active=True,
+    )
+
+
+@pytest.fixture
+def token0(user0):
+    return APIToken.objects.create_token(
+        user=user0,
     )

tests/unit/test_models.py

@@ -1,10 +1,99 @@
-def test_baseemailuser_str(user0):
+import pytest
+
+from demo.accounts.models import User, APIToken
+
+
+def test_user_str(user0):
     assert str(user0) == '[email protected]'
 
 
-def test_baseemailuser_get_short_name(user0):
+def test_user_get_short_name(user0):
     assert user0.get_short_name() == '[email protected]'
 
 
-def test_baseemailuser_natural_key(user0):
+def test_user_natural_key(user0):
     assert user0.natural_key() == '[email protected]'
+
+
+def test_user_manager_get_by_natural_key(user0, user1):
+    user = User.objects.get_by_natural_key('[email protected]')
+
+    assert user == user1
+
+
+def test_user_manager_create_user(db):
+    user = User.objects.create_user('[email protected]', 'superpassword')
+
+    assert user.id >= 1
+    assert user.email == '[email protected]'
+    assert user.has_usable_password()
+    assert user.is_active
+    assert not user.is_staff
+    assert not user.is_superuser
+
+
+def test_user_manager_create_superuser(db):
+    user = User.objects.create_superuser('[email protected]', 'wowpassword')
+
+    assert user.id >= 1
+    assert user.email == '[email protected]'
+    assert user.has_usable_password()
+    assert user.is_active
+    assert user.is_staff
+    assert user.is_superuser
+
+
+@pytest.mark.parametrize('fields', [
+    {'is_staff': False},
+    {'is_superuser': False},
+    {'is_staff': False, 'is_superuser': False},
+])
+def test_user_manager_create_superuser_bad_fields(db, fields):
+    with pytest.raises(ValueError):
+        User.objects.create_superuser('[email protected]', 'wowpassword', **fields)
+
+
+def test_apitoken_str(token0):
+    assert str(token0) == 'API token for [email protected]'
+
+
+def test_apitoken_save_create_generates_key(user0):
+    token = APIToken(user=user0)
+    assert token.key == ''
+
+    token.save()
+
+    assert isinstance(token.key, str)
+    assert len(token.key) == 40
+
+
+def test_apitoken_save_create_keeps_key(user0):
+    token = APIToken(user=user0, key='1234abcd')
+
+    token.save()
+
+    assert token.key == '1234abcd'
+
+
+def test_apitoken_save_update_keeps_key(token0, user1):
+    key = token0.key
+
+    token0.user = user1
+    token0.save()
+
+    assert token0.key == key
+
+
+def test_apitoken_custom_generate_key(monkeypatch, user0):
+    monkeypatch.setattr(APIToken, 'generate_key', lambda self: '2345bcde')
+    token = APIToken(user=user0)
+
+    token.save()
+
+    assert token.key == '2345bcde'
+
+
+def test_apitoken_manager_create_token(user0):
+    token = APIToken.objects.create_token(user=user0)
+
+    assert len(token.key) == 40