def test_delete_user(self): UserController.add_user(self.user_1) self.assertEqual(UserController.delete_user('nickname_1', 'email_1'), 'SUCCESS') self.assertEqual( UserController.delete_user('fake_nickname_1', 'fake_email_1'), 'SUCCESS')
def test_retrieve_hit(): user_one = UserModel('0', 'chengyun yu', 'yuche', 'male', '*****@*****.**', '0,0', 'sports, dining', 'i am a sports guy who likes to eat. ' ) user_two = UserModel('0', 'chengyun yu', 'yuche', 'male', '*****@*****.**', '0,0', 'sports, dining', 'this is my new description. ' ) user_three = UserModel('0', 'non existent', 'noext', 'male', '*****@*****.**', '0.0', 'sports, dining', 'i am a sports guy who does not exist. ' ) print("Insert New: " + UserController.add_user(user_one)) print("Insert Conflict: " + UserController.add_user(user_two)) print("Update: " + UserController.edit_user(user_two)) print("Retrieve Hit: " + UserController.retrieve_user(UserFields.email.name, user_two.email)) print("Retrieve Miss: " + UserController.retrieve_user(UserFields.email.name, user_three.email)) return 'Tests Finished'
def test_retrieve_user(self): UserController.delete_user('nickname_1', 'email_1') UserController.delete_user('nickname_2', 'email_2') UserController.add_user(self.user_1) self.assertIsInstance(UserController.retrieve_user('email', 'email_1'), UserModel) self.assertEqual(UserController.retrieve_user('email', 'email_2'), 'MISSING') UserController.delete_user('nickname_1', 'email_1')
def test_host_event(self): UserController.add_user(self.user_1) user_2 = UserController.retrieve_user('email', 'email_1') event_id = EventController.add_event(self.event_1) self.event_1.eventid = event_id self.assertIsNotNone(EventController.host_event(user_2, self.event_1)) UserController.delete_user('nickname_1', 'email_1')
def test_edit_user_1(self): UserController.delete_user('nickname_3', 'email_2') UserController.delete_user('nickname_2', 'email_2') UserController.add_user(self.user_2) user_3 = UserController.retrieve_user('email', 'email_2') user_3.nickname = 'nickname_3' self.assertIsNot(UserController.edit_user(user_3), 'MISSING' or 'DUPLICATE' or 'FAILURE') self.assertEqual(UserController.edit_user(user_3), 'MISSING') self.assertEqual( UserController.retrieve_user('email', 'email_2').nickname, 'nickname_3') UserController.delete_user('nickname_3', 'email_2')
def register(data): user = UserModel(0, data['name'], data['nickname'], data['gender'], data['email'], data['location'], stringToEnum(data['tags']), data['description'], [], [], data['image'], data['google_id']) result = UserController.add_user(user) if result == Errors.DUPLICATE.name: return result, None elif result == Errors.FAILURE.name: return result, None else: user = UserController.retrieve_user(UserFields.googleid.name, data['google_id']) data = { 'id': user.uid, 'name': user.name, 'nickname': user.nickname, 'email': user.email, 'location': user.location, 'description': user.description, 'gender': user.gender, 'image': user.image, 'tags': user.tags, 'google_id': user.google_id } return 'OK', data
def register(): global current_user current_user = read_user() result = UserController.add_user(current_user) if result == Errors.DUPLICATE.name: add_output("A user with the same credentials already exists! \n") current_user = None elif result == Errors.FAILURE.name: return_failure() current_user = None else: add_output("User registered JoinMe with email " + current_user.email + ". \n") current_user = UserController.retrieve_user(UserFields.email.name, current_user.email) UserController.print_user(current_user) return
def test_add_user(self): UserController.delete_user('nickname_1', 'email_1') self.assertEqual(UserController.add_user(self.user_1), 'SUCCESS') self.assertEqual(UserController.add_user(self.user_1), 'DUPLICATE') UserController.delete_user('nickname_1', 'email_1')