def test_following(self):
users = make_dummy_users()
users[0].follow(users[1])
users[0].follow(users[2])
users[0].follow(users[3])
users[1].follow(users[2])
users[1].follow(users[3])
users[2].follow(users[3])
self.assertIn(users[1], users[0].following.all())
self.assertIn(users[2], users[0].following.all())
self.assertIn(users[3], users[0].following.all())
self.assertIn(users[2], users[1].following.all())
self.assertIn(users[3], users[1].following.all())
self.assertIn(users[3], users[2].following.all())
# 반대로 follower를 가진 유저는 follower_set 역참조 이름으로
self.assertIn(users[0], users[3].follower_set.all())
self.assertIn(users[1], users[3].follower_set.all())
self.assertIn(users[2], users[3].follower_set.all())
self.assertIn(users[0], users[2].follower_set.all())
self.assertIn(users[1], users[2].follower_set.all())
self.assertIn(users[0], users[1].follower_set.all())
def test_following_duplicate(self):
users = make_dummy_users()
users[0].follow(users[1])
with self.assertRaises(IntegrityError):
users[0].follow(users[1])
self.assertEqual(users[0].following.count(), 1)
self.assertEqual(users[1].follower_set.count(), 1)
def test_unfollow(self):
users = make_dummy_users()
users[0].follow(users[1])
users[0].follow(users[2])
users[0].follow(users[3])
# 이후 해당 유저의 following HTML필드에 팔로우한 유저들이 있는지 확인
users[0].unfollow(users[1])
self.assertNotIn(users[0], users[1].following.all())
users[0].unfollow(users[2])
self.assertNotIn(users[0], users[2].following.all())
users[0].unfollow(users[3])
self.assertNotIn(users[0], users[3].following.all())
def test_following_duplicate(self):
users = make_dummy_users()
# 중복 팔로우 시도
users[0].follow(users[1])
# 중복 팔로우 시도 IntegrityError 발생과 함께 실패함을 확인
# 클래스가 TransactionTestCase 를 상속받아야함
with self.assertRaises(IntegrityError):
users[0].follow(users[1])
# 중복으로 팔로우했어도 실제로는 하나만 남아있어야함
self.assertEqual(users[0].following.count(), 1)
self.assertEqual(users[1].follower_set.count(), 1)
def test_profile_display_status_following_count(self):
user = self.make_user_and_login()
users = make_dummy_users()
num = random.randrange(1, 10)
for i in range(num):
user.follow(users[i])
self.access_profile_page_and_check_status(
status_name='팔로우',
num=num,
unit='명',
)
def test_unfollow(self):
users = make_dummy_users()
users[0].follow(users[1])
users[0].follow(users[2])
users[0].follow(users[3])
users[0].unfollow(users[1])
self.assertEqual(users[0].following.count(), 2)
users[0].unfollow(users[2])
self.assertEqual(users[0].following.count(), 1)
users[0].unfollow(users[3])
self.assertEqual(users[0].following.count(), 0)
def test_following(self):
"""
dummy 유저를 여러명 만들어서 서로 팔로우 팔로잉 관계를 갖도록 테스트
:return:
"""
users = make_dummy_users()
# 임의의 유저가 다른 유저를 Follow
# 0은 123을 팔로우
users[0].follow(users[1])
users[0].follow(users[2])
users[0].follow(users[3])
# 1은 23을 팔로우
users[1].follow(users[2])
users[1].follow(users[3])
# 2는 3만 팔로우
users[2].follow(users[3])
# 이후 해당 유저의 following HTML필드에 팔로우한 유저들이 있는지 확인
self.assertIn(users[1], users[0].following.all())
self.assertIn(users[2], users[0].following.all())
self.assertIn(users[3], users[0].following.all())
self.assertIn(users[2], users[1].following.all())
self.assertIn(users[3], users[1].following.all())
self.assertIn(users[3], users[2].following.all())
# 반대로 follower를 가진 유저는 follower_set역참조 이름으로
# 자신을 follow하는 유저를 출력하는지 검사
self.assertIn(users[0], users[3].follower_set.all())
self.assertIn(users[1], users[3].follower_set.all())
self.assertIn(users[2], users[3].follower_set.all())
self.assertIn(users[0], users[2].follower_set.all())
self.assertIn(users[1], users[2].follower_set.all())
self.assertIn(users[0], users[1].follower_set.all())