def test_contructor_existing_message(self): m = chatter_classes.Message(1, db) self.assertEqual( m.content, "This is the first message in TestRoom1, sent by TestUser1. It has two attachments (a picture of Donald Trump and another of Gary Barlow)." ) self.assertEqual(m.chatroom.name, chatter_classes.Chatroom(1, db).name) self.assertEqual(m.sender.username, chatter_classes.User(1, db).username) self.assertEqual(m.timestamp.year, 2021)
def test_delete_user_by_admin(self): u = chatter_classes.User(5, db) # As well as deleting the user, we need to ensure that their messages have been reassigned users_messages = chatter_classes.Message.get_messages_for_user( 5, None, db) users_message_ids = [message.messageid for message in users_messages] u.delete(chatter_classes.User(6, db)) # User 6 is TestAdmin for message_id in users_message_ids: m = chatter_classes.Message(message_id, db) self.assertEqual( m.senderid, 0) # assert message's sender Id updated to DeletedUser 0 # Check it's no longer possible to instantiate the deleted user self.assertRaises(chatter_classes.UserNotFoundError, chatter_classes.User, 5, db)
def test_constructor_existing_attachment(self): a = chatter_classes.Attachment(1, db) self.assertEqual(a.filepath, "donald.png") self.assertEqual(a.message.messageid, chatter_classes.Message(1, db).messageid)
def test_message_json(self): m = chatter_classes.Message(1, db) js = m.json print(js) self.assertNotEqual(0, len(js))
def test_chatroom(self): m = chatter_classes.Message(7, db) self.assertEqual(2, m.chatroom.chatroomid)
def test_sender(self): m = chatter_classes.Message(1, db) self.assertEqual(1, m.sender.userid)