Set RoutersJSONAPI class on AtomicViewHandler

Author: mahenzon

fastapi_jsonapi/atomic/atomic_handler.py

@@ -11,6 +11,7 @@
     Callable,
     List,
     Optional,
+    Type,
     TypedDict,
     Union,
 )
@@ -64,6 +65,8 @@ async def wrapper(*a, operation: OperationBase, **kw):
 
 
 class AtomicViewHandler:
+    jsonapi_routers_cls: Type[RoutersJSONAPI] = RoutersJSONAPI
+
     def __init__(
         self,
         request: Request,
@@ -82,10 +85,10 @@ async def prepare_one_operation(self, operation: AtomicOperation):
         """
         operation_type = operation.ref and operation.ref.type or operation.data and operation.data.type
         assert operation_type
-        if operation_type not in RoutersJSONAPI.all_jsonapi_routers:
+        if operation_type not in self.jsonapi_routers_cls.all_jsonapi_routers:
             msg = f"Unknown resource type {operation_type!r}. Register it via RoutersJSONAPI"
             raise ValueError(msg)
-        jsonapi = RoutersJSONAPI.all_jsonapi_routers[operation_type]
+        jsonapi = self.jsonapi_routers_cls.all_jsonapi_routers[operation_type]
 
         one_operation = OperationBase.prepare(
             action=operation.op,