Fix chrysn#269: allow overriding server transports

roysjosh · Jan 21, 2022 · 06f6386 · 06f6386
1 parent 1f03d4c
commit 06f6386
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/aiocoap/protocol.py b/aiocoap/protocol.py
@@ -216,7 +216,7 @@ async def create_client_context(cls, *, loggername="coap", loop=None):
 
     @classmethod
     @AwaitOrAenter.decorate
-    async def create_server_context(cls, site, bind=None, *, loggername="coap-server", loop=None, _ssl_context=None, multicast=[], server_credentials=None):
+    async def create_server_context(cls, site, bind=None, *, loggername="coap-server", loop=None, _ssl_context=None, multicast=[], server_credentials=None, transports=[]):
         """Create a context, bound to all addresses on the CoAP port (unless
         otherwise specified in the ``bind`` argument).
 
@@ -250,7 +250,9 @@ async def create_server_context(cls, site, bind=None, *, loggername="coap-server
 
         multicast_done = not multicast
 
-        for transportname in defaults.get_default_servertransports(loop=loop):
+        selected_transports = len(transports) > 0 and transports or defaults.get_default_servertransports(loop=loop)
+
+        for transportname in selected_transports:
             if transportname == 'udp6':
                 from .transports.udp6 import MessageInterfaceUDP6