class TestViews(unittest.TestCase):
def setUp(self):
self.jacob = User.objects.create_user('jacob', '*****@*****.**',
'JKM')
self.david = Member.objects.create(name='david')
self.member_list_view = MemberListView(queryset = Member.objects.all())
def testMemberDetailsContext(self):
# test anonymous user
rf = RequestFactory()
anon = AnonymousUser()
request = rf.get('/')
request.user = anon
context = self.member_list_view.get_object_context(request, self.david.id)
self.assertFalse(context['watched_member'])
# test autherized user
request.user = self.jacob
context = self.member_list_view.get_object_context(request, self.david.id)
self.assertFalse(context['watched_member'])
# test autherized user that follows
p = self.jacob.get_profile()
p.followed_members.add(self.david)
context = self.member_list_view.get_object_context(request, self.david.id)
self.assertTrue(context['watched_member'])
def tearDown(self):
self.jacob.delete()
self.david.delete()
def setUp(self):
self.jacob = User.objects.create_user('jacob', '*****@*****.**',
'JKM')
self.david = Member.objects.create(name='david')
self.member_list_view = MemberListView(queryset = Member.objects.all())
