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)
