in memory-repository

pyproject.toml

@@ -8,6 +8,7 @@ authors = ["Przemysław Górecki <[email protected]>"]
 python = "^3.9"
 pytest = "^6.2.4"
 pydantic = "^1.8.2"
+black = "^21.5b1"
 
 [tool.poetry.dev-dependencies]
 

src/modules/listing/domain/entities.py

@@ -2,4 +2,4 @@
 
 
 class AuctionItem(Entity):
-    pass
+    pass

src/modules/listing/domain/events.py

@@ -3,5 +3,3 @@
 
 class DraftCreatedEvent(DomainEvent):
     pass
-
-

src/modules/listing/domain/rules.py

@@ -1,9 +1,10 @@
 from seedwork.domain.rules import BusinessRule
 from seedwork.domain.value_objects import Currency
 
+
 class AuctionItemPriceMustBeGreaterThanZero(BusinessRule):
     message = "Price must be greater that zero"
     price: Currency
 
     def is_broken(self) -> bool:
-        return self.price <= 0
+        return self.price <= 0

src/seedwork/application/command.py

@@ -2,4 +2,4 @@
 
 
 class Command(BaseModel):
-    pass
+    pass

src/seedwork/application/exceptions.py

@@ -0,0 +1,2 @@
+class EntityNotFoundException(Exception):
+    pass

src/seedwork/domain/entities.py

@@ -1,2 +1,6 @@
-class Entity:
-    pass
+from pydantic import BaseModel, Field
+import uuid
+
+
+class Entity(BaseModel):
+    id: uuid.UUID = Field(default_factory=uuid.uuid4)

src/seedwork/domain/events.py

@@ -2,4 +2,4 @@
 
 
 class DomainEvent(BaseModel):
-    pass
+    pass

src/seedwork/domain/exceptions.py

@@ -3,4 +3,4 @@ class DomainException(Exception):
 
 
 class BusinessRuleValidationException(DomainException):
-    pass
+    pass

src/seedwork/domain/rules.py

@@ -8,4 +8,4 @@ def get_message(self) -> str:
         return self.message
 
     def is_broken(self) -> bool:
-        pass
+        pass

src/seedwork/domain/value_objects.py

@@ -1,2 +1,2 @@
 class Currency(int):
-    pass
+    pass

src/seedwork/infrastructure/repository.py

@@ -1,2 +1,20 @@
+from seedwork.domain.entities import Entity
+from seedwork.application.exceptions import EntityNotFoundException
+
+
 class Repository:
-    pass
+    pass
+
+
+class InMemoryRepository(Repository):
+    def __init__(self) -> None:
+        self.objects = {}
+
+    def get_by_id(self, id) -> Entity:
+        try:
+            return self.objects[id]
+        except KeyError:
+            raise EntityNotFoundException
+
+    def persist(self, entity: Entity):
+        self.objects[entity.id] = entity

src/seedwork/infrastructure/test_repository.py

@@ -0,0 +1,34 @@
+from seedwork.infrastructure.repository import InMemoryRepository
+from seedwork.domain.entities import Entity
+
+
+class Person(Entity):
+    first_name: str
+    last_name: str
+
+
+def test_InMemoryRepository_persist_one():
+    # arrange
+    person = Person(first_name="John", last_name="Doe")
+    repository = InMemoryRepository()
+
+    # act
+    repository.persist(person)
+
+    # assert
+    assert repository.get_by_id(person.id) == person
+
+
+def test_InMemoryRepository_persist_two():
+    # arrange
+    person1 = Person(first_name="John", last_name="Doe")
+    person2 = Person(first_name="Mary", last_name="Doe")
+    repository = InMemoryRepository()
+
+    # act
+    repository.persist(person1)
+    repository.persist(person2)
+
+    # assert
+    assert repository.get_by_id(person1.id) == person1
+    assert repository.get_by_id(person2.id) == person2