12d40122d1
Our old manual wrapped didn't properly work in some aspects, this version now should hopefully take care of most common proxy object pitfalls.
56 lines
2.2 KiB
Python
56 lines
2.2 KiB
Python
# coding=utf-8
|
|
"""
|
|
Unit tests for octoprint.users.SessionUser
|
|
"""
|
|
from __future__ import absolute_import, division, print_function
|
|
|
|
__author__ = "Gina Häußge <osd@foosel.net>"
|
|
__license__ = 'GNU Affero General Public License http://www.gnu.org/licenses/agpl.html'
|
|
__copyright__ = "Copyright (C) 2015 The OctoPrint Project - Released under terms of the AGPLv3 License"
|
|
|
|
import unittest
|
|
|
|
import octoprint.users
|
|
|
|
class SessionUserTestCase(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
self.user = octoprint.users.User("username", "passwordHash", True, ("user",), apikey="apikey", settings=dict(key="value"))
|
|
|
|
def test_two_sessions(self):
|
|
session1 = octoprint.users.SessionUser(self.user)
|
|
session2 = octoprint.users.SessionUser(self.user)
|
|
|
|
# session should be different, wrapped object should be identical
|
|
self.assertNotEqual(session1.session, session2.session)
|
|
self.assertEqual(session1.__wrapped__, session2.__wrapped__)
|
|
self.assertEqual(session1.get_name(), session2.get_name())
|
|
|
|
def test_settings_change_propagates(self):
|
|
session1 = octoprint.users.SessionUser(self.user)
|
|
session2 = octoprint.users.SessionUser(self.user)
|
|
|
|
# change should propagate from User to SessionUser
|
|
self.user.set_setting("otherkey", "othervalue")
|
|
self.assertDictEqual(dict(key="value", otherkey="othervalue"), session1.get_all_settings())
|
|
|
|
# change should propagate from SessionUser to SessionUser
|
|
session2.set_setting("otherkey", "yetanothervalue")
|
|
self.assertDictEqual(dict(key="value", otherkey="yetanothervalue"), session1.get_all_settings())
|
|
|
|
def test_repr(self):
|
|
user = octoprint.users.SessionUser(self.user)
|
|
expected = "SessionUser({!r},session={},created={})".format(self.user, user.session, user.created)
|
|
self.assertEqual(expected, repr(user))
|
|
|
|
def test_isinstance(self):
|
|
session = octoprint.users.SessionUser(self.user)
|
|
|
|
# needs to be detected as User instance
|
|
self.assertTrue(isinstance(session, octoprint.users.User))
|
|
|
|
# also needs to be detected as SessionUser instance
|
|
self.assertTrue(isinstance(session, octoprint.users.SessionUser))
|
|
|
|
# but wrapped user should NOT be detected as SessionUser instance of course
|
|
self.assertFalse(isinstance(self.user, octoprint.users.SessionUser))
|