1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
|
from mockito import mock, when, any as ANY
from pixelated.resources.auth import PixelatedAuthSessionWrapper
from pixelated.resources.login_resource import LoginResource
from pixelated.resources.root_resource import RootResource
from test.unit.resources import DummySite
from twisted.cred import error
from twisted.cred.checkers import ANONYMOUS
from twisted.internet.defer import succeed, fail
from twisted.python import failure
from twisted.trial import unittest
from twisted.web._auth.wrapper import UnauthorizedResource
from twisted.web.resource import IResource
from twisted.web.test.requesthelper import DummyRequest
class TestPixelatedAuthSessionWrapper(unittest.TestCase):
def setUp(self):
self.portal_mock = mock()
self.user_uuid_mock = mock()
self.root_resource_mock = mock()
self.anonymous_resource_mock = mock()
credential_factories_mock = mock()
self.session_wrapper = PixelatedAuthSessionWrapper(self.portal_mock, self.root_resource_mock, self.anonymous_resource_mock, credential_factories_mock)
self.request = DummyRequest([])
self.request.prepath = ['']
self.request.path = '/'
def test_should_proxy_to_login_resource_when_the_user_is_not_logged_in(self):
when(self.portal_mock).login(ANY(), None, IResource).thenReturn(succeed((IResource, ANONYMOUS, lambda: None)))
deferred_resource = self.session_wrapper.getChildWithDefault('/', self.request)
d = deferred_resource.d
def assert_anonymous_resource(resource):
self.assertIs(resource, self.anonymous_resource_mock)
d.addCallback(assert_anonymous_resource)
return d
def test_should_proxy_to_root_resource_when_the_user_is_logged_in(self):
when(self.portal_mock).login(ANY(), None, IResource).thenReturn(succeed((IResource, self.user_uuid_mock, lambda: None)))
deferred_resource = self.session_wrapper.getChildWithDefault('/', self.request)
d = deferred_resource.d
def assert_root_resource(resource):
self.assertIs(resource, self.root_resource_mock)
d.addCallback(assert_root_resource)
return d
def test_should_proxy_to_unauthorized_resource_when_login_fails(self):
when(self.portal_mock).login(ANY(), None, IResource).thenReturn(fail(failure.Failure(error.UnhandledCredentials('dummy message'))))
deferred_resource = self.session_wrapper.getChildWithDefault('/', self.request)
d = deferred_resource.d
def assert_unauthorized_resource(resource):
self.assertIsInstance(resource, UnauthorizedResource)
d.addCallback(assert_unauthorized_resource)
return d
|
