47 lines
1.8 KiB
Python
47 lines
1.8 KiB
Python
|
from datetime import timedelta
|
||
|
|
|
||
|
|
from django.utils import timezone
|
||
|
|
|
||
|
|
from openslides.core.models import Session
|
||
|
|
from openslides.users.models import User
|
||
|
|
from openslides.utils.autoupdate import get_logged_in_users
|
||
|
|
from openslides.utils.test import TestCase
|
||
|
|
|
||
|
|
|
||
|
|
class TestGetLoggedInUsers(TestCase):
|
||
|
|
def test_call(self):
|
||
|
|
"""
|
||
|
|
Test to call the function with:
|
||
|
|
* A user that session has not expired
|
||
|
|
* A user that session has expired
|
||
|
|
* A user that has no session
|
||
|
|
* An anonymous user that session hot not expired
|
||
|
|
|
||
|
|
Only the user with the session that has not expired should be returned
|
||
|
|
"""
|
||
|
|
user1 = User.objects.create(username='user1')
|
||
|
|
user2 = User.objects.create(username='user2')
|
||
|
|
User.objects.create(username='user3')
|
||
|
|
|
||
|
|
# Create a session with a user, that expires in 5 hours
|
||
|
|
Session.objects.create(user=user1, expire_data=timezone.now() + timedelta(hours=5))
|
||
|
|
|
||
|
|
# Create a session with a user, that is expired before 5 hours
|
||
|
|
Session.objects.create(user=user2, expire_data=timezone.now() + timedelta(hours=-5))
|
||
|
|
|
||
|
|
# Create a session with an anonymous user, that expires in 5 hours
|
||
|
|
Session.objects.create(user=None, expire_data=timezone.now() + timedelta(hours=5))
|
||
|
|
|
||
|
|
self.assertEqual(list(get_logged_in_users()), [user1])
|
||
|
|
|
||
|
|
def test_unique(self):
|
||
|
|
"""
|
||
|
|
Test the function with a user that has two not expired session.
|
||
|
|
The user should be returned only once.
|
||
|
|
"""
|
||
|
|
user1 = User.objects.create(username='user1')
|
||
|
|
Session.objects.create(user=user1, expire_data=timezone.now() + timedelta(hours=1))
|
||
|
|
Session.objects.create(user=user1, expire_data=timezone.now() + timedelta(hours=2))
|
||
|
|
|
||
|
|
self.assertEqual(list(get_logged_in_users()), [user1])
|