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 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 _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())
def setUp(self): self.service_factory = ServicesFactory(Mock())
class ServicesFactoryTest(unittest.TestCase): def setUp(self): self.service_factory = ServicesFactory(Mock()) def test_users_has_no_default_sessions(self): user_id = ANY self.assertFalse(self.service_factory.has_session(user_id)) def test_add_user_sessions(self): user_id = 'irrelevant' some_service = Mock() self.service_factory.add_session(user_id, some_service) self.assertTrue(self.service_factory.has_session(user_id)) self.assertEqual(some_service, self.service_factory.services(user_id)) def test_online_sessions_counts_logged_in_users(self): self.service_factory.add_session('some_id1', mock()) self.service_factory.add_session('some_id2', mock()) self.assertEqual(2, self.service_factory.online_sessions()) @patch('pixelated.config.services.Services.setup') def test_create_services_from_leap_session_sets_up_services_and_add_a_user_session(self, mock_setup_services): leap_session = Mock() user_id = 'irrelevant' leap_session.user_auth.uuid = user_id self.service_factory.create_services_from(leap_session) self.assertTrue(mock_setup_services.called) self.assertTrue(self.service_factory.has_session(user_id)) def test_destroy_session_using_close_user_services_and_deletes_sessions(self): user_id = 'irrelevant' some_service = Mock() self.service_factory.add_session(user_id, some_service) self.service_factory.destroy_session(user_id) self.assertFalse(self.service_factory.has_session(user_id)) self.assertTrue(some_service.close.called) def test_sessions_can_be_destroyed_using_email_rather_than_uuid(self): user_id = 'irrelevant' username = '******' email = '*****@*****.**' % username some_service = Mock() self.service_factory.add_session(user_id, some_service) self.service_factory.map_email(username, user_id) self.service_factory.destroy_session(email, using_email=True) self.assertFalse(self.service_factory.has_session(user_id)) self.assertTrue(some_service.close.called)