def test_show_likes(self):
"""Does messages the user liked show up?"""
u3 = User.signup("testuser3", "*****@*****.**", "password3", None)
u3.id = 4545
u4 = User.signup("abcdef", "*****@*****.**", "password4", None)
u4.id = 5656
db.session.add_all([u3, u4])
db.session.commit()
m1 = Message(text='message one for testuser1', user_id=1212)
m1.id = 111
m2 = Message(text='message one for testuser2', user_id=2323)
m2.id = 222
m3 = Message(text='message two for testuser2', user_id=2323)
m3.id = 333
m4 = Message(text='message one for testuser3', user_id=4545)
m4.id = 444
m5 = Message(text='message one for abcdef', user_id=5656)
m5.id = 555
db.session.add_all([m1, m2, m3, m4, m5])
db.session.commit()
l1 = Likes(user_id=1212, message_id=222)
l1.id = 12
l2 = Likes(user_id=1212, message_id=444)
l2.id = 34
l3 = Likes(user_id=1212, message_id=555)
l3.id = 45
db.session.add_all([l1, l2, l3])
db.session.commit()
with self.client as c:
with c.session_transaction() as sess:
sess[CURR_USER_KEY] = self.u1.id
resp = c.get(f"/users/{self.u1.id}/likes")
html = resp.get_data(as_text=True)
self.assertEqual(resp.status_code, 200)
self.assertIn('message one for testuser2', html)
self.assertIn('message one for testuser3', html)
self.assertIn('message one for abcdef', html)
self.assertNotIn('message one for testuser1', html)
self.assertNotIn('message two for testuser2', html)
def set_up_likes(self):
"""set up likes"""
like = Likes(user_id=self.user_id, message_id=self.message_id)
like.id = 4000
self.like_id = like.id
db.session.add(like)
db.session.commit()
def setUp(self):
"""Create test client, add sample data."""
db.drop_all()
db.create_all()
user = User.signup(username="******",
email="*****@*****.**",
password="******",
image_url=None)
user.id = 1000
self.user_id = user.id
db.session.commit()
message = Message(text="Test Message", user_id=1000)
message.id = 2000
self.message_id = message.id
db.session.add(message)
db.session.commit()
self.user1 = User.query.get(self.user_id)
like = Likes(user_id=self.user1.id, message_id=self.message_id)
like.id = 2000
self.like_id = like.id
db.session.add(like)
db.session.commit()
self.client = app.test_client()
