Skip to content

Commit fc224b7

Browse files
committed
add unit test to satisfy coverage requirement
1 parent 05bcf6f commit fc224b7

File tree

1 file changed

+19
-0
lines changed

1 file changed

+19
-0
lines changed

tests/unit/test_python_package_support.py

Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -124,3 +124,22 @@ def test_warn_deprecation_for_versions_less_than(mock_get_version, mock_get_pack
124124
"Custom warning for dep-package (dep.package) used by my-package (my.package)."
125125
in str(record[0].message)
126126
)
127+
128+
129+
from google.api_core._python_package_support import (
130+
check_dependency_versions,
131+
DependencyConstraint,
132+
)
133+
134+
135+
@patch("google.api_core._python_package_support.warn_deprecation_for_versions_less_than")
136+
def test_check_dependency_versions_with_custom_warnings(mock_warn):
137+
"""Test check_dependency_versions with custom warning parameters."""
138+
custom_warning1 = DependencyConstraint("pkg1", "1.0.0", "2.0.0")
139+
custom_warning2 = DependencyConstraint("pkg2", "2.0.0", "3.0.0")
140+
141+
check_dependency_versions("my-consumer", custom_warning1, custom_warning2)
142+
143+
assert mock_warn.call_count == 2
144+
mock_warn.assert_any_call("my-consumer", "pkg1", "1.0.0", recommended_version="2.0.0")
145+
mock_warn.assert_any_call("my-consumer", "pkg2", "2.0.0", recommended_version="3.0.0")

0 commit comments

Comments
 (0)