def setUp(self, problem_type, staff=False):
"""
Configure page objects to test Open Assessment.
Args:
problem_type (str): The type of problem being tested,
used to choose which part of the course to load.
staff (bool): If True, runs the test with a staff user (defaults to False).
"""
super(OpenAssessmentTest, self).setUp()
self.problem_loc = self.PROBLEM_LOCATIONS[problem_type]
self.auto_auth_page = AutoAuthPage(self.browser,
course_id=self.TEST_COURSE_ID,
staff=staff)
self.submission_page = SubmissionPage(self.browser, self.problem_loc)
self.self_asmnt_page = AssessmentPage('self-assessment', self.browser,
self.problem_loc)
self.peer_asmnt_page = AssessmentPage('peer-assessment', self.browser,
self.problem_loc)
self.student_training_page = AssessmentPage('student-training',
self.browser,
self.problem_loc)
self.grade_page = GradePage(self.browser, self.problem_loc)
