Add tests

Author: febus982

tests/socketio_app/namespaces/__init__.py

@@ -0,0 +1,47 @@
+from unittest.mock import AsyncMock, patch
+
+import pytest
+
+from socketio_app import ChatNamespace
+
+
+@pytest.fixture
+def chat_namespace():
+    return ChatNamespace("/chat")
+
+
+@pytest.mark.asyncio
+async def test_on_connect(chat_namespace):
+    sid = "test_session_id"
+    environ = {}
+
+    # Test that connect doesn't raise any exceptions
+    chat_namespace.on_connect(sid, environ)
+
+
+@pytest.mark.asyncio
+async def test_on_disconnect(chat_namespace):
+    sid = "test_session_id"
+    reason = "test_reason"
+
+    # Test that disconnect doesn't raise any exceptions
+    chat_namespace.on_disconnect(sid, reason)
+
+
+@pytest.mark.asyncio
+async def test_on_echo_message(chat_namespace):
+    sid = "test_session_id"
+    test_data = {"message": "Hello, World!"}
+
+    # Mock the emit method
+    chat_namespace.emit = AsyncMock()
+
+    # Mock the logging
+    with patch("logging.info") as mock_log:
+        await chat_namespace.on_echo_message(sid, test_data)
+
+        # Verify logging was called
+        mock_log.assert_called_once_with("received message")
+
+        # Verify emit was called with correct arguments
+        chat_namespace.emit.assert_called_once_with("echo_response", test_data)

tests/socketio_app/namespaces/test_chat.py

@@ -0,0 +1,55 @@
+from unittest.mock import patch
+
+import socketio
+from starlette.routing import Mount, Router
+
+from common import AppConfig
+from socketio_app import create_app
+
+
+def test_create_app_returns_router():
+    """Test that create_app returns a Router instance"""
+    app = create_app()
+    assert isinstance(app, Router)
+
+
+def test_create_app_with_custom_config():
+    """Test that create_app accepts custom config"""
+    test_config = AppConfig(DEBUG=True)
+    with patch("common.bootstrap.init_storage", return_value=None):
+        app = create_app(test_config=test_config)
+
+    assert isinstance(app, Router)
+
+
+def test_create_app_routes():
+    """Test that create_app creates all expected routes"""
+    with patch("common.bootstrap.init_storage", return_value=None):
+        app = create_app()
+
+    # Check that we have exactly 3 routes (docs JSON, docs HTML, and socketio mount)
+    assert len(app.routes) == 3
+
+    # Check routes paths and methods
+    routes = [(route.path, getattr(route, "methods", None)) for route in app.routes]
+    assert ("/docs/asyncapi.json", {"GET", "HEAD"}) in routes
+    assert ("/docs", {"GET", "HEAD"}) in routes
+
+    # Check that one route is a Mount instance for socketio
+    mount_routes = [route for route in app.routes if isinstance(route, Mount)]
+    assert len(mount_routes) == 1
+    assert mount_routes[0].name == "socketio"
+    assert isinstance(mount_routes[0].app, socketio.ASGIApp)
+
+
+def test_create_app_socketio_namespace():
+    """Test that socketio server has the chat namespace registered"""
+    with patch("common.bootstrap.init_storage", return_value=None):
+        app = create_app()
+
+    # Find the socketio mount
+    socketio_mount = next(route for route in app.routes if isinstance(route, Mount))
+    sio_app = socketio_mount.app
+
+    # Check that the chat namespace is registered
+    assert "/chat" in sio_app.engineio_server.namespace_handlers

tests/socketio_app/test_app_factory.py

@@ -0,0 +1,96 @@
+from unittest.mock import Mock, patch
+
+import pytest
+from pydantic import BaseModel
+from starlette.requests import Request
+
+from socketio_app.web_routes.docs import (
+    ASYNCAPI_CSS_DEFAULT_URL,
+    ASYNCAPI_JS_DEFAULT_URL,
+    NORMALIZE_CSS_DEFAULT_URL,
+    PydanticResponse,
+    asyncapi_json,
+    get_asyncapi_html,
+)
+
+
+# Test model
+class TestModel(BaseModel):
+    name: str
+    value: int
+
+
+# Fixtures
+@pytest.fixture
+def test_model():
+    return TestModel(name="test", value=42)
+
+
+@pytest.fixture
+def mock_request():
+    return Mock(spec=Request)
+
+
+@pytest.fixture
+def mock_app_config():
+    with patch("socketio_app.web_routes.docs.AppConfig") as mock:
+        mock.return_value.APP_NAME = "Test App"
+        yield mock
+
+
+# Tests for PydanticResponse
+def test_pydantic_response_render(test_model):
+    response = PydanticResponse(test_model)
+    expected = b'{"name":"test","value":42}'
+    assert response.render(test_model) == expected
+
+
+# Tests for asyncapi_json endpoint
+async def test_asyncapi_json(mock_request, test_model):
+    with patch("socketio_app.web_routes.docs.get_schema") as mock_get_schema:
+        mock_get_schema.return_value = test_model
+        response = await asyncapi_json(mock_request)
+        assert isinstance(response, PydanticResponse)
+        assert response.body == b'{"name":"test","value":42}'
+
+
+# Tests for get_asyncapi_html endpoint
+async def test_get_asyncapi_html_default_params(mock_request, mock_app_config):
+    mock_request.query_params = {}
+    response = await get_asyncapi_html(mock_request)
+
+    assert response.status_code == 200
+    assert response.headers["content-type"] == "text/html; charset=utf-8"
+
+    content = response.body.decode()
+    assert "Test App AsyncAPI" in content
+    assert ASYNCAPI_JS_DEFAULT_URL in content
+    assert NORMALIZE_CSS_DEFAULT_URL in content
+    assert ASYNCAPI_CSS_DEFAULT_URL in content
+    assert '"sidebar": true' in content
+    assert '"info": true' in content
+
+
+async def test_get_asyncapi_html_custom_params(mock_request, mock_app_config):
+    mock_request.query_params = {
+        "sidebar": "false",
+        "info": "false",
+        "servers": "false",
+        "operations": "false",
+        "messages": "false",
+        "schemas": "false",
+        "errors": "false",
+        "expand_message_examples": "true",
+    }
+
+    response = await get_asyncapi_html(mock_request)
+    content = response.body.decode()
+
+    assert '"sidebar": false' in content
+    assert '"info": false' in content
+    assert '"servers": false' in content
+    assert '"operations": false' in content
+    assert '"messages": false' in content
+    assert '"schemas": false' in content
+    assert '"errors": false' in content
+    assert '"messageExamples": true' in content