Add more unit tests

Author: seratch

tests/slack_bolt/listener/__init__.py

@@ -0,0 +1,27 @@
+import logging
+
+from slack_bolt import BoltRequest
+from slack_bolt.listener.listener_completion_handler import (
+    CustomListenerCompletionHandler,
+)
+
+
+class TestListenerCompletionHandler:
+    def test_handler(self):
+        result = {"called": False}
+
+        def f():
+            result["called"] = True
+
+        handler = CustomListenerCompletionHandler(
+            logger=logging.getLogger(__name__),
+            func=f,
+        )
+        request = BoltRequest(
+            body="{}",
+            query={},
+            headers={"content-type": ["application/json"]},
+            context={},
+        )
+        handler.handle(request, None)
+        assert result["called"] is True

tests/slack_bolt/listener/test_listener_completion_handler.py

@@ -0,0 +1,27 @@
+import logging
+
+from slack_bolt import BoltRequest
+from slack_bolt.listener.listener_start_handler import (
+    CustomListenerStartHandler,
+)
+
+
+class TestListenerStartHandler:
+    def test_handler(self):
+        result = {"called": False}
+
+        def f():
+            result["called"] = True
+
+        handler = CustomListenerStartHandler(
+            logger=logging.getLogger(__name__),
+            func=f,
+        )
+        request = BoltRequest(
+            body="{}",
+            query={},
+            headers={"content-type": ["application/json"]},
+            context={},
+        )
+        handler.handle(request, None)
+        assert result["called"] is True

tests/slack_bolt/listener/test_listener_start_handler.py

@@ -0,0 +1,42 @@
+import pytest
+
+from slack_bolt import Ack
+from slack_bolt.error import BoltError
+from slack_bolt.workflows.step import WorkflowStep
+
+
+class TestStep:
+    def test_build(self):
+        step = WorkflowStep.builder("foo")
+        step.edit(just_ack)
+        step.save(just_ack)
+        step.execute(just_ack)
+        assert step.build() is not None
+
+    def test_build_errors(self):
+        with pytest.raises(BoltError):
+            step = WorkflowStep.builder("foo")
+            step.edit(None)
+            step.save(just_ack)
+            step.execute(just_ack)
+            step.build()
+        with pytest.raises(BoltError):
+            step = WorkflowStep.builder("foo")
+            step.edit(just_ack)
+            step.save(None)
+            step.execute(just_ack)
+            step.build()
+        with pytest.raises(BoltError):
+            step = WorkflowStep.builder("foo")
+            step.edit(just_ack)
+            step.save(just_ack)
+            step.execute(None)
+            step.build()
+
+
+def just_ack(ack: Ack):
+    ack()
+
+
+def execute():
+    pass

tests/slack_bolt/workflows/__init__.py

@@ -0,0 +1,29 @@
+import logging
+import pytest
+
+from slack_bolt.request.async_request import AsyncBoltRequest
+from slack_bolt.listener.async_listener_completion_handler import (
+    AsyncCustomListenerCompletionHandler,
+)
+
+
+class TestAsyncListenerCompletionHandler:
+    @pytest.mark.asyncio
+    async def test_handler(self):
+        result = {"called": False}
+
+        async def f():
+            result["called"] = True
+
+        handler = AsyncCustomListenerCompletionHandler(
+            logger=logging.getLogger(__name__),
+            func=f,
+        )
+        request = AsyncBoltRequest(
+            body="{}",
+            query={},
+            headers={"content-type": ["application/json"]},
+            context={},
+        )
+        await handler.handle(request, None)
+        assert result["called"] is True

tests/slack_bolt/workflows/step/__init__.py

@@ -0,0 +1,29 @@
+import logging
+import pytest
+
+from slack_bolt.request.async_request import AsyncBoltRequest
+from slack_bolt.listener.async_listener_start_handler import (
+    AsyncCustomListenerStartHandler,
+)
+
+
+class TestAsyncListenerStartHandler:
+    @pytest.mark.asyncio
+    async def test_handler(self):
+        result = {"called": False}
+
+        async def f():
+            result["called"] = True
+
+        handler = AsyncCustomListenerStartHandler(
+            logger=logging.getLogger(__name__),
+            func=f,
+        )
+        request = AsyncBoltRequest(
+            body="{}",
+            query={},
+            headers={"content-type": ["application/json"]},
+            context={},
+        )
+        await handler.handle(request, None)
+        assert result["called"] is True