diff --git a/tests/test_user_management.py b/tests/test_user_management.py index b5d8c3b1..245b028f 100644 --- a/tests/test_user_management.py +++ b/tests/test_user_management.py @@ -1013,6 +1013,7 @@ def test_enroll_auth_factor( type = "totp" totp_issuer = "WorkOS" totp_user = "marcelina@foo-corp.com" + totp_secret = "secret-test" mock_request_method("post", mock_enroll_auth_factor_response, 200) @@ -1021,6 +1022,7 @@ def test_enroll_auth_factor( type=type, totp_issuer=totp_issuer, totp_user=totp_user, + totp_secret=totp_secret, ) assert enroll_auth_factor == mock_enroll_auth_factor_response diff --git a/workos/__about__.py b/workos/__about__.py index e12b6abb..a4bf0d09 100644 --- a/workos/__about__.py +++ b/workos/__about__.py @@ -12,7 +12,7 @@ __package_url__ = "https://github.com/workos-inc/workos-python" -__version__ = "4.9.0" +__version__ = "4.10.0" __author__ = "WorkOS" diff --git a/workos/user_management.py b/workos/user_management.py index a322bb0b..502b1259 100644 --- a/workos/user_management.py +++ b/workos/user_management.py @@ -1113,6 +1113,7 @@ def enroll_auth_factor( type, totp_issuer=None, totp_user=None, + totp_secret=None, ): """Enrolls a user in a new auth factor. @@ -1121,6 +1122,7 @@ def enroll_auth_factor( type (str): The type of factor to enroll (Only option available is 'totp'). totp_issuer (str): Name of the Organization (Optional) totp_user (str): Email of user (Optional) + totp_secret (str): The secret key for the TOTP factor. Generated if not provided. (Optional) Returns: { WorkOSAuthenticationFactorTotp, WorkOSChallenge} """ @@ -1134,6 +1136,7 @@ def enroll_auth_factor( "type": type, "totp_issuer": totp_issuer, "totp_user": totp_user, + "totp_secret": totp_secret, } response = self.request_helper.request(