'Visit the course outline page, restrict access to a unit.
Verify that there is a restricted group warning.
Remove the group access restriction and verify that there
is no longer a warning.'

def test_units_can_be_restricted(self):

    self.course_outline_page.visit()
    self.course_outline_page.expand_all_subsections()
    unit = self.course_outline_page.section_at(0).subsection_at(0).unit_at(0)
    self._set_restriction_on_unrestricted_unit(unit)
