Parameters added to the endpoints of users

lccruz · lccruz · commit 9e78639f49e8 · 2018-02-21T14:53:55.000-03:00
diff --git a/HISTORY.md b/HISTORY.md
@@ -1,5 +1,9 @@
 # History
 
+## 0.3.4 (2018-02-21)
+
+* Parameters added to the endpoints of users [lccruz]
+
 ## 0.3.3 (2017-12-04)
 
 * Parameters added to the endpoints of categories [lccruz]
diff --git a/python_wpapi/python_wpapi.py b/python_wpapi/python_wpapi.py
@@ -85,8 +85,8 @@ def delete_media(self, id, force=False):
         endpoint = '{}/media/{}'.format(self.base_url, id)
         return self._request(endpoint, method='DELETE', force=force)
 
-    def get_users(self):
-        endpoint = '{}/users'.format(self.base_url)
+    def get_users(self, parameters='?per_page=10'):
+        endpoint = '{}/users/{}'.format(self.base_url, parameters)
         return self._request(endpoint)
 
     def get_user(self, id, context='view'):
diff --git a/setup.cfg b/setup.cfg
@@ -1,5 +1,5 @@
 [bumpversion]
-current_version = 0.3.3
+current_version = 0.3.4
 commit = True
 tag = True
 
diff --git a/setup.py b/setup.py
@@ -19,7 +19,7 @@
 
 setup(
     name='python_wpapi',
-    version='0.3.3',
+    version='0.3.4',
     description="Simple wrapper around the Wordpress REST API",
     long_description=readme + '\n\n' + history,
     author="Lucas Lobosque",
diff --git a/tests/test_python_wpapi.py b/tests/test_python_wpapi.py
@@ -129,7 +129,16 @@ def test_delete_media(mock, api):
 @patch.object(python_wpapi.WpApi, '_request')
 def test_get_users(mock, api):
     api.get_users()
-    mock.assert_called_with('http://base.url/wp-json/wp/v2/users')
+    mock.assert_called_with('http://base.url/wp-json/wp/v2/users/?per_page=10')
+
+@patch.object(python_wpapi.WpApi, '_request')
+def test_get_users_with_20_items(mock, api):
+    users_len = 20
+    mock.return_value = range(users_len)
+    users = api.get_users('?per_page={}'.format(users_len))
+    url_result = 'http://base.url/wp-json/wp/v2/users/?per_page={}'.format(users_len)
+    mock.assert_called_with(url_result)
+    assert len(users) == users_len
 
 @patch.object(python_wpapi.WpApi, '_request')
 def test_get_user(mock, api):
