blob: f583c836525137f4f7c14d00a17b32c8d75189fe (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
|
from django.conf import settings as django_settings
from askbot.tests.utils import AskbotTestCase
if 'followit' in django_settings.INSTALLED_APPS:
TEST_PROTOTYPE = AskbotTestCase
else:
TEST_PROTOTYPE = object
class FollowUserTests(TEST_PROTOTYPE):
def setUp(self):
self.u1 = self.create_user('user1')
self.u2 = self.create_user('user2')
self.u3 = self.create_user('user3')
def test_multiple_follow(self):
self.u1.follow_user(self.u2)
self.u1.follow_user(self.u3)
self.u2.follow_user(self.u1)
self.assertEquals(
set(self.u1.get_followers()),
set([self.u2])
)
self.assertEquals(
set(self.u2.get_followers()),
set([self.u1])
)
self.assertEquals(
set(self.u1.get_followed_users()),
set([self.u2, self.u3])
)
def test_unfollow(self):
self.u1.follow_user(self.u2)
self.u1.unfollow_user(self.u2)
self.assertEquals(self.u1.get_followed_users().count(), 0)
|