def test_normal_upload_does_not_call_centralauth(self): data = 'Actual validation tested elsewhere' centralauth_called = [False] def expand_via_ca_mock(cohort_users, session): centralauth_called[0] = True return [] ca_service_mock = CentralAuthService() ca_service_mock.expand_via_centralauth = expand_via_ca_mock with centralauth_service_set(app, ca_service_mock): self.app.post('/cohorts/upload', data=dict(name='new_cohort_name', project='wiki', csv=(StringIO(data), 'cohort.csv'), validate_as_user_ids='True', centralauth=False)) assert_true(not centralauth_called[0])
def test_normal_upload_does_not_call_centralauth(self): data = 'Actual validation tested elsewhere' centralauth_called = [False] def expand_via_ca_mock(cohort_users, session): centralauth_called[0] = True return [] ca_service_mock = CentralAuthService() ca_service_mock.expand_via_centralauth = expand_via_ca_mock with centralauth_service_set(app, ca_service_mock): self.app.post('/cohorts/upload', data=dict( name='new_cohort_name', project='wiki', csv=(StringIO(data), 'cohort.csv'), validate_as_user_ids='True', centralauth=False )) assert_true(not centralauth_called[0])
def setup_cohort_service(): if request.endpoint is not None: if request.path.startswith('/cohorts'): cohort_service = getattr(g, 'cohort_service', None) tag_service = getattr(g, 'tag_service', None) centralauth_service = getattr(g, 'centralauth_service', None) if cohort_service is None: g.cohort_service = CohortService() if tag_service is None: g.tag_service = TagService() if centralauth_service is None: g.centralauth_service = CentralAuthService()
def test_upload_with_centralauth_expansion_works(self): data = 'Actual validation tested elsewhere' centralauth_called = [False] def expand_via_ca_mock(cohort_users, session): assert_equal(cohort_users[0]['raw_id_or_name'], data) centralauth_called[0] = True return [] ca_service_mock = CentralAuthService() ca_service_mock.expand_via_centralauth = expand_via_ca_mock with centralauth_service_set(app, ca_service_mock): response = self.app.post('/cohorts/upload', data=dict( name='new_cohort_name', project='wiki', csv=(StringIO(data), 'cohort.csv'), validate_as_user_ids='False', centralauth=True )) assert_true(centralauth_called[0]) assert_equal(response.status_code, 302) assert_true(response.data.find('href="/cohorts/#') >= 0)
def test_upload_with_centralauth_expansion_works(self): data = 'Actual validation tested elsewhere' centralauth_called = [False] def expand_via_ca_mock(cohort_users, session): assert_equal(cohort_users[0]['raw_id_or_name'], data) centralauth_called[0] = True return [] ca_service_mock = CentralAuthService() ca_service_mock.expand_via_centralauth = expand_via_ca_mock with centralauth_service_set(app, ca_service_mock): response = self.app.post('/cohorts/upload', data=dict(name='new_cohort_name', project='wiki', csv=(StringIO(data), 'cohort.csv'), validate_as_user_ids='False', centralauth=True)) assert_true(centralauth_called[0]) assert_equal(response.status_code, 302) assert_true(response.data.find('href="/cohorts/#') >= 0)
def setup_filemanager(): if request.endpoint is not None: if request.path.startswith('/reports'): file_manager = getattr(g, 'file_manager', None) if file_manager is None: g.file_manager = PublicReportFileManager( app.logger, app.absolute_path_to_app_root) cohort_service = getattr(g, 'cohort_service', None) centralauth_service = getattr(g, 'centralauth_service', None) if cohort_service is None: g.cohort_service = CohortService() if centralauth_service is None: g.centralauth_service = CentralAuthService()
def setUp(self): DatabaseTest.setUp(self) self.expand = CentralAuthService().expand_via_centralauth