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)