pytest-girder 5.0.8.dev2__tar.gz → 5.0.8.dev6__tar.gz

{pytest_girder-5.0.8.dev2/pytest_girder.egg-info → pytest_girder-5.0.8.dev6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pytest-girder
-Version: 5.0.8.dev2
+Version: 5.0.8.dev6
 Summary: A set of pytest fixtures for testing Girder applications.
 Author: Kitware, Inc.
 Author-email: kitware@kitware.com

{pytest_girder-5.0.8.dev2 → pytest_girder-5.0.8.dev6}/pytest_girder/fixtures.py

@@ -1,13 +1,15 @@
 import hashlib
 import os
 import shutil
+import threading
+import time
 import unittest.mock
 
 import mongomock
 import pytest
 
 from .plugin_registry import PluginRegistry
-from .utils import serverContext
+from .utils import _findFreePort, serverContext
 
 
 def _uid(node):
@@ -145,6 +147,52 @@ def boundServer(db, request):
             yield server
 
 
+@pytest.fixture
+def asgiBoundServer(db, request):
+    """
+    Require an ASGI server that listens on a port.
+
+    Provides a started uvicorn server with a bound port. The returned value has
+    a `boundPort` property identifying where the server can be reached.
+    """
+    import uvicorn
+
+    from girder.asgi import _WSGIBridge
+
+    registry = PluginRegistry()
+    with registry():
+        plugins = _getPluginsFromMarker(request, registry)
+        with serverContext(plugins, bindPort=False) as server_fixture:
+            port = _findFreePort()
+            server_fixture.boundPort = port
+            config = uvicorn.Config(
+                _WSGIBridge(server_fixture.serverRoot),
+                host='127.0.0.1',
+                port=port,
+                log_level='warning',
+                server_header=False,
+                date_header=False,
+                lifespan='off',
+                access_log=False,
+                loop='asyncio',
+            )
+            asgi_server = uvicorn.Server(config)
+            thread = threading.Thread(target=asgi_server.run, daemon=True)
+            thread.start()
+            deadline = time.monotonic() + 10
+            while not asgi_server.started and thread.is_alive() and time.monotonic() < deadline:
+                time.sleep(0.01)
+            if not asgi_server.started:
+                asgi_server.should_exit = True
+                thread.join(timeout=5)
+                raise RuntimeError('ASGI server failed to start')
+            try:
+                yield server_fixture
+            finally:
+                asgi_server.should_exit = True
+                thread.join(timeout=5)
+
+
 @pytest.fixture
 def email_stdout(db):
     old_env_val = os.environ.get('GIRDER_EMAIL_TO_CONSOLE')
@@ -209,11 +257,12 @@ def fsAssetstore(db, request):
 
 __all__ = (
     'admin',
+    'asgiBoundServer',
+    'boundServer',
     'db',
+    'eagerWorkerTasks',
+    'email_stdout',
     'fsAssetstore',
     'server',
-    'boundServer',
     'user',
-    'email_stdout',
-    'eagerWorkerTasks',
 )

{pytest_girder-5.0.8.dev2 → pytest_girder-5.0.8.dev6/pytest_girder.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pytest-girder
-Version: 5.0.8.dev2
+Version: 5.0.8.dev6
 Summary: A set of pytest fixtures for testing Girder applications.
 Author: Kitware, Inc.
 Author-email: kitware@kitware.com