124 lines
3.5 KiB
Python
124 lines
3.5 KiB
Python
import json
|
|
|
|
from django.urls import reverse
|
|
from rest_framework.test import APIClient
|
|
|
|
from tests.test_case import TestCase
|
|
|
|
|
|
class TestWhoAmIView(TestCase):
|
|
url = reverse("user_whoami")
|
|
|
|
def setUp(self):
|
|
pass
|
|
|
|
def test_get_anonymous(self):
|
|
response = self.client.get(self.url)
|
|
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertEqual(
|
|
json.loads(response.content.decode()),
|
|
{
|
|
"auth_type": "default",
|
|
"user_id": None,
|
|
"user": None,
|
|
"permissions": [],
|
|
"guest_enabled": False,
|
|
},
|
|
)
|
|
|
|
def test_get_authenticated_user(self):
|
|
self.client.login(username="admin", password="admin")
|
|
|
|
response = self.client.get(self.url)
|
|
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertEqual(json.loads(response.content.decode()).get("user_id"), 1)
|
|
self.assertEqual(
|
|
json.loads(response.content.decode()).get("guest_enabled"), False
|
|
)
|
|
|
|
def test_post(self):
|
|
response = self.client.post(self.url)
|
|
|
|
self.assertEqual(response.status_code, 405)
|
|
|
|
|
|
class TestUserLogoutView(TestCase):
|
|
url = reverse("user_logout")
|
|
|
|
def setUp(self):
|
|
pass
|
|
|
|
def test_get(self):
|
|
response = self.client.get(self.url)
|
|
|
|
self.assertEqual(response.status_code, 405)
|
|
|
|
def test_post_anonymous(self):
|
|
response = self.client.post(self.url)
|
|
|
|
self.assertEqual(response.status_code, 400)
|
|
|
|
def test_post_authenticated_user(self):
|
|
self.client.login(username="admin", password="admin")
|
|
self.client.session["test_key"] = "test_value"
|
|
|
|
response = self.client.post(self.url)
|
|
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertFalse(hasattr(self.client.session, "test_key"))
|
|
self.assertEqual(
|
|
json.loads(response.content.decode()),
|
|
{
|
|
"auth_type": "default",
|
|
"user_id": None,
|
|
"user": None,
|
|
"permissions": [],
|
|
"guest_enabled": False,
|
|
},
|
|
)
|
|
|
|
|
|
class TestUserLoginView(TestCase):
|
|
url = reverse("user_login")
|
|
|
|
def setUp(self):
|
|
self.client = APIClient()
|
|
|
|
def test_get(self):
|
|
response = self.client.get(self.url)
|
|
|
|
self.assertEqual(response.status_code, 200)
|
|
content = json.loads(response.content.decode())
|
|
self.assertTrue("login_info_text" in content)
|
|
self.assertTrue("privacy_policy" in content)
|
|
self.assertTrue("legal_notice" in content)
|
|
self.assertTrue("theme" in content)
|
|
self.assertTrue("logo_web_header" in content)
|
|
|
|
def test_post_no_data(self):
|
|
response = self.client.post(self.url)
|
|
|
|
self.assertEqual(response.status_code, 400)
|
|
|
|
def test_post_correct_data(self):
|
|
response = self.client.post(
|
|
self.url, {"username": "admin", "password": "admin"}
|
|
)
|
|
|
|
self.assertEqual(response.status_code, 200)
|
|
content = json.loads(response.content.decode())
|
|
self.assertEqual(content.get("user_id"), 1)
|
|
self.assertTrue(isinstance(content.get("user"), dict))
|
|
self.assertTrue(isinstance(content.get("permissions"), list))
|
|
self.assertFalse(content.get("guest_enabled", True))
|
|
self.assertEqual(content.get("auth_type"), "default")
|
|
|
|
def test_post_incorrect_data(self):
|
|
response = self.client.post(
|
|
self.url, {"username": "wrong", "password": "wrong"}
|
|
)
|
|
|
|
self.assertEqual(response.status_code, 400)
|