fix unit test

Author: dmjb

src/codegate/api/v1.py

@@ -728,7 +728,7 @@ async def stream_sse():
 @v1.get("/version", tags=["Dashboard"], generate_unique_id_function=uniq_name)
 async def version_check():
     try:
-        latest_version = __get_latest_version()
+        latest_version = _get_latest_version()
         # normalize the versions as github will return them with a 'v' prefix
         current_version = __version__.lstrip("v")
         latest_version_stripped = latest_version.lstrip("v")
@@ -885,6 +885,6 @@ async def delete_persona(persona_name: str):
 
 
 @cachetools.func.ttl_cache(maxsize=128, ttl=20 * 60)
-def __get_latest_version():
+def _get_latest_version():
     update_client = get_update_client_singleton()
     return update_client.get_latest_version(Origin.FrontEnd)

tests/test_server.py

@@ -16,6 +16,7 @@
 from codegate.pipeline.factory import PipelineFactory
 from codegate.providers.registry import ProviderRegistry
 from codegate.server import init_app
+from codegate.updates.client import Origin
 from src.codegate.cli import UvicornServer, cli
 from src.codegate.codegate_logging import LogFormat, LogLevel
 
@@ -75,18 +76,20 @@ def test_health_check(test_client: TestClient) -> None:
     assert response.json() == {"status": "healthy"}
 
 
-@patch("codegate.api.v1_processing.fetch_latest_version", return_value="foo")
-def test_version_endpoint(mock_fetch_latest_version, test_client: TestClient) -> None:
+@patch("codegate.api.v1._get_latest_version")
+def test_version_endpoint(mock_get_latest_version, test_client: TestClient) -> None:
     """Test the version endpoint."""
+    # Mock the __get_latest_version function to return a specific version
+    mock_get_latest_version.return_value = "v1.2.3"
+
     response = test_client.get("/api/v1/version")
     assert response.status_code == 200
 
     response_data = response.json()
-
-    assert response_data["current_version"] == __version__.lstrip("v")
-    assert response_data["latest_version"] == "foo"
-    assert isinstance(response_data["is_latest"], bool)
+    assert response_data["current_version"] == "0.1.7"
+    assert response_data["latest_version"] == "1.2.3"
     assert response_data["is_latest"] is False
+    assert response_data["error"] is None
 
 
 @patch("codegate.pipeline.sensitive_data.manager.SensitiveDataManager")