add CRUD for organization member roles

Author: lbalmaceda

auth0/v3/management/organizations.py

@@ -22,14 +22,11 @@ def __init__(self, domain, token, telemetry=True, timeout=5.0):
         self.domain = domain
         self.client = RestClient(jwt=token, telemetry=telemetry, timeout=timeout)
 
-    def _url(self, id=None, path=None, secondary_id=None):
+    def _url(self, *args):
         url = 'https://{}/api/v2/organizations'.format(self.domain)
-        if id is not None:
-            url = '{}/{}'.format(url, id)
-        if path is not None:
-            url = '{}/{}'.format(url, path)
-            if secondary_id is not None:
-                url = '{}/{}'.format(url, secondary_id)
+        for p in args:
+            if p is not None:
+                url = '{}/{}'.format(url, p)
         return url
 
     # Organizations
@@ -221,4 +218,57 @@ def delete_organization_members(self, id, body):
         See: https://auth0.com/docs/api/management/v2#!/Clients/delete_clients_by_id
         """
 
-        return self.client.delete(self._url(id, 'members'), data=body)
+        return self.client.delete(self._url(id, 'members'), data=body)
+
+    # Organization Member Roles
+    def all_organization_member_roles(self, id, user_id, page=None, per_page=None):
+        """Retrieves a list of all the roles from the given organization member.
+
+        Args:
+           id (str): the ID of the organization.
+           
+           user_id (str): the ID of the user member of the organization.
+
+           page (int): The result's page number (zero based). When not set,
+              the default value is up to the server.
+
+           per_page (int, optional): The amount of entries per page. When not set,
+              the default value is up to the server.
+
+        See: https://auth0.com/docs/api/management/v2#!/Clients/get_clients
+        """
+        params = {}
+        params['page'] = page
+        params['per_page'] = per_page
+
+        return self.client.get(self._url(id, 'members', user_id, 'roles'), params=params)
+
+    def create_organization_member_roles(self, id, user_id, body):
+        """Adds roles to a member of an organization.
+
+        Args:
+           id (str): the ID of the organization.
+
+           user_id (str): the ID of the user member of the organization.
+
+           body (dict): Attributes from the members to add.
+
+        See: https://auth0.com/docs/api/v2#!/Clients/post_clients
+        """
+
+        return self.client.post(self._url(id, 'members', user_id, 'roles'), data=body)
+
+    def delete_organization_member_roles(self, id, user_id, body):
+        """Deletes roles from a member of an organization.
+
+        Args:
+           id (str): Id of organization.
+
+           user_id (str): the ID of the user member of the organization.
+
+           body (dict): Attributes from the members to delete
+
+        See: https://auth0.com/docs/api/management/v2#!/Clients/delete_clients_by_id
+        """
+
+        return self.client.delete(self._url(id, 'members', user_id, 'roles'), data=body)

auth0/v3/test/management/test_organizations.py

@@ -155,7 +155,7 @@ def test_delete_organization_connection(self, mock_rc):
             'https://domain/api/v2/organizations/test-org/enabled_connections/test-con'
         )
 
-    # Organization Connections
+    # Organization Members
     @mock.patch('auth0.v3.management.organizations.RestClient')
     def test_all_organization_members(self, mock_rc):
         mock_instance = mock_rc.return_value
@@ -202,4 +202,53 @@ def test_delete_organization_members(self, mock_rc):
         mock_instance.delete.assert_called_with(
             'https://domain/api/v2/organizations/test-org/members',
             data={'a': 'b', 'c': 'd'}
+        )
+
+    # Organization Member Roles
+    @mock.patch('auth0.v3.management.organizations.RestClient')
+    def test_all_organization_member_roles(self, mock_rc):
+        mock_instance = mock_rc.return_value
+
+        c = Organizations(domain='domain', token='jwttoken')
+
+        # Default parameters are requested
+        c.all_organization_member_roles('test-org', 'test-user')
+
+        args, kwargs = mock_instance.get.call_args
+
+        self.assertEqual('https://domain/api/v2/organizations/test-org/members/test-user/roles', args[0])
+        self.assertEqual(kwargs['params'], {'page': None,
+                                            'per_page': None})
+
+        # Specific pagination
+        c.all_organization_member_roles('test-org', 'test-user', page=7, per_page=25)
+
+        args, kwargs = mock_instance.get.call_args
+
+        self.assertEqual('https://domain/api/v2/organizations/test-org/members/test-user/roles', args[0])
+        self.assertEqual(kwargs['params'], {'page': 7,
+                                            'per_page': 25})
+
+    @mock.patch('auth0.v3.management.organizations.RestClient')
+    def test_create_organization_member_roles(self, mock_rc):
+        mock_instance = mock_rc.return_value
+
+        c = Organizations(domain='domain', token='jwttoken')
+        c.create_organization_member_roles('test-org', 'test-user', {'a': 'b', 'c': 'd'})
+
+        mock_instance.post.assert_called_with(
+            'https://domain/api/v2/organizations/test-org/members/test-user/roles',
+            data={'a': 'b', 'c': 'd'}
+        )
+
+    @mock.patch('auth0.v3.management.organizations.RestClient')
+    def test_delete_organization_member_roles(self, mock_rc):
+        mock_instance = mock_rc.return_value
+
+        c = Organizations(domain='domain', token='jwttoken')
+        c.delete_organization_member_roles('test-org', 'test-user', {'a': 'b', 'c': 'd'})
+
+        mock_instance.delete.assert_called_with(
+            'https://domain/api/v2/organizations/test-org/members/test-user/roles',
+            data={'a': 'b', 'c': 'd'}
         )