def test_message_model(self):
"""Does basic model work?"""
m = Message(text="Test", user_id=1)
db.session.add(m)
db.session.commit()
# User should have no messages & no followers
self.assertEqual(m.text, "Test")
self.assertEqual(m.user_id, 1)
self.assertEqual(len(m.user.followers), 0)
self.assertEqual(len(m.user.messages), 1)
# Test __repr__
self.assertEqual(m.__repr__(), f"<Message text={m.text}>")
def test_message_repr(self):
"""Does the repr method in the Message model work as expected?"""
u = User(email="*****@*****.**",
username="******",
password="******")
db.session.add(u)
db.session.commit()
message = Message(text="This is a test.", user_id=u.id)
db.session.add(message)
db.session.commit()
self.assertEqual(
message.__repr__(),
f"<Message #{message.id}: {message.user_id}, {message.timestamp}, {message.text}>"
)
