Tivix/django-rest-auth
1
1
Fork 0
mirror of https://github.com/Tivix/django-rest-auth.git synced 2025-08-23 17:54:44 +03:00
Code Issues Packages Projects Releases Wiki Activity

Stuggling with the logout tests...

Browse Source
This commit is contained in:
Daniel Stanton 2017-01-26 18:44:31 +00:00
parent 5961c48a07
commit eb2684fd2e
2 changed files with 60 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
rest_auth/tests/settings.py
View File

@ -66,6 +66,15 @@ REST_FRAMEWORK = {
) )
} }
REST_FRAMEWORK_KNOX = {
'DEFAULT_AUTHENTICATION_CLASSES': (
'knox.auth.TokenAuthentication',
),
'DEFAULT_PERMISSION_CLASSES': (
'rest_framework.permissions.IsAuthenticated',
),
}
INSTALLED_APPS = [ INSTALLED_APPS = [
'django.contrib.admin', 'django.contrib.admin',
'django.contrib.auth', 'django.contrib.auth',

51
rest_auth/tests/test_api.py
View File

@ -1,15 +1,26 @@
from django.core.urlresolvers import reverse from django.core.urlresolvers import reverse
from django.test import TestCase, override_settings from django.test import TestCase, override_settings
from django.contrib.auth import get_user_model from django.contrib.auth import get_user_model
from django.core import mail from django.core import mail
from django.conf import settings from django.conf import settings
from django.utils.encoding import force_text from django.utils.encoding import force_text
from django.utils.six.moves import reload_module
from rest_framework import status from rest_framework import status
from rest_framework.test import APIClient
from rest_framework import views as drf_views
from unittest.mock import patch
from allauth.account import app_settings as account_app_settings from allauth.account import app_settings as account_app_settings
from knox.models import AuthToken
from .test_base import BaseAPITestCase from .test_base import BaseAPITestCase
from .settings import REST_FRAMEWORK_KNOX
client = APIClient()
@override_settings(ROOT_URLCONF="tests.urls") @override_settings(ROOT_URLCONF="tests.urls")
class APITestCase1(TestCase, BaseAPITestCase): class APITestCase1(TestCase, BaseAPITestCase):
""" """
@ -541,3 +552,43 @@ class APITestCase1(TestCase, BaseAPITestCase):
self.post(self.login_url, data=payload, status_code=status.HTTP_200_OK) self.post(self.login_url, data=payload, status_code=status.HTTP_200_OK)
self.get(self.logout_url, status_code=status.HTTP_405_METHOD_NOT_ALLOWED) self.get(self.logout_url, status_code=status.HTTP_405_METHOD_NOT_ALLOWED)
# @override_settings(REST_USE_KNOX=True, REST_FRAMEWORK = {
# 'DEFAULT_AUTHENTICATION_CLASSES': ('knox.auth.TokenAuthentication')},)
def test_logout_knox(self):
with override_settings(REST_USE_KNOX=True, REST_FRAMEWORK=REST_FRAMEWORK_KNOX):
reload_module(drf_views)
payload = {
"username": self.USERNAME,
"password": self.PASS
}
get_user_model().objects.create_user(self.USERNAME, '', self.PASS)
self.client = APIClient()
response = self.client.post(self.login_url, data=payload, status_code=status.HTTP_200_OK)
self.client.credentials(HTTP_AUTHORIZATION=('Token %s' % response.data['token']))
self.client.post(self.logout_url, status_code=status.HTTP_200_OK)
self.assertEqual(AuthToken.objects.count(), 0)
reload_module(drf_views)
@override_settings(REST_USE_KNOX=True, REST_FRAMEWORK = {
'DEFAULT_AUTHENTICATION_CLASSES': ('knox.auth.TokenAuthentication')},)
def test_logout_all_knox(self):
payload = {
"username": self.USERNAME,
"password": self.PASS
}
get_user_model().objects.create_user(self.USERNAME, '', self.PASS)
self.logout_all_url = reverse('rest_logout_all')
self.client = APIClient()
self.client.post(self.login_url, data=payload, status_code=status.HTTP_200_OK)
response = self.client.post(self.login_url, data=payload, status_code=status.HTTP_200_OK)
self.client.credentials(HTTP_AUTHORIZATION=('Token %s' % response.data['token']))
self.client.post(self.logout_all_url, status_code=status.HTTP_200_OK)
self.assertEqual(AuthToken.objects.count(), 0)