|
@@ -186,8 +186,8 @@ class UserModelTestCase(unittest.TestCase):
|
186
|
186
|
self.assertTrue(u1.is_following(u2))
|
187
|
187
|
self.assertFalse(u1.is_followed_by(u2))
|
188
|
188
|
self.assertTrue(u2.is_followed_by(u1))
|
189
|
|
- self.assertTrue(u1.followed.count() == 1)
|
190
|
|
- self.assertTrue(u2.followers.count() == 1)
|
|
189
|
+ self.assertTrue(u1.followed.count() == 2)
|
|
190
|
+ self.assertTrue(u2.followers.count() == 2)
|
191
|
191
|
f = u1.followed.all()[-1]
|
192
|
192
|
self.assertTrue(f.followed == u2)
|
193
|
193
|
self.assertTrue(timestamp_before <= f.timestamp <= timestamp_after)
|
|
@@ -196,13 +196,13 @@ class UserModelTestCase(unittest.TestCase):
|
196
|
196
|
u1.unfollow(u2)
|
197
|
197
|
db.session.add(u1)
|
198
|
198
|
db.session.commit()
|
199
|
|
- self.assertTrue(u1.followed.count() == 0)
|
200
|
|
- self.assertTrue(u2.followers.count() == 0)
|
201
|
|
- self.assertTrue(Follow.query.count() == 0)
|
|
199
|
+ self.assertTrue(u1.followed.count() == 1)
|
|
200
|
+ self.assertTrue(u2.followers.count() == 1)
|
|
201
|
+ self.assertTrue(Follow.query.count() == 2)
|
202
|
202
|
u2.follow(u1)
|
203
|
203
|
db.session.add(u1)
|
204
|
204
|
db.session.add(u2)
|
205
|
205
|
db.session.commit()
|
206
|
206
|
db.session.delete(u2)
|
207
|
207
|
db.session.commit()
|
208
|
|
- self.assertTrue(Follow.query.count() == 0)
|
|
208
|
+ self.assertTrue(Follow.query.count() == 1)
|