Merge pull request #11 from davidjnevin/use_case_get_all_functionality

Use case get all functionality

src/chatcleaner/adapters/use_cases/clean.py

@@ -16,7 +16,7 @@ def __init__(
         self.uow = uow
         self.service = service
 
-    def _add(self, chat: str):
+    def _clean(self, chat: str):
         with self.uow:
             # clean
             result = self.service.clean(chat)
@@ -26,3 +26,7 @@ def _add(self, chat: str):
             model = cleaning_factory(**data_)
             self.uow.cleaning.add(model)
             self.uow.commit()
+
+    def _get_all(self):
+        with self.uow:
+            return self.uow.cleaning.get_all()

src/chatcleaner/domain/ports/use_cases/clean.py

@@ -9,9 +9,16 @@ class CleanUseCaseInterface(abc.ABC):
     def __init__(self, uow: CleaningUnitOfWorkInterface, chat: ChatServiceInterface):
         raise NotImplementedError
 
-    def add(self, chat: str) -> None:
-        return self._add(chat)
+    def clean(self, chat: str) -> None:
+        return self._clean(chat)
+
+    def get_all(self):
+        return self._get_all()
+
+    @abc.abstractmethod
+    def _clean(self, chat: str) -> None:
+        raise NotImplementedError
 
     @abc.abstractmethod
-    def _add(self, chat: str) -> None:
+    def _get_all(self):
         raise NotImplementedError

tests/integrations/test_cleaning_use_case.py

@@ -1,13 +1,34 @@
 from chatcleaner.configurator.containers import Container
 
 
-def test_clean_use_case_add(get_fake_container, get_clean_use_case):
+def test_clean_use_case_clean(get_fake_container, get_clean_use_case):
     with Container.cleaning_uow.override(get_fake_container.cleaning_uow):
         with Container.chat_service.override(get_fake_container.chat_service):
-            get_clean_use_case.add("\n19:10:00 from David to Everyone:\ntest")
+            get_clean_use_case.clean("\n19:10:00 from David to Everyone:\ntest")
             uow_ = get_fake_container.cleaning_uow()
             with uow_:
                 result = uow_.cleaning.get_all()
                 assert len(result) == 1
                 assert result[0].chat == "\n19:10:00 from David to Everyone:\ntest"
                 assert result[0].cleaned_chat == "test"
+
+
+def test_clean_use_case_get_all(get_fake_container, get_clean_use_case):
+    with Container.cleaning_uow.override(get_fake_container.cleaning_uow):
+        with Container.chat_service.override(get_fake_container.chat_service):
+            get_clean_use_case.clean("\n19:10:00 from David to Everyone:\ntest 1")
+            get_clean_use_case.clean("\n19:10:00 from David to Everyone:\ntest 2")
+            get_clean_use_case.clean("\n19:10:00 from David to Everyone:\ntest 3")
+            uow_ = get_fake_container.cleaning_uow()
+            with uow_:
+                result = get_clean_use_case.get_all()
+                # fixtures are scopes to module, so this should be 4
+                assert len(result) == 4
+                assert result[0].chat == "\n19:10:00 from David to Everyone:\ntest"
+                assert result[0].cleaned_chat == "test"
+                assert result[1].chat == "\n19:10:00 from David to Everyone:\ntest 1"
+                assert result[1].cleaned_chat == "test 1"
+                assert result[2].chat == "\n19:10:00 from David to Everyone:\ntest 2"
+                assert result[2].cleaned_chat == "test 2"
+                assert result[3].chat == "\n19:10:00 from David to Everyone:\ntest 3"
+                assert result[3].cleaned_chat == "test 3"