Source code for camcops_server.cc_modules.tests.cc_session_tests

#!/usr/bin/env python

"""
camcops_server/cc_modules/tests/cc_session_tests.py

===============================================================================

    Copyright (C) 2012, University of Cambridge, Department of Psychiatry.
    Created by Rudolf Cardinal (rnc1001@cam.ac.uk).

    This file is part of CamCOPS.

    CamCOPS is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    CamCOPS is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with CamCOPS. If not, see <https://www.gnu.org/licenses/>.

===============================================================================

"""

from pendulum import DateTime as Pendulum

from camcops_server.cc_modules.cc_session import CamcopsSession, generate_token
from camcops_server.cc_modules.cc_taskfilter import TaskFilter
from camcops_server.cc_modules.cc_unittest import (
    BasicDatabaseTestCase,
    DemoDatabaseTestCase,
)
from camcops_server.cc_modules.cc_user import User


# =============================================================================
# Unit tests
# =============================================================================


[docs]class SessionTests(DemoDatabaseTestCase): """ Unit tests. """ def test_sessions(self) -> None: self.announce("test_sessions") req = self.req dbsession = self.dbsession self.assertIsInstance(generate_token(), str) CamcopsSession.delete_old_sessions(req) self.assertIsInstance( CamcopsSession.get_oldest_last_activity_allowed(req), Pendulum ) s = req.camcops_session u = self.dbsession.query(User).first() # type: User assert u, "Missing user in demo database!" self.assertIsInstance(s.last_activity_utc_iso, str) self.assertIsInstanceOrNone(s.username, str) s.logout() s.login(u) self.assertIsInstance(s.get_task_filter(), TaskFilter) # Now test deletion cascade dbsession.commit() numfilters = dbsession.query(TaskFilter).count() assert numfilters == 1, "TaskFilter count should be 1" dbsession.delete(s) dbsession.commit() numfilters = dbsession.query(TaskFilter).count() assert ( numfilters == 0 ), "TaskFilter count should be 0; cascade delete not working"
[docs]class GetSessionTests(BasicDatabaseTestCase): old_ip_addr = "192.0.2.1" new_ip_addr = "192.0.2.2"
[docs] def setUp(self) -> None: super().setUp() CamcopsSession.delete_old_sessions(self.req) self.old_session = CamcopsSession( ip_addr=self.old_ip_addr, last_activity_utc=self.req.now_utc ) self.dbsession.add(self.old_session) self.dbsession.flush()
def test_old_session_for_same_ip(self) -> None: self.req.remote_addr = self.old_ip_addr new_session = CamcopsSession.get_session( self.req, str(self.old_session.id), self.old_session.token ) self.dbsession.add(new_session) self.dbsession.flush() self.assertEqual(self.old_session.id, new_session.id) def test_old_session_for_different_ip_when_ip_ignored(self) -> None: self.req.config.session_check_user_ip = False self.req.remote_addr = self.new_ip_addr new_session = CamcopsSession.get_session( self.req, str(self.old_session.id), self.old_session.token ) self.dbsession.add(new_session) self.dbsession.flush() self.assertEqual(self.old_session.id, new_session.id) def test_new_session_for_different_ip_when_ip_checked(self) -> None: self.req.config.session_check_user_ip = True self.req.remote_addr = self.new_ip_addr new_session = CamcopsSession.get_session( self.req, str(self.old_session.id), self.old_session.token ) self.dbsession.add(new_session) self.dbsession.flush() self.assertNotEqual(self.old_session.id, new_session.id)