update integrations endpoints to v3

duosecurity · Feb 11, 2025 · 0e70b9d · 0e70b9d
1 parent f8a16b9
commit 0e70b9d
Show file tree

Hide file tree

Showing 3 changed files with 25 additions and 17 deletions.
diff --git a/duo_client/admin.py b/duo_client/admin.py
@@ -2529,7 +2529,7 @@ def get_integrations_generator(self):
         """
         return self.json_paging_api_call(
             'GET',
-            '/admin/v2/integrations',
+            '/admin/v3/integrations',
             {},
         )
 
@@ -2549,7 +2549,7 @@ def get_integrations(self, limit=None, offset=0):
         if limit:
             return self.json_api_call(
                 'GET',
-                '/admin/v2/integrations',
+                '/admin/v3/integrations',
                 {'limit': limit, 'offset': offset},
             )
 
@@ -2568,7 +2568,7 @@ def get_integration(self, integration_key):
         params = {}
         response = self.json_api_call(
             'GET',
-            '/admin/v2/integrations/' + integration_key,
+            '/admin/v3/integrations/' + integration_key,
             params,
         )
         return response
@@ -2593,7 +2593,8 @@ def create_integration(self,
                            ip_whitelist_enroll_policy=None,
                            groups_allowed=None,
                            self_service_allowed=None,
-                           sso=None):
+                           sso=None,
+                           user_access=None):
         """Creates a new integration.
 
         name - The name of the integration (required)
@@ -2671,8 +2672,11 @@ def create_integration(self,
             params['self_service_allowed'] = '1' if self_service_allowed else '0'
         if sso is not None:
             params['sso'] = sso
+        if user_access is not None:
+            params['user_access'] = user_access
+
         response = self.json_api_call('POST',
-                                      '/admin/v2/integrations',
+                                      '/admin/v3/integrations',
                                       params,
         )
         return response
@@ -2766,7 +2770,7 @@ def delete_integration(self, integration_key):
 
         """
         integration_key = urllib.parse.quote_plus(str(integration_key))
-        path = '/admin/v2/integrations/%s' % integration_key
+        path = '/admin/v3/integrations/%s' % integration_key
         return self.json_api_call(
             'DELETE',
             path,
@@ -2794,7 +2798,9 @@ def update_integration(self,
                            ip_whitelist_enroll_policy=None,
                            groups_allowed=None,
                            self_service_allowed=None,
-                           sso=None):
+                           sso=None,
+                           user_access=None
+                           ):
         """Updates an integration.
 
         integration_key - The key of the integration to update. (required)
@@ -2833,7 +2839,7 @@ def update_integration(self,
 
         """
         integration_key = urllib.parse.quote_plus(str(integration_key))
-        path = '/admin/v2/integrations/%s' % integration_key
+        path = '/admin/v3/integrations/%s' % integration_key
         params = {}
         if name is not None:
             params['name'] = name
@@ -2877,6 +2883,8 @@ def update_integration(self,
             params['self_service_allowed'] = '1' if self_service_allowed else '0'
         if sso is not None:
             params['sso'] = sso
+        if user_access is not None:
+            params['user_access'] = user_access
 
         if not params:
             raise TypeError("No new values were provided")

diff --git a/tests/admin/test_integration.py b/tests/admin/test_integration.py
@@ -16,15 +16,15 @@ def test_get_integration(self):
         (uri, args) = response['uri'].split('?')
 
         self.assertEqual(response['method'], 'GET')
-        self.assertEqual(uri, '/admin/v2/integrations/{}'.format(self.integration_key))
+        self.assertEqual(uri, '/admin/v3/integrations/{}'.format(self.integration_key))
         self.assertEqual(util.params_to_dict(args), {'account_id': [self.client.account_id]})
 
     def test_delete_integration(self):
         response = self.client.delete_integration(self.integration_key)
         (uri, args) = response['uri'].split('?')
 
         self.assertEqual(response['method'], 'DELETE')
-        self.assertEqual(uri, '/admin/v2/integrations/{}'.format(self.integration_key))
+        self.assertEqual(uri, '/admin/v3/integrations/{}'.format(self.integration_key))
         self.assertEqual(util.params_to_dict(args), {'account_id': [self.client.account_id]})
 
     def test_create_integration(self):
@@ -38,7 +38,7 @@ def test_create_integration(self):
         )
 
         self.assertEqual(response['method'], 'POST')
-        self.assertEqual(response['uri'], '/admin/v2/integrations')
+        self.assertEqual(response['uri'], '/admin/v3/integrations')
         self.assertEqual(json.loads(response['body']),
             {
                 "account_id": self.client.account_id,
@@ -63,7 +63,7 @@ def test_update_integration_success(self):
         )
 
         self.assertEqual(response['method'], 'POST')
-        self.assertEqual(response['uri'], '/admin/v2/integrations/{}'.format(self.integration_key))
+        self.assertEqual(response['uri'], '/admin/v3/integrations/{}'.format(self.integration_key))
         self.assertEqual(json.loads(response['body']),
             {
                 "account_id": self.client.account_id,

diff --git a/tests/admin/test_integrations.py b/tests/admin/test_integrations.py
@@ -11,7 +11,7 @@ def test_get_integrations_generator(self):
         response = next(generator)
         self.assertEqual(response['method'], 'GET')
         (uri, args) = response['uri'].split('?')
-        self.assertEqual(uri, '/admin/v2/integrations')
+        self.assertEqual(uri, '/admin/v3/integrations')
         self.assertEqual(
             util.params_to_dict(args),
             {
@@ -27,7 +27,7 @@ def test_get_integrations(self):
         response = response[0]
         self.assertEqual(response['method'], 'GET')
         (uri, args) = response['uri'].split('?')
-        self.assertEqual(uri, '/admin/v2/integrations')
+        self.assertEqual(uri, '/admin/v3/integrations')
         self.assertEqual(
             util.params_to_dict(args),
             {
@@ -43,7 +43,7 @@ def test_get_integrations_with_limit(self):
         response = response[0]
         self.assertEqual(response['method'], 'GET')
         (uri, args) = response['uri'].split('?')
-        self.assertEqual(uri, '/admin/v2/integrations')
+        self.assertEqual(uri, '/admin/v3/integrations')
         self.assertEqual(
             util.params_to_dict(args),
             {
@@ -59,7 +59,7 @@ def test_get_integrations_with_limit_offset(self):
         response = response[0]
         self.assertEqual(response['method'], 'GET')
         (uri, args) = response['uri'].split('?')
-        self.assertEqual(uri, '/admin/v2/integrations')
+        self.assertEqual(uri, '/admin/v3/integrations')
         self.assertEqual(
             util.params_to_dict(args),
             {
@@ -75,7 +75,7 @@ def test_get_integrations_with_offset(self):
         response = response[0]
         self.assertEqual(response['method'], 'GET')
         (uri, args) = response['uri'].split('?')
-        self.assertEqual(uri, '/admin/v2/integrations')
+        self.assertEqual(uri, '/admin/v3/integrations')
         self.assertEqual(
             util.params_to_dict(args),
             {