unittest: EventBiz

Author: freemindcore

tests/demo_app/domain.py

@@ -1,5 +1,12 @@
 from easy.domain import BaseDomain
+from tests.demo_app.models import Event
 
 
 class EventDomain(BaseDomain):
     pass
+
+
+class EventBiz(EventDomain):
+    def __init__(self, model=Event):
+        self.model = model
+        super(EventBiz, self).__init__(self.model)

tests/demo_app/services.py

@@ -2,7 +2,7 @@
 
 from django.http import HttpRequest
 
-from tests.demo_app.models import Event
+from tests.demo_app.domain import EventBiz
 
 if TYPE_CHECKING:
     from ninja_extra.controllers.base import ControllerBase  # pragma: no cover
@@ -11,8 +11,8 @@
 
 
 class EventService(BaseService):
-    def __init__(self, model=Event):
-        super().__init__(model=model)
+    def __init__(self, biz=EventBiz()):
+        super().__init__(biz.model)
 
     @staticmethod
     async def prepare_create_event_data(data):