Skip to content

Commit 393dd6f

Browse files
committed
finish fixing tests
1 parent 6290c35 commit 393dd6f

File tree

1 file changed

+11
-8
lines changed

1 file changed

+11
-8
lines changed

unit_tests/sources/streams/http/requests_native_auth/test_requests_native_auth.py

Lines changed: 11 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)