chore: import upstream snapshot with attribution

2026-07-13 12:59:42 +08:00
commit 59f8f60dad
348 changed files with 139133 additions and 0 deletions
@@ -0,0 +1,91 @@
+"""Tests for hybrid_server."""
+
+import logging
+import sys
+from unittest.mock import MagicMock, patch
+
+
+def test_gpu_detected_logging(caplog):
+    """GPU detection should log GPU name and CUDA version when available."""
+    mock_torch = MagicMock()
+    mock_torch.cuda.is_available.return_value = True
+    mock_torch.cuda.get_device_name.return_value = "NVIDIA A100"
+    mock_torch.version.cuda = "12.1"
+
+    with patch.dict("sys.modules", {"torch": mock_torch}):
+        # Re-import to pick up the mock
+        import importlib
+        from opendataloader_pdf import hybrid_server
+
+        importlib.reload(hybrid_server)
+
+        with caplog.at_level(logging.INFO):
+            # Simulate the GPU detection block from main()
+            try:
+                import torch
+                if torch.cuda.is_available():
+                    gpu_name = torch.cuda.get_device_name(0)
+                    cuda_version = torch.version.cuda
+                    logging.getLogger(__name__).info(
+                        f"GPU detected: {gpu_name} (CUDA {cuda_version})"
+                    )
+            except ImportError:
+                pass
+
+    assert "GPU detected: NVIDIA A100 (CUDA 12.1)" in caplog.text
+
+
+def test_no_gpu_logging(caplog):
+    """Should log CPU fallback when no GPU is available."""
+    mock_torch = MagicMock()
+    mock_torch.cuda.is_available.return_value = False
+
+    with patch.dict("sys.modules", {"torch": mock_torch}):
+        with caplog.at_level(logging.INFO):
+            try:
+                import torch
+                if torch.cuda.is_available():
+                    pass
+                else:
+                    logging.getLogger(__name__).info("No GPU detected, using CPU.")
+            except ImportError:
+                pass
+
+    assert "No GPU detected, using CPU." in caplog.text
+
+
+def test_no_pytorch_logging(caplog):
+    """Should log CPU fallback when PyTorch is not installed."""
+    with patch.dict("sys.modules", {"torch": None}):
+        with caplog.at_level(logging.INFO):
+            try:
+                import torch  # noqa: F811
+                if torch.cuda.is_available():
+                    pass
+                else:
+                    logging.getLogger(__name__).info("No GPU detected, using CPU.")
+            except (ImportError, TypeError):
+                logging.getLogger(__name__).info(
+                    "No GPU detected, using CPU. (PyTorch not installed)"
+                )
+
+    assert "No GPU detected, using CPU. (PyTorch not installed)" in caplog.text
+
+
+def test_get_loop_setting_returns_asyncio_on_windows():
+    """On Windows, should return 'asyncio' to avoid uvloop errors (#323)."""
+    from opendataloader_pdf.hybrid_server import _get_loop_setting
+
+    with patch("sys.platform", "win32"):
+        assert _get_loop_setting() == "asyncio"
+
+
+def test_get_loop_setting_returns_auto_on_non_windows():
+    """On non-Windows platforms, should return 'auto' (uvloop if available)."""
+    from opendataloader_pdf.hybrid_server import _get_loop_setting
+
+    with patch("sys.platform", "darwin"):
+        assert _get_loop_setting() == "auto"
+
+    with patch("sys.platform", "linux"):
+        assert _get_loop_setting() == "auto"