def start_client(self, mode=UserAgentMode(is_single_user=True)):
self._initialize()
self._test_account = AppTestAccount('test', self._tmp_dir.name)
yield self._test_account.start()
self.cleanup = lambda: self._test_account.cleanup()
self.soledad = self._test_account.soledad
self.service_factory = ServicesFactory(UserAgentMode(is_single_user=False))
root_resource = RootResource(self.service_factory)
leap_provider = mock()
self.resource = set_up_protected_resources(root_resource, leap_provider, self.service_factory)
def test_is_admin_is_queried_from_leap_auth(self):
leap_session = mock()
auth = mock()
auth.uuid = 'some_id1'
leap_session.user_auth = auth
leap_session.config = mock()
services = Services(leap_session)
service_factory = ServicesFactory(mock())
service_factory.add_session('some_id1', services)
when(auth).is_admin().thenReturn(True)
request = mock()
resource = UsersResource(service_factory)
when(resource)._get_user_id_from_request(request).thenReturn('some_id1')
self.assertTrue(resource.is_admin(request))
verify(auth).is_admin()
def _create_service_factory(args):
if args.single_user:
return SingleUserServicesFactory(UserAgentMode(is_single_user=True))
else:
return ServicesFactory(UserAgentMode(is_single_user=False))
def test_online_sessions_counts_logged_in_users(self):
service_factory = ServicesFactory(mock())
service_factory.add_session('some_id1', mock())
service_factory.add_session('some_id2', mock())
self.assertEqual(2, service_factory.online_sessions())
