More test coverage

tests/test_version_extra.py

@@ -1,7 +1,6 @@
 from __future__ import annotations
 
-# The version module is hard to test fully because it uses importlib.metadata
-# which is difficult to mock. We'll test what we can.
+import importlib.metadata as metadata
 
 
 def test_get_enroll_version_returns_string():
@@ -10,3 +9,51 @@ def test_get_enroll_version_returns_string():
     result = get_enroll_version()
     assert isinstance(result, str)
     assert len(result) > 0
+
+
+def test_get_enroll_version_falls_back_after_bad_mapped_dist(monkeypatch):
+    from enroll.version import get_enroll_version
+
+    monkeypatch.setattr(
+        metadata,
+        "packages_distributions",
+        lambda: {"enroll": ["wrong-dist", "enroll"]},
+    )
+
+    def fake_version(name):
+        if name == "enroll":
+            return "0.7.0"
+        raise metadata.PackageNotFoundError(name)
+
+    monkeypatch.setattr(metadata, "version", fake_version)
+
+    assert get_enroll_version() == "0.7.0"
+
+
+def test_get_enroll_version_uses_default_when_mapping_fails(monkeypatch):
+    from enroll.version import get_enroll_version
+
+    def broken_mapping():
+        raise RuntimeError("metadata unavailable")
+
+    monkeypatch.setattr(metadata, "packages_distributions", broken_mapping)
+    monkeypatch.setattr(
+        metadata,
+        "version",
+        lambda name: "0.7.0" if name == "enroll" else "bad",
+    )
+
+    assert get_enroll_version() == "0.7.0"
+
+
+def test_get_enroll_version_returns_unknown_when_all_lookups_fail(monkeypatch):
+    from enroll.version import get_enroll_version
+
+    monkeypatch.setattr(metadata, "packages_distributions", lambda: {"enroll": ["bad"]})
+
+    def missing(_name):
+        raise metadata.PackageNotFoundError(_name)
+
+    monkeypatch.setattr(metadata, "version", missing)
+
+    assert get_enroll_version() == "unknown"