@@ -668,9 +668,10 @@ def test_given_message_repository_when_get_access_token_then_emit_message(
668668 message_repository = Mock ()
669669 authenticator = SingleUseRefreshTokenOauth2Authenticator (
670670 connector_config ,
671- token_refresh_endpoint = "foobar " ,
671+ token_refresh_endpoint = "https://refresh_endpoint.com " ,
672672 client_id = connector_config ["credentials" ]["client_id" ],
673673 client_secret = connector_config ["credentials" ]["client_secret" ],
674+ token_expiry_is_time_of_expiration = True ,
674675 token_expiry_date_format = "YYYY-MM-DD" ,
675676 message_repository = message_repository ,
676677 )
@@ -681,9 +682,6 @@ def test_given_message_repository_when_get_access_token_then_emit_message(
681682 )
682683 mocker .patch .object (requests , "request" , side_effect = mock_request , autospec = True )
683684
684- # authenticator.refresh_access_token = mocker.Mock(
685- # return_value=("new_access_token", "2023-04-04", "new_refresh_token")
686- # )
687685 authenticator .token_has_expired = mocker .Mock (return_value = True )
688686
689687 authenticator .get_access_token ()
@@ -744,21 +742,26 @@ def test_given_message_repository_when_get_access_token_then_log_request(
744742 def test_refresh_access_token (self , mocker , connector_config ):
745743 authenticator = SingleUseRefreshTokenOauth2Authenticator (
746744 connector_config ,
747- token_refresh_endpoint = "foobar " ,
745+ token_refresh_endpoint = "https://refresh_endpoint.com " ,
748746 client_id = connector_config ["credentials" ]["client_id" ],
749747 client_secret = connector_config ["credentials" ]["client_secret" ],
750748 )
751-
752- authenticator ._make_handled_request = mocker .Mock (
749+
750+ # Mock the response from the refresh token endpoint
751+ resp .status_code = 200
752+ mocker .patch .object (
753+ resp , "json" ,
753754 return_value = {
754755 authenticator .get_access_token_name (): "new_access_token" ,
755756 authenticator .get_expires_in_name (): "42" ,
756757 authenticator .get_refresh_token_name (): "new_refresh_token" ,
757758 }
758759 )
760+ mocker .patch .object (requests , "request" , side_effect = mock_request , autospec = True )
761+
759762 assert authenticator .refresh_access_token () == (
760763 "new_access_token" ,
761- "42" ,
764+ ab_datetime_now (). add ( timedelta ( seconds = 42 )) ,
762765 "new_refresh_token" ,
763766 )
764767
0 commit comments