def test_get_geography_data(self, mock_enrollment): mock_enrollment.return_value = utils.get_mock_api_enrollment_geography_data(self.course_id) expected_summary, expected_data = utils.get_mock_presenter_enrollment_geography_data() summary, actual_data = self.presenter.get_geography_data() self.assertDictEqual(summary, expected_summary) self.assertListEqual(actual_data, expected_data) # test with a small set of countries mock_data = utils.get_mock_api_enrollment_geography_data_limited(self.course_id) mock_enrollment.return_value = mock_data expected_summary, expected_data = utils.get_mock_presenter_enrollment_geography_data_limited() summary, actual_data = self.presenter.get_geography_data() self.assertDictEqual(summary, expected_summary) self.assertListEqual(actual_data, expected_data)
def get_mock_data(self, course_id): return get_mock_api_enrollment_geography_data(course_id)
def get_mock_enrollment_data(self): return get_mock_api_enrollment_geography_data(self.course_id)
def get_mock_data(self, course_id): return utils.get_mock_api_enrollment_geography_data(course_id)