Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Backend for show auth token. #1683

Merged
merged 3 commits into from May 28, 2018
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Jump to
Jump to file
Failed to load files.
Diff view
Diff view
1 change: 1 addition & 0 deletions apps/accounts/urls.py
Expand Up @@ -5,5 +5,6 @@
urlpatterns = [

url(r'^user/disable$', views.disable_user, name='disable_user'),
url(r'^user/get_auth_token$', views.get_auth_token, name='get_auth_token'),

]
29 changes: 28 additions & 1 deletion apps/accounts/views.py
@@ -1,12 +1,18 @@
from django.contrib.auth import logout
from django.contrib.auth.models import User

from rest_framework.authtoken.models import Token
from rest_framework.response import Response
from rest_framework import permissions, status
from rest_framework.decorators import (api_view,
authentication_classes,
permission_classes,)
permission_classes,
throttle_classes,)
from rest_framework.throttling import UserRateThrottle
from rest_framework_expiring_authtoken.authentication import (ExpiringTokenAuthentication,)

from .permissions import HasVerifiedEmail


@api_view(['POST'])
@permission_classes((permissions.IsAuthenticated,))
Expand All @@ -18,3 +24,24 @@ def disable_user(request):
user.save()
logout(request)
return Response(status=status.HTTP_200_OK)


@throttle_classes([UserRateThrottle])
@api_view(['GET'])
@permission_classes((permissions.IsAuthenticated, HasVerifiedEmail))
@authentication_classes((ExpiringTokenAuthentication,))
def get_auth_token(request):
try:
user = User.objects.get(email=request.user.email)
except User.DoesNotExist:
response_data = {"error": "This User account doesn't exist."}
Response(response_data, status.HTTP_404_NOT_FOUND)

try:
token = Token.objects.get(user=user)
except Token.DoesNotExist:
token = Token.objects.create(user=user)
token.save()

response_data = {"token": "{}".format(token)}
return Response(response_data, status=status.HTTP_200_OK)
3 changes: 3 additions & 0 deletions tests/unit/accounts/test_urls.py
Expand Up @@ -30,3 +30,6 @@ class TestStringMethods(BaseAPITestClass):
def test_disable_user(self):
url = reverse_lazy('accounts:disable_user')
self.assertEqual(unicode(url), '/api/accounts/user/disable')

url = reverse_lazy('accounts:get_auth_token')
self.assertEqual(unicode(url), '/api/accounts/user/get_auth_token')
14 changes: 14 additions & 0 deletions tests/unit/accounts/test_views.py
@@ -1,5 +1,7 @@
import os

from rest_framework.authtoken.models import Token

from django.core.urlresolvers import reverse_lazy
from django.contrib.auth.models import User

Expand Down Expand Up @@ -48,3 +50,15 @@ def test_cannot_update_username(self):
self.assertNotContains(response, 'anotheruser')
self.assertContains(response, 'someuser')
self.assertEqual(response.status_code, status.HTTP_200_OK)


class GetAuthTokenTest(BaseAPITestClass):

url = reverse_lazy('accounts:get_auth_token')

def test_get_auth_token(self):
response = self.client.get(self.url, {})
token = Token.objects.get(user=self.user)
expected_data = {"token": "{}".format(token)}
self.assertEqual(response.status_code, status.HTTP_200_OK)
self.assertEqual(response.data, expected_data)