def view_ta_assign(self, parse_cmd): current_user = self.whos_logged_in() if len(parse_cmd) != 1: return "View TA assignments not of the right format: [view_ta_assign]" if current_user.type == "ta": tee_ayy = TA(current_user.email, current_user.password, current_user.type) return tee_ayy.view_ta_assignments() if current_user.type == "instructor": instructor = Instructor(current_user.email, current_user.password, current_user.type) return instructor.view_ta_assign() else: return "You don't have access to that command."
class TestTA(TestCase): def setUp(self): self.ta0 = TA("*****@*****.**", "password", "ta") self.ta1 = TA("*****@*****.**", "password", "ta") self.ta2 = TA("*****@*****.**", "password", "ta") self.ta3 = TA("*****@*****.**", "password", "ta") self.course1 = Course("CS101", 0) self.course2 = Course("CS201", 0) self.course3 = Course("CS301", 0) def test_view_ta_assignments1(self): mod_course1 = models.ModelCourse.objects.get(course_id="CS101") mod_ta_course1 = models.ModelTACourse() mod_ta_course1.course = mod_course1 mod_ta1 = models.ModelPerson.objects.get(email="*****@*****.**") mod_ta_course1.TA = mod_ta1 mod_ta_course1.save() self.assertEqual(self.ta1.view_ta_assignments()[0], "Course: CS101 TA: DEFAULT, [email protected]") def test_view_ta_assignments2(self): mod_course2 = models.ModelCourse.objects.get(course_id="CS201") mod_ta_course2 = models.ModelTACourse() mod_ta_course2.course = mod_course2 mod_ta2 = models.ModelPerson.objects.get(email="*****@*****.**") mod_ta_course2.TA = mod_ta2 mod_ta_course2.save() self.assertEqual(self.ta2.view_ta_assignments()[0], "Course: CS201 TA: DEFAULT, [email protected]") def test_view_ta_assignments3(self): mod_course3 = models.ModelCourse.objects.get(course_id="CS301") mod_ta_course3 = models.ModelTACourse() mod_ta_course3.course = mod_course3 mod_ta3 = models.ModelPerson.objects.get(email="*****@*****.**") mod_ta_course3.TA = mod_ta3 mod_ta_course3.save() self.assertEqual(self.ta3.view_ta_assignments()[0], "Course: CS301 TA: DEFAULT, [email protected]") def test_view_ta_assignments4(self): mod_course1 = models.ModelCourse.objects.get(course_id="CS101") mod_ta_course1 = models.ModelTACourse() mod_ta_course1.course = mod_course1 mod_ta1 = models.ModelPerson.objects.get(email="*****@*****.**") mod_ta_course1.TA = mod_ta1 mod_ta_course1.save() mod_course2 = models.ModelCourse.objects.get(course_id="CS201") mod_ta_course2 = models.ModelTACourse() mod_ta_course2.course = mod_course2 mod_ta2 = models.ModelPerson.objects.get(email="*****@*****.**") mod_ta_course2.TA = mod_ta2 mod_ta_course2.save() mod_course3 = models.ModelCourse.objects.get(course_id="CS301") mod_ta_course3 = models.ModelTACourse() mod_ta_course3.course = mod_course3 mod_ta3 = models.ModelPerson.objects.get(email="*****@*****.**") mod_ta_course3.TA = mod_ta3 mod_ta_course3.save() self.assertEqual(self.ta0.view_ta_assignments()[0], "Course: CS101 TA: DEFAULT, [email protected]") self.assertEqual(self.ta0.view_ta_assignments()[1], "Course: CS201 TA: DEFAULT, [email protected]") self.assertEqual(self.ta0.view_ta_assignments()[2], "Course: CS301 TA: DEFAULT, [email protected]") self.assertEqual(self.ta0.view_ta_assignments(), [ 'Course: CS101 TA: DEFAULT, [email protected]', 'Course: CS201 TA: DEFAULT, [email protected]', 'Course: CS301 TA: DEFAULT, [email protected]' ]) def test_read_public_contact(self): # self.assertEqual(self.ta1.read_public_contact(self.class_list)) pass def test_change_password(self): self.TA1 = TA("*****@*****.**", "DEFAULT_PASSWORD", "DEFAULT") self.assertTrue(self.TA1.change_password("password")) self.assertEquals(self.TA1.password, "password") self.assertNotEquals(self.TA1.password, "DEFAULT_PASSWORD") model_person1 = models.ModelPerson.objects.get(email=self.TA1.email) self.assertEquals(model_person1.password, "password") def test_change_email(self): self.TA1 = TA("*****@*****.**", "DEFAULT_PASSWORD", "DEFAULT") self.person2 = Person("*****@*****.**", "DEFAULT_PASSWORD", "DEFAULT") self.TA1.change_email("*****@*****.**") model_person1 = models.ModelPerson.objects.get(email=self.TA1.email) self.assertEquals(model_person1.email, "*****@*****.**") self.assertEquals(self.TA1.email, "*****@*****.**") self.assertNotEquals(self.TA1.email, "*****@*****.**") self.assertFalse(self.TA1.change_email("*****@*****.**")) self.assertFalse( self.TA1.change_email("no_at_symbol_or_dot_something")) self.assertFalse(self.TA1.change_email("*****@*****.**")) def test_change_phone(self): self.TA1 = TA("*****@*****.**", "DEFAULT_PASSWORD", "DEFAULT") self.TA1.change_phone("414.414.4141") model_person1 = models.ModelPerson.objects.get(email=self.TA1.email) self.assertEquals(model_person1.phone, "414.414.4141") self.assertEquals(self.TA1.phone_number, "414.414.4141") self.assertNotEquals(self.TA1.phone_number, "000.000.0000") self.assertFalse(self.TA1.change_phone("1234567890")) self.assertFalse(self.TA1.change_phone("414-414-4141")) self.assertFalse(self.TA1.change_phone("(414)414-4141")) self.assertFalse(self.TA1.change_phone("abc.abc.abcd")) self.assertFalse(self.TA1.change_phone("1234.1234.1234")) def test_change_name(self): self.TA1 = TA("*****@*****.**", "DEFAULT_PASSWORD", "DEFAULT") self.TA1.change_name("Snoop Doggy Dog") model_person1 = models.ModelPerson.objects.get(email=self.TA1.email) self.assertEquals(model_person1.name, "Snoop Doggy Dog") self.assertEquals(self.TA1.name, "Snoop Doggy Dog") self.assertNotEquals(self.TA1.name, "DEFAULT")