class DjangoModelInterfaceTests(TestCase):
def setUp(self):
self.di = DjangoModelInterface()
@staticmethod
def hashed_password(password):
h = hashlib.new("md5")
h.update(f"{password}".encode("ascii"))
return h.hexdigest()
# region Testing User variations
def test_get_user_exists(self):
self.di.create_account("root", "root", "administrator")
self.assertIsNotNone(self.di.get_user("root"))
def test_get_user_doesnt_exist(self):
self.assertIsNone(self.di.get_user("nonexistent"))
def test_valid_role_admin(self):
self.assertTrue(self.di.is_valid_role("administrator"))
def test_valid_role_supervisor(self):
self.assertTrue(self.di.is_valid_role("supervisor"))
def test_valid_role_instructor(self):
self.assertTrue(self.di.is_valid_role("instructor"))
def test_valid_role_TA(self):
self.assertTrue(self.di.is_valid_role("TA"))
def test_invalid_role(self):
self.assertFalse(self.di.is_valid_role("invalid"))
# endregion
# region Testing account variations
def test_create_account(self):
name = "account"
password = "******"
role = "role"
self.di.create_account(name, password, role)
account = Account.objects.get(name=name)
self.assertEqual(account.name, name)
self.assertEqual(account.password, self.hashed_password(password))
self.assertEqual(account.role, role)
def test_create_account_no_user(self):
name = "name"
password = "******"
role = "role"
# wrong name
response = self.di.create_account("", password, role)
self.assertIsNone(response)
# wrong role
response = self.di.create_account(name, password, "")
self.assertIsNone(response)
def test_delete_account(self):
name = "account2delete"
self.di.create_account("account", "pass", "role")
self.di.create_account(name, "pass", "role")
self.di.delete_account("account")
self.di.delete_account(name)
self.assertEqual(len(Account.objects.filter(name=name)), 0)
def test_update_account(self):
name = "account"
new_pass = "******"
new_role = "newrole"
self.di.create_account(name, "pass", "role")
self.di.update_account(name, new_pass, new_role)
account = Account.objects.get(name=name)
self.assertNotEqual(account.password, self.hashed_password("pass"))
self.assertEqual(account.password, self.hashed_password(new_pass))
self.assertEqual(account.role, new_role)
def test_get_accounts(self):
password1 = "pass"
password2 = "pass2"
self.di.create_account("account", password1, "role")
self.di.create_account("account2", password2, "role2")
accounts = self.di.get_accounts()
self.assertEqual(accounts, [{
"name": "account",
"password": self.hashed_password(password1),
"role": "role"
}, {
"name": "account2",
"password": self.hashed_password(password2),
"role": "role2"
}])
# endregion
# region Testing Login variations
def test_get_set_logged_in(self):
self.di.create_account("account", "pass", "TA")
self.di.set_logged_in("account")
response = self.di.get_logged_in()
self.assertEqual(response, "account")
def test_get_not_logged_in(self):
self.di.create_account("account", "pass", "TA")
response = self.di.get_logged_in()
self.assertEqual(response, "")
def test_set_logged_in_set_logged_out(self):
self.di.create_account("account", "pass", "TA")
self.di.set_logged_in("account")
response = self.di.get_logged_in()
self.assertEqual(response, "account")
self.di.set_logged_out()
response = self.di.get_logged_in()
self.assertEqual(response, "")
def test_set_logged_in_twice(self):
self.di.create_account("account", "pass", "TA")
self.di.set_logged_in("account")
self.di.create_account("account_fail", "pass", "TA")
self.di.set_logged_in("account_fail")
# Tried to sign another account in before logout
response = self.di.get_logged_in()
self.assertEqual(response, "account")
# endregion
# region Testing Course variations
def test_create_course_get_courses(self):
number = "361"
name = "CompSci361"
self.di.create_course(number, name)
self.assertIsNotNone(
Course.objects.filter(number=number, name=name).first())
def test_create_set_course_instructor(self):
number = "361"
instructor_name = "jayson"
self.di.create_course(number, "CompSci")
self.di.create_account(instructor_name, "pass", "instructor")
self.di.set_course_instructor(number, instructor_name)
course = Course.objects.get(number=number)
self.assertEqual(course.instructor.first().name, instructor_name)
def test_create_set_course_no_instructor(self):
number = "361"
name = "CompSci361"
self.di.create_course(number, name)
course = Course.objects.get(number=number)
self.assertIsNone(course.instructor.first())
def test_create_set_course_instructor_twice(self):
number = "361"
instructor_name = "jayson"
self.di.create_course(number, "CompSci")
self.di.create_account(instructor_name, "pass", "instructor")
self.di.set_course_instructor(number, instructor_name)
# Try to set again
self.di.create_account("Not jayson", "pass", "instructor")
self.di.set_course_instructor(number, "Not jayson")
course = Course.objects.get(number=number)
self.assertEqual(course.instructor.first().name, instructor_name)
def test_get_course_exists(self):
self.di.create_course("123", "test_course")
self.assertIsNotNone(self.di.course_exists("123"))
def test_get_course_doesnt_exist(self):
self.assertIsNotNone(self.di.course_exists("000"))
def test_get_course_assigned(self):
self.di.create_account("teacher", "root", "instructor")
self.di.create_course("123", "test_course")
self.di.set_course_instructor("123", "teacher")
self.assertTrue(self.di.is_course_assigned("123"))
def test_get_course_not_assigned(self):
self.di.create_course("000", "test_course")
self.assertFalse(self.di.is_course_assigned("000"))
# endregion
# region Testing Lab variations
def test_create_lab_get_labs(self):
course_number = "361"
lab_number = "801"
self.di.create_course(course_number, "courseName")
self.di.create_lab(course_number, lab_number)
lab = Course.objects.get(number=course_number, labs__number=lab_number)
self.assertIsNotNone(lab)
def test_assign_lab(self):
course_number = "361"
lab_number = "801"
ta_name = "apoorv"
self.di.create_course(course_number, "courseName")
self.di.create_lab(course_number, lab_number)
self.di.create_account(ta_name, "pass", "TA")
self.di.set_lab_assignment(course_number, lab_number, ta_name)
lab = Course.objects.get(number=course_number).labs.get(
number=lab_number)
self.assertEqual(lab.ta.first().name, ta_name)
def test_get_lab_exists(self):
self.di.create_course("123", "test_course")
self.di.create_lab("123", "001")
self.assertTrue(self.di.lab_exists("123", "001"))
def test_get_lab_doesnt_exist(self):
self.di.create_course("123", "test_course")
self.assertFalse(self.di.lab_exists("123", "1231231"))
def test_get_lab_assigned(self):
self.di.create_account("ta", "pass", "TA")
self.di.create_course("123", "test_course")
self.di.create_lab("123", "001")
self.di.set_lab_assignment("123", "001", "ta")
self.assertTrue(self.di.is_lab_assigned("123", "001"))
def test_get_lab_not_assigned(self):
self.di.create_course("234", "test_course2")
self.di.create_lab("234", "000")
self.assertFalse(self.di.is_lab_assigned("234", "000"))
# endregion
# region Information variations
def test_basic_get_private_info(self):
user = TaCLI.User.User("admin", "master")
self.di.create_account("admin", "pass", "master")
self.assertEqual(
self.di.get_private_info(user), {
'username': '******',
'name': '',
'lname': '',
'email': '',
'address': '',
'phone': '',
'hours': '',
'role': 'master'
})
def test_edit_phone_get_private_info(self):
user = TaCLI.User.User("admin", "master")
self.di.create_account("admin", "pass", "master")
self.di.edit_phone("admin", "1414")
self.assertEqual(
self.di.get_private_info(user), {
'username': '******',
'name': '',
'lname': '',
'email': '',
'address': '',
'phone': '1414',
'hours': '',
'role': 'master'
})
def test_edit_name_get_private_info(self):
user = TaCLI.User.User("admin", "master")
self.di.create_account("admin", "pass", "master")
self.di.edit_name("admin", "first", "second")
self.assertEqual(
self.di.get_private_info(user), {
'username': '******',
'name': 'first',
'lname': 'second',
'email': '',
'address': '',
'phone': '',
'hours': '',
'role': 'master'
})
def test_edit_address_get_private_info(self):
user = TaCLI.User.User("admin", "master")
self.di.create_account("admin", "pass", "master")
self.di.edit_address("admin", "somewhere")
self.assertEqual(
self.di.get_private_info(user), {
'username': '******',
'name': '',
'lname': '',
'email': '',
'address': 'somewhere',
'phone': '',
'hours': '',
'role': 'master'
})
def test_edit_email_get_private_info(self):
user = TaCLI.User.User("admin", "master")
self.di.create_account("admin", "pass", "master")
self.di.edit_email("admin", "@uwm.edu")
self.assertEqual(
self.di.get_private_info(user), {
'username': '******',
'name': '',
'lname': '',
'email': '@uwm.edu',
'address': '',
'phone': '',
'hours': '',
'role': 'master'
})
def test_edit_office_hours_get_private_info(self):
user = TaCLI.User.User("admin", "master")
self.di.create_account("admin", "pass", "master")
self.di.edit_office_hours("admin", "Monday: 1:00PM-2:00PM")
self.assertEqual(
self.di.get_private_info(user), {
'username': '******',
'name': '',
'lname': '',
'email': '',
'address': '',
'phone': '',
'hours': 'MONDAY: 1:00PM-2:00PM, ',
'role': 'master'
})
def test_basic_get_public_info(self):
user = TaCLI.User.User("admin", "master")
self.di.create_account("admin", "pass", "master")
self.assertEqual(
self.di.get_public_info(user), {
'username': '******',
'name': '',
'lname': '',
'email': '',
'hours': '',
'role': 'master'
})
def test_edit_name_get_public_info(self):
user = TaCLI.User.User("admin", "master")
self.di.create_account("admin", "pass", "master")
self.di.edit_name("admin", "first", "second")
self.assertEqual(
self.di.get_public_info(user), {
'username': '******',
'name': 'first',
'lname': 'second',
'email': '',
'hours': '',
'role': 'master'
})
def test_edit_email_get_public_info(self):
user = TaCLI.User.User("admin", "master")
self.di.create_account("admin", "pass", "master")
self.di.edit_email("admin", "@uwm.edu")
self.assertEqual(
self.di.get_public_info(user), {
'username': '******',
'name': '',
'lname': '',
'email': '@uwm.edu',
'hours': '',
'role': 'master'
})
def test_edit_office_hours_get_public_info(self):
user = TaCLI.User.User("admin", "master")
self.di.create_account("admin", "pass", "master")
self.di.edit_office_hours("admin", "Monday: 1:00PM-2:00PM")
self.assertEqual(
self.di.get_public_info(user), {
'username': '******',
'name': '',
'lname': '',
'email': '',
'hours': 'MONDAY: 1:00PM-2:00PM ',
'role': 'master'
})
class ContactInfoTests(TestCase):
def setUp(self):
self.di = DjangoModelInterface()
self.environment = Environment.Environment(self.di, DEBUG=True)
self.ui = UI.UI(self.environment)
def test_command_view_private_info(self):
"""
Only public information is available for a normal user to see. Only those with permission can view
private information like address and phone number.
When view_info command is entered for a user you only have public access to, it expects one argument:
- username
The response is a string:
- If successful: "username role"
- If failure: "Username does not exist"
"""
# Create a TA account and log in as part of the setup
self.di.create_account("userTA", "userPassword", "TA")
self.ui.command("login", {
"username": "******",
"password": "******"
})
self.ui.command("edit_info", {
"field": "email",
"email": "erin [email protected]"
})
self.ui.command("edit_info", {"field": "phone", "phone": "1231231234"})
self.ui.command(
"edit_info", {
"field": "address",
"address": "Address: 2311 E Hartford Ave Milwaukee, WI 53211"
})
# view own contact info, expect success
self.assertEquals(
self.ui.command("view_info", ""), {
'username': '******',
'name': '',
'lname': '',
'email': 'erin [email protected]',
'address': 'Address: 2311 E Hartford Ave Milwaukee, WI 53211',
'phone': '1231231234',
'hours': '',
'role': 'TA'
})
def test_command_view_public_info(self):
self.di.create_account("userTA", "userPassword", "TA")
self.di.edit_email("userTA", "*****@*****.**")
self.di.create_account("userInstructor", "userPassword", "instructor")
self.ui.command("login", {
"username": "******",
"password": "******"
})
self.assertEquals(
self.ui.command("view_info", {"username": "******"}), {
'username': '******',
'name': '',
'lname': '',
'email': '*****@*****.**',
'hours': '',
'role': 'TA'
})
def test_command_view_public_info_fake(self):
self.di.create_account("userInstructor", "userPassword", "instructor")
self.ui.command("login", {
"username": "******",
"password": "******"
})
# Command: "view_info fake", expect success
self.assertEquals(self.ui.command("view_info", {"username": "******"}),
"ERROR")
def test_command_Instructor_change_contact_info_valid(self):
self.di.create_account("userInstructor", "password", "instructor")
self.ui.command("login", {
"username": "******",
"password": "******"
})
# Instructor changes own email, phone number, password; expected success
self.assertEquals(
self.ui.command("edit_info", {
"field": "email",
"email": "erin [email protected]"
}), "Email has been updated successfully.")
self.assertEquals(
self.ui.command("edit_info", {
"field": "phone",
"phone": "1231231234"
}), "Phone Number has been updated successfully")
self.assertEquals(
self.ui.command(
"edit_info", {
"field": "address",
"address":
"Address: 2311 E Hartford Ave Milwaukee, WI 53211"
}), "Address has been updated successfully.")
def test_command_TA_change_contact_info(self):
self.di.create_account("userTA", "password", "TA")
self.ui.command("login", {
"username": "******",
"password": "******"
})
# Instructor changes own email, phone number, password; expected success
self.assertEquals(
self.ui.command("edit_info", {
"field": "email",
"email": "erin [email protected]"
}), "Email has been updated successfully.")
self.assertEquals(
self.ui.command("edit_info", {
"field": "phone",
"phone": "1231231234"
}), "Phone Number has been updated successfully")
self.assertEquals(
self.ui.command(
"edit_info", {
"field": "address",
"address":
"Address: 2311 E Hartford Ave Milwaukee, WI 53211"
}), "Address has been updated successfully.")
def test_command_Instructor_change_contact_info(self):
self.di.create_account("userTA", "password", "Instructor")
self.ui.command("login", {
"username": "******",
"password": "******"
})
# Instructor changes own email, phone number, password; expected success
self.assertEquals(
self.ui.command("edit_info", {
"field": "email",
"email": "erin [email protected]"
}), "Email has been updated successfully.")
self.assertEquals(
self.ui.command("edit_info", {
"field": "phone",
"phone": "1231231234"
}), "Phone Number has been updated successfully")
self.assertEquals(
self.ui.command(
"edit_info", {
"field": "address",
"address":
"Address: 2311 E Hartford Ave Milwaukee, WI 53211"
}), "Address has been updated successfully.")
def test_command_Administrator_change_contact_info(self):
self.di.create_account("userTA", "password", "Administrator")
self.ui.command("login", {
"username": "******",
"password": "******"
})
# Instructor changes own email, phone number, password; expected success
self.assertEquals(
self.ui.command("edit_info", {
"field": "email",
"email": "erin [email protected]"
}), "Email has been updated successfully.")
self.assertEquals(
self.ui.command("edit_info", {
"field": "phone",
"phone": "1231231234"
}), "Phone Number has been updated successfully")
self.assertEquals(
self.ui.command(
"edit_info", {
"field": "address",
"address":
"Address: 2311 E Hartford Ave Milwaukee, WI 53211"
}), "Address has been updated successfully.")
def test_command_edit_contact_format(self):
self.di.create_account("userInstructor", "password", "instructor")
self.ui.command("login", {
"username": "******",
"password": "******"
})
# no argument given
self.assertEquals(self.ui.command("edit_info", {}), "ERROR")