Пример #1
0
class TestRabbitMQPublisher(object):

    @mock.patch('qpaca.backend.rabbitmq.InfluxDB')
    @mock.patch('qpaca.backend.rabbitmq.RabbitMQPublisher._connect')
    def setup_class(self, mock1, mock2):
        self.publisher = RabbitMQPublisher()

    @mock.patch('qpaca.backend.rabbitmq.InfluxDB')
    @mock.patch('qpaca.backend.rabbitmq.RabbitMQPublisher._connect')
    def test_call_create_monitor(self, mocked_class, mocked_function):
        RabbitMQPublisher()
        assert mocked_class.called

    @mock.patch('qpaca.backend.rabbitmq.RabbitMQPublisher._connect')
    def test_call_connect(self, mocked_function):
        RabbitMQPublisher()
        assert mocked_function.called

    @mock.patch('qpaca.backend.rabbitmq.RabbitMQPublisher._create_exchange')
    def test_call_create_exchange(self, mocked_function):
        self.publisher.start()
        assert mocked_function.called

    @mock.patch('qpaca.backend.rabbitmq.RabbitMQPublisher._create_exchange')
    @mock.patch('qpaca.backend.rabbitmq.RabbitMQPublisher._create_producer')
    def test_call_create_producer(self, mocked_exchange, mocked_producer):
        self.publisher.start()
        assert self.publisher._create_producer.called

    def test_call_producer_publish(self):
        self.publisher.config = {'publish': {}}
        self.publisher._producer = mock.Mock()
        self.publisher.publish(None)
        assert self.publisher._producer.publish.called

    @mock.patch('kombu.Exchange')
    def test_create_exchange(self, mocked_exchange):
        self.publisher.config = {'exchange': {}}
        exchange = self.publisher._create_exchange()
        assert exchange is not None

    @mock.patch('kombu.Connection')
    def test_connect(self, mocked_connection):
        self.publisher.config = {'connection': {}}
        connection = self.publisher._connect()
        assert connection is not None
Пример #2
0
class TestRabbitMQPublisher(object):
    @mock.patch('qpaca.backend.rabbitmq.InfluxDB')
    @mock.patch('qpaca.backend.rabbitmq.RabbitMQPublisher._connect')
    def setup_class(self, mock1, mock2):
        self.publisher = RabbitMQPublisher()

    @mock.patch('qpaca.backend.rabbitmq.InfluxDB')
    @mock.patch('qpaca.backend.rabbitmq.RabbitMQPublisher._connect')
    def test_call_create_monitor(self, mocked_class, mocked_function):
        RabbitMQPublisher()
        assert mocked_class.called

    @mock.patch('qpaca.backend.rabbitmq.RabbitMQPublisher._connect')
    def test_call_connect(self, mocked_function):
        RabbitMQPublisher()
        assert mocked_function.called

    @mock.patch('qpaca.backend.rabbitmq.RabbitMQPublisher._create_exchange')
    def test_call_create_exchange(self, mocked_function):
        self.publisher.start()
        assert mocked_function.called

    @mock.patch('qpaca.backend.rabbitmq.RabbitMQPublisher._create_exchange')
    @mock.patch('qpaca.backend.rabbitmq.RabbitMQPublisher._create_producer')
    def test_call_create_producer(self, mocked_exchange, mocked_producer):
        self.publisher.start()
        assert self.publisher._create_producer.called

    def test_call_producer_publish(self):
        self.publisher.config = {'publish': {}}
        self.publisher._producer = mock.Mock()
        self.publisher.publish(None)
        assert self.publisher._producer.publish.called

    @mock.patch('kombu.Exchange')
    def test_create_exchange(self, mocked_exchange):
        self.publisher.config = {'exchange': {}}
        exchange = self.publisher._create_exchange()
        assert exchange is not None

    @mock.patch('kombu.Connection')
    def test_connect(self, mocked_connection):
        self.publisher.config = {'connection': {}}
        connection = self.publisher._connect()
        assert connection is not None