def test_stop_following(self): """Does user you want to stop following not appear?""" u3 = User.signup("abcdef", "*****@*****.**", "password4", None) u3.id = 5656 db.session.add(u3) db.session.commit() f1 = Follows(user_being_followed_id=u3.id, user_following_id=self.u1.id) f1.id = 11 f2 = Follows(user_being_followed_id=self.u2.id, user_following_id=self.u1.id) f2.id = 22 db.session.add_all([f1, f2]) db.session.commit() with self.client as c: with c.session_transaction() as sess: sess[CURR_USER_KEY] = self.u1.id u = User.query.get(2323) resp = c.post(f"/users/stop-following/{u.id}", follow_redirects=True) html = resp.get_data(as_text=True) self.assertEqual(resp.status_code, 200) self.assertIn('@abcdef', html) self.assertNotIn('@testuser2', html)
def setup_follows(self): 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() f1 = Follows(user_being_followed_id=u3.id, user_following_id=self.u1.id) f1.id = 11 f2 = Follows(user_being_followed_id=u4.id, user_following_id=self.u1.id) f2.id = 22 f3 = Follows(user_being_followed_id=self.u1.id, user_following_id=self.u2.id) f3.id = 33 f4 = Follows(user_being_followed_id=self.u1.id, user_following_id=u4.id) f4.id = 44 db.session.add_all([f1, f2, f3, f4]) db.session.commit()