def test_cohort_membership_basic_detail(self): def get_for_display_mock(session, user_id, by_id): assert_equal(by_id, unicode(self.cohort.id)) return self.cohort cohort_service_mock = CohortService() cohort_service_mock.get_for_display = get_for_display_mock with cohort_service_set(app, cohort_service_mock): response = self.app.get('/cohorts/{0}/membership'.format( self.cohort.id)) assert_equal(response.status_code, 200) assert_true(response.data.find(self.cohort.name) >= 0)
def test_cohort_membership_basic_detail(self): def get_for_display_mock(session, user_id, by_id): assert_equal(by_id, unicode(self.cohort.id)) return self.cohort cohort_service_mock = CohortService() cohort_service_mock.get_for_display = get_for_display_mock with cohort_service_set(app, cohort_service_mock): response = self.app.get('/cohorts/{0}/membership'.format( self.cohort.id )) assert_equal(response.status_code, 200) assert_true(response.data.find(self.cohort.name) >= 0)
def test_cohort_membership_full_detail(self): def get_for_display_mock(session, user_id, by_id): assert_equal(by_id, unicode(self.cohort.id)) return self.cohort def get_membership_mock(cohort, session): assert_equal(cohort, self.cohort) return 'mock' cohort_service_mock = CohortService() cohort_service_mock.get_for_display = get_for_display_mock cohort_service_mock.get_membership = get_membership_mock with cohort_service_set(app, cohort_service_mock): response = self.app.get('/cohorts/{0}/membership?full_detail=true'.format( self.cohort.id )) assert_equal(response.status_code, 200) assert_equal(json.loads(response.data), {'membership': 'mock'})
def test_cohort_membership_full_detail(self): def get_for_display_mock(session, user_id, by_id): assert_equal(by_id, unicode(self.cohort.id)) return self.cohort def get_membership_mock(cohort, session): assert_equal(cohort, self.cohort) return 'mock' cohort_service_mock = CohortService() cohort_service_mock.get_for_display = get_for_display_mock cohort_service_mock.get_membership = get_membership_mock with cohort_service_set(app, cohort_service_mock): response = self.app.get( '/cohorts/{0}/membership?full_detail=true'.format( self.cohort.id)) assert_equal(response.status_code, 200) assert_equal(json.loads(response.data), {'membership': 'mock'})