Add Bus clean method

Author: José Márquez Doblas

plato_cqrs/_command_bus.py

@@ -21,3 +21,6 @@ def publish(self, cmd: Command):
         self._semaphore.acquire()
         self._commands[cmd.__class__].handle(cmd)
         self._semaphore.release()
+        
+    def clean(self):
+        self._commands = {}

plato_cqrs/_query_bus.py

@@ -24,4 +24,6 @@ def publish(self, query: Query) -> QueryResponse:
         response: QueryResponse = self._querys[query.__class__].handle(query)
         self._semaphore.release()
         return response
-        
+
+    def clean(self):
+        self._querys = {}

plato_cqrs/tests/test_command_bus.py

@@ -30,4 +30,11 @@ def test_cannot_subscribe_to_two_handlers(self):
     def test_cannot_call_a_command_without_handler(self):
         command_bus: CommandBus = CommandBus()
         with self.assertRaises(CommandHandlerDoesNotExistException):
-            command_bus.publish(self.mockedCommnand)
+            command_bus.publish(self.mockedCommnand)
+            
+    def test_clean_command_bus(self):
+        command_bus: CommandBus = CommandBus()
+        command_bus.subscribe(self.mockedCommnand.__class__, self.mockedCommnandHandler)
+        command_bus.clean()
+        self.assertEqual(command_bus._commands, {})
+        

plato_cqrs/tests/test_query_bus.py

@@ -44,4 +44,10 @@ def test_cannot_subscribe_to_two_handlers(self):
     def test_cannot_call_a_query_without_handler(self):
         query_bus: QueryBus = QueryBus()
         with self.assertRaises(QueryHandlerDoesNotExistException):
-            query_bus.publish(self.mockedQuery)
+            query_bus.publish(self.mockedQuery)
+
+    def test_clean_query_bus(self):
+        query_bus: QueryBus = QueryBus()
+        query_bus.subscribe(self.mockedQuery.__class__, self.mockedQueryHandler)
+        query_bus.clean()
+        self.assertEqual(query_bus._querys, {})