Take version from loaded package when metadata are missing

Author: Fab Micheloni

redshift_connector/idp_auth_helper.py

@@ -47,7 +47,13 @@ def get_pkg_version(module_name: str) -> Version:
 
             return Version(pkg_resources.get_distribution(module_name).version)
 
-        return Version(version(module_name))
+        pkg_version = version(module_name)
+        if pkg_version is not None:
+            return Version(pkg_version)
+
+        import importlib
+        imported_module = importlib.import_module(module_name)
+        return Version(imported_module.__version__)
 
     @staticmethod
     def set_auth_properties(info: RedshiftProperty):

test/unit/test_idp_auth_helper.py

@@ -0,0 +1,16 @@
+from unittest.mock import MagicMock
+
+from packaging.version import Version
+from redshift_connector.idp_auth_helper import IdpAuthHelper
+
+
+def test_get_pkg_version(mocker) -> None:
+    mocker.patch("importlib.metadata.version", return_value=None)
+
+    module_mock = MagicMock()
+    module_mock.__version__ = "9.8.7"
+    mocker.patch("importlib.import_module", return_value=module_mock)
+
+    actual_version: Version = IdpAuthHelper.get_pkg_version("test_module")
+
+    assert actual_version == Version("9.8.7")