Pass rest_options through Auth0 constructor

adamjmcgrath · adamjmcgrath · commit 16febb71975f · 2022-06-10T13:02:32.000+01:00
diff --git a/auth0/v3/management/auth0.py b/auth0/v3/management/auth0.py
@@ -80,7 +80,15 @@ def __init__(self, domain, token, rest_options=None):
 
             for name, cls in modules.items():
                 cls = asyncify(cls)
-                setattr(self, name, cls(domain=domain, token=token, rest_options=None))
+                setattr(
+                    self,
+                    name,
+                    cls(domain=domain, token=token, rest_options=rest_options),
+                )
         else:
             for name, cls in modules.items():
-                setattr(self, name, cls(domain=domain, token=token, rest_options=None))
+                setattr(
+                    self,
+                    name,
+                    cls(domain=domain, token=token, rest_options=rest_options),
+                )
diff --git a/auth0/v3/test/management/test_auth0.py b/auth0/v3/test/management/test_auth0.py
@@ -1,5 +1,7 @@
 import unittest
 
+import mock
+
 from ...management.actions import Actions
 from ...management.attack_protection import AttackProtection
 from ...management.auth0 import Auth0
@@ -29,6 +31,7 @@
 from ...management.user_blocks import UserBlocks
 from ...management.users import Users
 from ...management.users_by_email import UsersByEmail
+from ...rest import RestClientOptions
 
 
 class TestAuth0(unittest.TestCase):
@@ -120,3 +123,16 @@ def test_users_by_email(self):
 
     def test_users(self):
         self.assertIsInstance(self.a0.users, Users)
+
+    @mock.patch("auth0.v3.management.users.Users.__init__")
+    def test_args(self, users):
+        rest_options = RestClientOptions(retries=99)
+        Auth0(self.domain, self.token, rest_options=rest_options)
+        users.assert_called_with(
+            "user.some.domain",
+            "a-token",
+            True,
+            5.0,
+            "https",
+            rest_options,
+        )
