Skip to content

Commit 3689ea7

Browse files
committed
add unit test to increase coverage
1 parent b888d1e commit 3689ea7

File tree

2 files changed

+13
-0
lines changed

2 files changed

+13
-0
lines changed

tests/compute_engine/test__metadata.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -329,6 +329,7 @@ def test_get_success_custom_root_old_variable():
329329
timeout=_metadata._METADATA_DEFAULT_TIMEOUT,
330330
)
331331

332+
332333
def test_get_success_custom_root():
333334
request = make_request("{}", headers={"content-type": "application/json"})
334335

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)