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()