Skip to content

Commit 8947e48

Browse files
committed
add unit test to increase coverage
1 parent b888d1e commit 8947e48

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

tests/compute_engine/test__mtls.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -113,3 +113,15 @@ def test_create_session(mock_adapter, mock_session, mock_mds_mtls_config):
113113
session_instance.mount.assert_called_once_with(
114114
"https://", mock_adapter.return_value
115115
)
116+
117+
118+
@mock.patch("ssl.create_default_context")
119+
@mock.patch("requests.adapters.HTTPAdapter.proxy_manager_for")
120+
def test_mds_mtls_adapter_proxy_manager_for(
121+
mock_proxy_manager_for, mock_ssl_context, mock_mds_mtls_config
122+
):
123+
adapter = _mtls.MdsMtlsAdapter(mock_mds_mtls_config)
124+
adapter.proxy_manager_for("test_proxy")
125+
mock_proxy_manager_for.assert_called_once_with(
126+
"test_proxy", ssl_context=adapter.ssl_context
127+
)

0 commit comments

Comments
 (0)