add chat events, tests

Author: RivanDev

Dockerfile

@@ -1,5 +1,8 @@
 FROM python:3.12.1-slim-bookworm
 
+ENV PYTHONDONTWRITEBYTECODE 1
+ENV PYTHONUNBUFFERED 1
+
 WORKDIR /app
 
 RUN apt update -y && \

Makefile

@@ -19,4 +19,8 @@ app-shell:
 
 .PHONY: app-logs
 app-logs:
-	${LOGS} ${APP_CONTAINER} -f
+	${LOGS} ${APP_CONTAINER} -f
+
+.PHONY: test
+test:
+	${EXEC} ${APP_CONTAINER} pytest tests

app/domain/entities/base.py

@@ -1,10 +1,13 @@
 from abc import ABC
+from copy import copy
 from dataclasses import dataclass, field
 from datetime import datetime
 from uuid import uuid4
 
+from domain.events.base import BaseEvent
 
-@dataclass(frozen=True)
+
+@dataclass
 class BaseEntity(ABC):
     oid: str = field(
         default_factory=lambda: str(uuid4()),
@@ -16,8 +19,21 @@ class BaseEntity(ABC):
         kw_only=True,
     )
 
+    _events: list[BaseEvent] = field(
+        default_factory=list,
+        kw_only=True,
+    )
+
     def __hash__(self) -> int:
         return hash(self.oid)
 
-    def __eq__(self, __value: "BaseEntity") -> bool:
+    def __eq__(self, __value: "BaseEvent") -> bool:
         return self.oid == __value.oid
+
+    def register_event(self, event: BaseEvent) -> None:
+        self._events.append(event)
+
+    def pull_events(self) -> list[BaseEvent]:
+        registered_events = copy(self._events)
+        self._events.clear()
+        return registered_events

app/domain/entities/messages.py

@@ -1,21 +1,35 @@
 from dataclasses import dataclass, field
 
 from domain.entities.base import BaseEntity
+from domain.events.messages import NewMessageReceivedEvent
 from domain.values.messages import Text, Title
 
 
-@dataclass(frozen=True)
+@dataclass(eq=False)
 class Message(BaseEntity):
     text: Text
 
 
-@dataclass(frozen=True)
+@dataclass(eq=False)
 class Chat(BaseEntity):
     title: Title
     messages: set[Message] = field(
         default_factory=set,
         kw_only=True,
     )
 
+    def __hash__(self) -> int:
+        return hash(self.oid)
+
+    def __eq__(self, __value: "Chat") -> bool:
+        return self.oid == __value.oid
+
     def add_message(self, message: Message):
         self.messages.add(message)
+        self.register_event(
+            NewMessageReceivedEvent(
+                message_text=message.text.as_generic_type(),
+                chat_oid=self.oid,
+                message_oid=message.oid,
+            )
+        )

app/domain/events/__init__.py

@@ -0,0 +1,8 @@
+from abc import ABC
+from dataclasses import dataclass, field
+from uuid import uuid4, UUID
+
+
+@dataclass
+class BaseEvent(ABC):
+    event_id: UUID = field(default_factory=uuid4, kw_only=True)

app/domain/events/base.py

@@ -0,0 +1,10 @@
+from dataclasses import dataclass
+
+from domain.events.base import BaseEvent
+
+
+@dataclass
+class NewMessageReceivedEvent(BaseEvent):
+    message_oid: str
+    message_text: str
+    chat_oid: str

app/domain/events/messages.py

@@ -16,4 +16,4 @@ def __post_init__(self):
     def validate(self): ...
 
     @abstractmethod
-    def as_generic_type(self): ...
+    def as_generic_type(self) -> VT: ...

app/domain/values/base.py

@@ -26,5 +26,5 @@ def validate(self):
         if len(self.value) > 255:
             raise TitleTooLongException(self.value)
 
-    def as_generic_type(self):
+    def as_generic_type(self) -> str:
         return str(self.value)

app/domain/values/messages.py

@@ -3,6 +3,7 @@
 import pytest
 
 from domain.entities.messages import Message, Chat
+from domain.events.messages import NewMessageReceivedEvent
 from domain.exceptions.messages import TitleTooLongException
 from domain.values.messages import Text, Title
 
@@ -38,3 +39,25 @@ def test_add_message_to_chat():
 
     chat.add_message(message)
     assert message in chat.messages
+
+
+def test_new_message_events():
+    text = Text("Hello word")
+    message = Message(text=text)
+
+    title = Title("title")
+    chat = Chat(title=title)
+
+    chat.add_message(message)
+    events = chat.pull_events()
+    pulled_events = chat.pull_events()
+
+    assert not pulled_events, pulled_events
+    assert len(events) == 1, events
+
+    new_event = events[0]
+
+    assert isinstance(new_event, NewMessageReceivedEvent), new_event
+    assert new_event.message_oid == message.oid
+    assert new_event.message_text == message.text.as_generic_type()
+    assert new_event.chat_oid == chat.oid