def messages_destroy(message_id):
"""Delete a message."""
Message.delete_message(message_id)
db.session.commit()
return redirect(f"/users/{g.user.id}")
def test_message_model(self):
"""Create a user with a message to test"""
hashed_pwd1 = bcrypt.generate_password_hash("HASHED_PASSWORD1").decode(
'UTF-8')
user1 = User(email="*****@*****.**",
username="******",
password=hashed_pwd1)
db.session.add(user1)
db.session.commit()
message1 = Message(text="TestMessage", user_id=user1.id)
db.session.add(message1)
db.session.commit()
"""Test get_message_by_id"""
testmessage = Message.get_message_by_id(message1.id)
self.assertEqual(user1.id, testmessage.user_id)
"""Test get_filtered_messages"""
filtered_List = [user1.id]
testmessage = Message.get_filtered_messages(filtered_List)
self.assertEqual(user1.id, testmessage[0].user_id)
"""Test get_liked_messages"""
hashed_pwd2 = bcrypt.generate_password_hash("HASHED_PASSWORD2").decode(
'UTF-8')
user2 = User(email="*****@*****.**",
username="******",
password=hashed_pwd2)
db.session.add(user1)
db.session.commit()
user2.likes.append(message1.id)
liked = [message1.id]
testmessage = Message.get_liked_messages(liked)
self.assertEqual(message1.id, testmessage[0].id)
"""Test delete_message"""
before = db.session.query(Message).count()
Message.delete_message(message1.id)
after = db.session.query(Message).count()
self.assertEqual(before - 1, after)
