More tests and fixes

nats-io · Jul 17, 2024 · 7ccc454 · 7ccc454
1 parent 2268103
commit 7ccc454
Show file tree

Hide file tree

Showing 4 changed files with 339 additions and 157 deletions.
diff --git a/nats/micro/__init__.py b/nats/micro/__init__.py
@@ -12,14 +12,21 @@
 # limitations under the License.
 #
 
+from dataclasses import replace
 from nats.aio.client import Client
+from typing import Optional
 
 from .service import Service, ServiceConfig
 from .request import Request, Handler
 
 
-async def add_service(nc: Client, config: ServiceConfig) -> Service:
+async def add_service(nc: Client, config: Optional[ServiceConfig] = None, **kwargs) -> Service:
     """Add a service."""
+    if config:
+        config = replace(config, **kwargs)
+    else:
+        config = ServiceConfig(**kwargs)
+
     service = Service(nc, config)
     await service.start()
 

diff --git a/nats/micro/request.py b/nats/micro/request.py
@@ -74,7 +74,11 @@ async def respond_error(
         :param data: The error data.
         :param headers: Additional response headers.
         """
-        headers = headers or {}
+        if headers:
+            headers = headers.copy()
+        else:
+            headers = {}
+
         headers.update(
             {
                 ERROR_HEADER: description,