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'})
