Exemplo n.º 1
0
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()
Exemplo n.º 2
0
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()