Merge branch 'master' of github.com:robinhood/faust

robinhood · Jan 8, 2020 · a9450e2 · a9450e2
2 parents abcd7d2 + 3c329bf
commit a9450e2
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 2 deletions.
diff --git a/faust/agents/agent.py b/faust/agents/agent.py
@@ -435,7 +435,7 @@ def info(self) -> Mapping:
             'channel': self.channel,
             'concurrency': self.concurrency,
             'help': self.help,
-            'sinks': self._sinks,
+            'sink': self._sinks,
             'on_error': self._on_error,
             'supervisor_strategy': self.supervisor_strategy,
             'isolated_partitions': self.isolated_partitions,

diff --git a/t/unit/agents/test_agent.py b/t/unit/agents/test_agent.py
@@ -361,7 +361,7 @@ def test_info(self, *, agent):
             'channel': agent.channel,
             'concurrency': agent.concurrency,
             'help': agent.help,
-            'sinks': agent._sinks,
+            'sink': agent._sinks,
             'on_error': agent._on_error,
             'supervisor_strategy': agent.supervisor_strategy,
             'isolated_partitions': agent.isolated_partitions,
@@ -898,3 +898,15 @@ def test_channel_iterator(self, *, agent):
 
     def test_label(self, *, agent):
         assert label(agent)
+
+    async def test_context_calls_sink(self, *, agent):
+        class SinkCalledException(Exception):
+            pass
+
+        def dummy_sink(_):
+            raise SinkCalledException()
+
+        agent.add_sink(dummy_sink)
+        async with agent.test_context() as agent_mock:
+            with pytest.raises(SinkCalledException):
+                await agent_mock.put('hello')