class DescribeRun(_BaseTestCase): def configure(self): self.agent = ConsumerAgent(sentinel.consumer, sentinel.broker, sentinel.bindings) self.agent.connect = MagicMock() self.agent.connection = MagicMock() def execute(self): self.agent.run() def should_connect(self): self.agent.connect.assert_called_once_with() def should_start_ioloop(self): self.agent.connection.ioloop.start.assert_called_once_with()