Skip to content

Commit cd45832

Browse files
authored
Support follow count in user get (#104)
Add query param support in user get for follow counts.
1 parent 171b3e4 commit cd45832

File tree

2 files changed

+9
-2
lines changed

2 files changed

+9
-2
lines changed

stream/tests/test_client.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1377,6 +1377,13 @@ def test_user_get(self):
13771377
self.assertTrue("updated_at" in user)
13781378
self.assertTrue("id" in user)
13791379

1380+
def test_user_get_with_follow_counts(self):
1381+
response = self.c.users.add(str(uuid1()))
1382+
user = self.c.users.get(response["id"], with_follow_counts=True)
1383+
self.assertEqual(user["id"], response["id"])
1384+
self.assertTrue("followers_count" in user)
1385+
self.assertTrue("following_count" in user)
1386+
13801387
def test_user_update(self):
13811388
response = self.c.users.add(str(uuid1()))
13821389
self.c.users.update(response["id"], {"changed": True})

stream/users.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -19,9 +19,9 @@ def add(self, user_id, data=None, get_or_create=False):
1919
params={"get_or_create": get_or_create},
2020
)
2121

22-
def get(self, user_id):
22+
def get(self, user_id, **params):
2323
return self.client.get(
24-
"user/%s" % user_id, service_name="api", signature=self.token
24+
"user/%s" % user_id, service_name="api", params=params, signature=self.token
2525
)
2626

2727
def update(self, user_id, data=None):

0 commit comments

Comments
 (0)