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