def test_get_year_from_course_itgk_3rd_grader_fall(self):
course = load_course(ITGK_EXPIRED, years_ago=2)
course_finish_date = get_course_finish_date(course)
year = get_year_from_course(course, course_finish_date)
self.assertEqual(3, year)
def test_get_year_from_course_project1_2nd_grader_fall(self):
course = load_course(PROJECT1_ACTIVE, years_ago=0)
course_finish_date = get_course_finish_date(course)
year = get_year_from_course(course, course_finish_date)
self.assertEqual(2, year)
def test_get_year_from_course_itgk_1st_grader_spring(self):
course = load_course(ITGK_EXPIRED, years_ago=1)
year = get_year_from_course(course, get_course_finish_date(course))
self.assertEqual(1, year)
def test_get_course_finish_date_illegal_date(self):
course = load_course(ITGK_ACTIVE)
course['membership']['notAfter'] = 'definitely not a date'
self.assertEqual(None, get_course_finish_date(course))
def test_get_course_finish_date_no_date(self):
course = load_course(ITGK_ACTIVE)
# Sneaky hacky
del course['membership']['notAfter']
self.assertEqual(None, get_course_finish_date(course))
def test_get_course_finish_date(self):
course = load_course(ITGK_EXPIRED)
expiry_date = parse(course.get('membership').get('notAfter'))
self.assertEqual(expiry_date, get_course_finish_date(course))
def test_get_course_finish_date_illegal_date(self):
course = load_course(ITGK_ACTIVE)
course["membership"]["notAfter"] = "definitely not a date"
self.assertEqual(None, get_course_finish_date(course))
