90 lines
2.4 KiB
Python
90 lines
2.4 KiB
Python
|
import json
|
||
|
|
|
||
|
|
from rest_framework.test import APIClient
|
||
|
|
|
||
|
|
from openslides.utils.test import TestCase
|
||
|
|
|
||
|
|
|
||
|
|
class TestWhoAmIView(TestCase):
|
||
|
|
url = '/users/whoami/'
|
||
|
|
|
||
|
|
def test_get_anonymous(self):
|
||
|
|
response = self.client.get(self.url)
|
||
|
|
|
||
|
|
self.assertEqual(response.status_code, 200)
|
||
|
|
self.assertEqual(response.content, b'{"user_id":null}')
|
||
|
|
|
||
|
|
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(response.content, b'{"user_id":1}')
|
||
|
|
|
||
|
|
def test_post(self):
|
||
|
|
response = self.client.post(self.url)
|
||
|
|
|
||
|
|
self.assertEqual(response.status_code, 405)
|
||
|
|
|
||
|
|
|
||
|
|
class TestUserLogoutView(TestCase):
|
||
|
|
url = '/users/logout/'
|
||
|
|
|
||
|
|
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, 200)
|
||
|
|
|
||
|
|
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'))
|
||
|
|
|
||
|
|
|
||
|
|
class TestUserLoginView(TestCase):
|
||
|
|
url = '/users/login/'
|
||
|
|
|
||
|
|
def setUp(self):
|
||
|
|
self.client = APIClient()
|
||
|
|
|
||
|
|
def test_get(self):
|
||
|
|
response = self.client.get(self.url)
|
||
|
|
|
||
|
|
self.assertEqual(response.status_code, 405)
|
||
|
|
|
||
|
|
def test_post_no_data(self):
|
||
|
|
response = self.client.post(self.url)
|
||
|
|
|
||
|
|
self.assertEqual(response.status_code, 200)
|
||
|
|
self.assertEqual(response.content, b'{"success":false}')
|
||
|
|
|
||
|
|
def test_post_correct_data(self):
|
||
|
|
response = self.client.post(
|
||
|
|
self.url,
|
||
|
|
{'username': 'admin', 'password': 'admin'})
|
||
|
|
|
||
|
|
self.assertEqual(response.status_code, 200)
|
||
|
|
self.assertEqual(
|
||
|
|
json.loads(response.content.decode('utf-8')),
|
||
|
|
{'success': True, 'user_id': 1})
|
||
|
|
|
||
|
|
def test_post_incorrect_data(self):
|
||
|
|
response = self.client.post(
|
||
|
|
self.url,
|
||
|
|
{'username': 'wrong', 'password': 'wrong'})
|
||
|
|
|
||
|
|
self.assertEqual(response.status_code, 200)
|
||
|
|
self.assertEqual(
|
||
|
|
json.loads(response.content.decode('utf-8')),
|
||
|
|
{'success': False})
|