def test_get_prerequisite_milestone_returns_milestone(self): """ Test test_get_prerequisite_milestone_returns_milestone """ gating_api.add_prerequisite(self.course.id, self.seq1.location) prereq = gating_api._get_prerequisite_milestone(self.seq1.location) # pylint: disable=protected-access self.assertIsNotNone(prereq)
def test_get_prerequisite_milestone_returns_none(self, mock_log): """ Test test_get_prerequisite_milestone_returns_none """ prereq = gating_api._get_prerequisite_milestone(self.seq1.location) # pylint: disable=protected-access self.assertIsNone(prereq) self.assertTrue(mock_log.called)