Skip to content

Commit b888d1e

Browse files
committed
add custom root unit test to increase coverage
1 parent d31d480 commit b888d1e

File tree

2 files changed

+15
-1
lines changed

2 files changed

+15
-1
lines changed

tests/compute_engine/test__metadata.py

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

332+
def test_get_success_custom_root():
333+
request = make_request("{}", headers={"content-type": "application/json"})
334+
335+
fake_root = "http://another.metadata.service"
336+
337+
_metadata.get(request, PATH, root=fake_root)
338+
339+
request.assert_called_once_with(
340+
method="GET",
341+
url="{}/{}".format(fake_root, PATH),
342+
headers=_metadata._METADATA_HEADERS,
343+
timeout=_metadata._METADATA_DEFAULT_TIMEOUT,
344+
)
345+
332346

333347
@mock.patch("time.sleep", return_value=None)
334348
def test_get_failure(mock_sleep):

tests/compute_engine/test__mtls.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@
1616
#
1717

1818
import mock
19-
import pytest
19+
import pytest # type: ignore
2020

2121
from google.auth import environment_vars, exceptions
2222
from google.auth.compute_engine import _mtls

0 commit comments

Comments
 (0)