From dbab455b1337d6fdb2ff0d124deefd53dca283a6 Mon Sep 17 00:00:00 2001
From: Artem Muterko <artieua@gmail.com>
Date: Sat, 14 Jan 2017 19:45:25 +0200
Subject: [PATCH] Add couple of test for AuthToken

---
 tests/test_authtoken.py | 31 +++++++++++++++++++++++++++++++
 1 file changed, 31 insertions(+)
 create mode 100644 tests/test_authtoken.py

diff --git a/tests/test_authtoken.py b/tests/test_authtoken.py
new file mode 100644
index 000000000..331f4b564
--- /dev/null
+++ b/tests/test_authtoken.py
@@ -0,0 +1,31 @@
+import pytest
+from django.contrib.admin import site
+from django.contrib.auth.models import User
+from django.test import TestCase
+
+from rest_framework.authtoken.admin import TokenAdmin
+from rest_framework.authtoken.models import Token
+from rest_framework.authtoken.serializers import AuthTokenSerializer
+from rest_framework.exceptions import ValidationError
+
+
+class AuthTokenTests(TestCase):
+
+    def setUp(self):
+        self.site = site
+        self.user = User.objects.create_user(username='test_user')
+        self.token = Token.objects.create(key='test token', user=self.user)
+
+    def test_model_admin_displayed_fields(self):
+        mock_request = object()
+        token_admin = TokenAdmin(self.token, self.site)
+        assert token_admin.get_fields(mock_request) == ('user',)
+
+    def test_token_string_representation(self):
+        assert str(self.token) == 'test token'
+
+    def test_validate_raise_error_if_no_credentials_provided(self):
+        with pytest.raises(ValidationError):
+            AuthTokenSerializer().validate({})
+
+