set current user for user directory

gitcoinco · Jun 12, 2019 · fce4ea1 · fce4ea1
1 parent 3fd1a63
commit fce4ea1
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/app/dashboard/tests/test_users_list.py b/app/dashboard/tests/test_users_list.py
@@ -33,11 +33,18 @@ class UsersListTest(TestCase):
 
     def setUp(self):
         self.request = RequestFactory()
+        self.current_user = User.objects.create(
+            password="asdfasdf", username="asdfasdf")
+        current_user_profile = Profile.objects.create(
+            user=self.current_user, data={}, hide_profile=False, handle="asdfasdf")
+
         for i in range(20):
             user = User.objects.create(password="{}".format(i),
                                        username="user{}".format(i))
             profile = Profile.objects.create(
                 user=user, data={}, hide_profile=False, handle="{}".format(i))
 
     def test_user_list(self):
-        assert json.loads(users_fetch(self.request.get('/api/v0.1/users_fetch/')).content)['count'] == 20
+        request = self.request
+        request.user = self.current_user
+        assert json.loads(users_fetch(request.get('/api/v0.1/users_fetch/')).content)['count'] == 21