def ems():
ems1 = EMS()
staff = Staff(createUser('z5135009', 'hahaha', 'name5135009', '*****@*****.**'))
presenter = Guest(createUser(None, 'samplepass', 'Vicky', '*****@*****.**'))
ems1.addUser(staff)
ems1.addUser(presenter)
return ems1
def test_guest_registration_success():
ems1 = EMS()
user1 = Guest(createUser(None, 'hahaha', 'testName23',
'*****@*****.**'))
ems1.addUser(user1)
assert (ems1.getUserByEmail('*****@*****.**').name == 'testName23')
registerEMS(None)
def test_dumpAndLoad2():
ems1 = EMS(userCSV='user.csv')
user1 = Guest(
createUser('z5135009', 'hahaha', 'name5135009', '*****@*****.**'))
ems1.addUser(user1)
ems1.dumpData(file='user.data')
ems2 = EMS(binFile='user.data')
assert (ems2.getUserByEmail('*****@*****.**').password == 'hahaha')
assert (ems2.getUserByEmail('*****@*****.**').password == 'pass24064')
registerEMS(None)
def test_guest_registration_email_exist():
ems1 = EMS()
user1 = Guest(createUser(None, 'hahaha1', 'hahah', '*****@*****.**'))
ems1.addUser(user1)
with pytest.raises(UserDataException) as exc_info:
user2 = Guest(createUser(None, 'hahaha', 'hahah', '*****@*****.**'))
ems1.addUser(user2)
assert (exc_info.type == UserDataException)
assert (exc_info.value.fieldname == 'email')
assert (str(exc_info.value) == 'Email already exists')
registerEMS(None)
def test_dumpAndLoad3():
ems1 = EMS(userCSV='user.csv')
user1 = Guest(
createUser('z5135009', 'hahaha', 'name5135009', '*****@*****.**'))
ems1.addUser(user1)
ems1.dumpData(file='test-user.data')
ems2 = EMS(binFile='test-user.data')
os.remove('test-user.data')
assert (
ems2.getUserByEmail('*****@*****.**').email == '*****@*****.**')
assert (
ems2.getUserByEmail('*****@*****.**').email == '*****@*****.**')
registerEMS(None)
