Commit d240f3e6 by Dennis Jen

Fill enrollment trend with previous day's count.

parent fb77533e
......@@ -191,7 +191,7 @@ class CourseEnrollmentPresenter(BaseCourseEnrollmentPresenter):
# add zero for the day prior (prevents just a single point in the chart)
if len(trends) == 1:
day_before = self.parse_api_date(trends[0]['date']) - datetime.timedelta(days=1)
trends.insert(0, self._build_empty_trend(day_before))
trends.insert(0, self._create_enrollment_datapoint(day_before, 0))
return summary, trends
......@@ -205,12 +205,13 @@ class CourseEnrollmentPresenter(BaseCourseEnrollmentPresenter):
expected_date = start_date + datetime.timedelta(days=day_change)
current_date = self.parse_api_date(api_response[day_change]['date'])
if current_date > expected_date:
api_response.insert(day_change, self._build_empty_trend(expected_date))
api_response.insert(day_change, self._create_enrollment_datapoint(
expected_date, api_response[day_change]['count']))
return api_response
def _build_empty_trend(self, day):
trend = {'date': day.isoformat(), 'count': 0}
def _create_enrollment_datapoint(self, day, count):
trend = {'date': day.isoformat(), 'count': count}
return trend
def _translate_country_names(self, data):
......
......@@ -50,12 +50,12 @@ def get_mock_enrollment_summary_and_trend(course_id):
def get_mock_presenter_enrollment_trend(course_id):
trend = get_mock_api_enrollment_data(course_id)
# presenter data has gaps filled in
empty_enrollment = {
'count': 0,
# presenter data has gaps filled in with previous day's count
filled_enrollment = {
'count': 1,
'date': '2013-12-31'
}
trend.insert(1, empty_enrollment)
trend.insert(1, filled_enrollment)
return trend
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment