Skip to content

Commit d1919ec

Browse files
committed
Fix tests
1 parent 2c052e9 commit d1919ec

File tree

1 file changed

+13
-2
lines changed

1 file changed

+13
-2
lines changed

unit_tests/sources/streams/http/test_http.py

Lines changed: 13 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -92,7 +92,10 @@ def test_requests_native_token_authenticator():
9292

9393
def test_request_kwargs_used(mocker, requests_mock):
9494
stream = StubBasicReadHttpStream()
95-
request_kwargs = {"cert": None, "proxies": "google.com"}
95+
request_kwargs = {
96+
"cert": None,
97+
"proxies": {"http": "http://example.com", "https": "http://example.com"},
98+
}
9699
mocker.patch.object(stream, "request_kwargs", return_value=request_kwargs)
97100
send_mock = mocker.patch.object(
98101
stream._http_client._session, "send", wraps=stream._http_client._session.send
@@ -101,8 +104,16 @@ def test_request_kwargs_used(mocker, requests_mock):
101104

102105
list(stream.read_records(sync_mode=SyncMode.full_refresh))
103106

104-
stream._http_client._session.send.assert_any_call(ANY, **request_kwargs)
105107
assert send_mock.call_count == 1
108+
call_args = send_mock.call_args_list[0]
109+
call_kwargs = call_args.kwargs
110+
111+
assert call_kwargs.get("cert") is None
112+
113+
proxies = call_kwargs.get("proxies")
114+
assert proxies is not None
115+
assert proxies["http"] == "http://example.com"
116+
assert proxies["https"] == "http://example.com"
106117

107118

108119
def test_stub_basic_read_http_stream_read_records(mocker):

0 commit comments

Comments
 (0)