Issues/172

README.md

@@ -74,6 +74,18 @@ You should expect to see a message like `Application startup complete.` in the a
 python scripts/test.py
 ```
 
+For more comprehensive testing, install pytest
+```sh
+conda activate ndif-dev
+pip install pytest
+```
+Navigate to the service you wish to test
+```sh
+cd src/services/api
+pytest
+```
+
+
 This will send a test NNsight request to the API service running in the local container.
 
 ## Additional Commands

src/services/api/.gitignore

@@ -0,0 +1 @@
+*.egg-info

src/services/api/README.md

@@ -0,0 +1,15 @@
+## NDIF API Service
+
+## Testing
+
+install dependencies from requirements.in, ideally in a virtual environment
+
+run pytests with 
+```bash 
+pytest
+```
+
+run pytests in certain file with
+```bash
+pytest -k test_app 
+```

src/services/api/pyproject.toml

@@ -0,0 +1,23 @@
+[build-system]
+requires = ["setuptools>=61.0", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "api"
+version = "0.1.0"
+requires-python = ">=3.10"
+[tool.setuptools.packages.find]
+where = ["."]
+include = ["src*"]
+
+[tool.pytest.ini_options]
+minversion = "6.0"
+testpaths = ["tests"]
+pythonpath = ["."]
+
+[dependency-groups]
+dev = [
+    "httpx>=0.28.1",
+    "pytest>=8.4.0",
+    "pytest-asyncio>=1.3.0",
+]

src/services/api/tests/unit/conftest.py

@@ -0,0 +1,13 @@
+import os
+os.environ["DEV_MODE"] = "true"
+os.environ.setdefault("BROKER_URL", "redis://localhost:6379")
+
+from unittest.mock import AsyncMock, Mock, patch
+
+mock_redis_manager = Mock()
+mock_redis_client = AsyncMock()
+mock_socket_manager = Mock()
+
+patch("socketio.AsyncRedisManager", return_value=mock_redis_manager).start()
+patch("redis.asyncio.Redis.from_url", return_value=mock_redis_client).start()
+patch("fastapi_socketio.SocketManager", return_value=mock_socket_manager).start()

src/services/api/tests/unit/test_app.py

@@ -0,0 +1,11 @@
+import pytest
+from httpx import AsyncClient, ASGITransport
+from src.app import app
+
+
+@pytest.mark.asyncio
+async def test_ping():
+    async with AsyncClient(transport=ASGITransport(app=app), base_url="http://test") as client:
+        response = await client.get("/ping")
+        assert response.status_code == 200
+        assert response.json() == "pong"

src/services/api/tests/unit/test_dependencies.py

@@ -0,0 +1,33 @@
+import asyncio
+from unittest.mock import patch
+from src.dependencies import check_hotswapping_access, validate_nnsight_version, validate_python_version, authenticate_api_key
+import pytest
+import os
+
+
+@pytest.mark.asyncio
+async def test_validate_python_version():
+    with patch("sys.version", "3.12.10"):
+        res = await validate_python_version("3.12")
+        assert res == "3.12"
+
+
+@pytest.mark.asyncio
+async def test_validate_nnsight_version():
+    os.environ["MIN_NNSIGHT_VERSION"] = "0.5.10"
+    res = await validate_nnsight_version("0.5.10")
+    assert res == "0.5.10"
+
+
+@pytest.mark.asyncio
+async def test_authenticate_api_key_dev_mode():
+    os.environ["DEV_MODE"] = "true"
+    res = await authenticate_api_key("test-key")
+    assert res == "test-key"
+
+
+@pytest.mark.asyncio
+async def test_check_hotswapping_access_dev_mode():
+    os.environ["DEV_MODE"] = "true"
+    res = await check_hotswapping_access("test-key")
+    assert res == True

src/services/api/tests/unit/test_providers_util.py

@@ -0,0 +1,10 @@
+from unittest.mock import MagicMock, patch
+
+from src.providers.util import verify_connection
+
+
+def test_verify_connection_success():
+    with patch("socket.create_connection") as mock_conn:
+        mock_conn.return_value.__enter__.return_value = MagicMock()
+        assert verify_connection("127.0.0.1", 8000) is True
+        mock_conn.assert_called_once_with(("127.0.0.1", 8000), timeout=2)

src/services/api/tests/unit/test_queue_util.py

@@ -0,0 +1,28 @@
+import time
+from functools import lru_cache
+from src.queue.util import cache_maintainer
+
+
+def test_cache_maintainer():
+    calls = {"count": 0}
+
+    @cache_maintainer(1)       # clear every 1 second
+    @lru_cache(None)
+    def compute(x):
+        calls["count"] += 1
+        return x * 2
+
+    # First call — calls function
+    assert compute(5) == 10
+    assert calls["count"] == 1
+
+    # Second call immediately — should use cache
+    assert compute(5) == 10
+    assert calls["count"] == 1  # NO new calls
+
+    # Wait for 1 second → decorator clears cache
+    time.sleep(1.1)
+
+    # Next call — should trigger a new computation
+    assert compute(5) == 10
+    assert calls["count"] == 2

src/services/ray/src/ray/nn/security/protected_environment.py

@@ -83,6 +83,7 @@ def check(self, name: str) -> bool:
     WhitelistedModule(name="typing", strict=False),
     WhitelistedModule(name="_operator", strict=True),
     WhitelistedModule(name="operator", strict=True),
+    WhitelistedModule(name="pandas", strict = False),
 ]
 
 # Modules allowed during deserialization