3232def test_get_dependency_version_py38_plus (mock_version ):
3333 """Test get_dependency_version on Python 3.8+."""
3434 mock_version .return_value = "1.2.3"
35- assert get_dependency_version ("some-package" ) == parse_version ("1.2.3" )
35+ assert get_dependency_version ("some-package" ) == ( parse_version ("1.2.3" ), "1.2.3" )
3636 mock_version .assert_called_once_with ("some-package" )
3737
3838 # Test package not found
3939 mock_version .side_effect = ImportError
40- assert get_dependency_version ("not-a-package" ) is None
40+ assert get_dependency_version ("not-a-package" ) == ( None , "--" )
4141
4242
4343# TODO(https://github.com/googleapis/python-api-core/issues/835): Remove
@@ -49,14 +49,14 @@ def test_get_dependency_version_py37(mock_get_distribution):
4949 mock_dist = MagicMock ()
5050 mock_dist .version = "4.5.6"
5151 mock_get_distribution .return_value = mock_dist
52- assert get_dependency_version ("another-package" ) == parse_version ("4.5.6" )
52+ assert get_dependency_version ("another-package" ) == ( parse_version ("4.5.6" ), "4.5.6" )
5353 mock_get_distribution .assert_called_once_with ("another-package" )
5454
5555 # Test package not found
5656 mock_get_distribution .side_effect = (
5757 Exception # pkg_resources has its own exception types
5858 )
59- assert get_dependency_version ("not-a-package" ) is None
59+ assert get_dependency_version ("not-a-package" ) == ( None , "--" )
6060
6161
6262@patch ("google.api_core._python_package_support._get_distribution_and_import_packages" )
@@ -71,9 +71,8 @@ def test_warn_deprecation_for_versions_less_than(
7171 ("my-package (my.package)" , "my-package" ),
7272 ]
7373
74- # Case 1: Installed version is less than required, should warn.
75- mock_get_version .return_value = parse_version ("1.0.0" )
76- with pytest .warns (UserWarning ) as record :
74+ mock_get_version .return_value = (parse_version ("1.0.0" ), "1.0.0" )
75+ with pytest .warns (FutureWarning ) as record :
7776 warn_deprecation_for_versions_less_than ("my.package" , "dep.package" , "2.0.0" )
7877 assert len (record ) == 1
7978 assert (
@@ -87,17 +86,17 @@ def test_warn_deprecation_for_versions_less_than(
8786
8887 # Case 2: Installed version is equal to required, should not warn.
8988 mock_get_packages .reset_mock ()
90- mock_get_version .return_value = parse_version ("2.0.0" )
89+ mock_get_version .return_value = ( parse_version ("2.0.0" ), "2.0.0" )
9190 warn_deprecation_for_versions_less_than ("my.package" , "dep.package" , "2.0.0" )
9291
9392 # Case 3: Installed version is greater than required, should not warn.
9493 mock_get_packages .reset_mock ()
95- mock_get_version .return_value = parse_version ("3.0.0" )
94+ mock_get_version .return_value = ( parse_version ("3.0.0" ), "3.0.0" )
9695 warn_deprecation_for_versions_less_than ("my.package" , "dep.package" , "2.0.0" )
9796
9897 # Case 4: Dependency not found, should not warn.
9998 mock_get_packages .reset_mock ()
100- mock_get_version .return_value = None
99+ mock_get_version .return_value = ( None , "--" )
101100 warn_deprecation_for_versions_less_than ("my.package" , "dep.package" , "2.0.0" )
102101
103102 # Assert that no warnings were recorded
@@ -109,9 +108,9 @@ def test_warn_deprecation_for_versions_less_than(
109108 ("dep-package (dep.package)" , "dep-package" ),
110109 ("my-package (my.package)" , "my-package" ),
111110 ]
112- mock_get_version .return_value = parse_version ("1.0.0" )
111+ mock_get_version .return_value = ( parse_version ("1.0.0" ), "1.0.0" )
113112 template = "Custom warning for {dependency_package} used by {dependent_package}."
114- with pytest .warns (UserWarning ) as record :
113+ with pytest .warns (FutureWarning ) as record :
115114 warn_deprecation_for_versions_less_than (
116115 "my.package" , "dep.package" , "2.0.0" , message_template = template
117116 )
0 commit comments