tests/unit/network_services/nfvi/test_collectd.py

   1 # Copyright (c) 2016-2017 Intel Corporation
   2 #
   3 # Licensed under the Apache License, Version 2.0 (the "License");
   4 # you may not use this file except in compliance with the License.
   5 # You may obtain a copy of the License at
   6 #
   7 #      http://www.apache.org/licenses/LICENSE-2.0
   8 #
   9 # Unless required by applicable law or agreed to in writing, software
  10 # distributed under the License is distributed on an "AS IS" BASIS,
  11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12 # See the License for the specific language governing permissions and
  13 # limitations under the License.
  14
  15 from __future__ import absolute_import
  16 import unittest
  17 import multiprocessing
  18 import mock
  19
  20 from yardstick.network_services.nfvi.collectd import AmqpConsumer
  21
  22
  23 class TestAmqpConsumer(unittest.TestCase):
  24     def setUp(self):
  25         self.queue = multiprocessing.Queue()
  26         self.url = 'amqp://admin:admin@127.0.0.1:5672/%2F'
  27         self.amqp_consumer = AmqpConsumer(self.url, self.queue)
  28
  29     def test___init__(self):
  30         self.assertEqual(self.url, self.amqp_consumer._url)
  31
  32     def test_on_connection_open(self):
  33         self.amqp_consumer._connection = mock.Mock(autospec=AmqpConsumer)
  34         self.amqp_consumer._connection.add_on_close_callback = \
  35             mock.Mock(return_value=0)
  36         self.amqp_consumer._connection.channel = mock.Mock(return_value=0)
  37         self.assertIsNone(self.amqp_consumer.on_connection_open(10))
  38
  39     def test_on_connection_closed(self):
  40         self.amqp_consumer._connection = mock.Mock(autospec=AmqpConsumer)
  41         self.amqp_consumer._connection.ioloop = mock.Mock()
  42         self.amqp_consumer._connection.ioloop.stop = mock.Mock(return_value=0)
  43         self.amqp_consumer._connection.add_timeout = mock.Mock(return_value=0)
  44         self.amqp_consumer._closing = True
  45         self.assertIsNone(
  46             self.amqp_consumer.on_connection_closed("", 404, "Not Found"))
  47         self.amqp_consumer._closing = False
  48         self.assertIsNone(
  49             self.amqp_consumer.on_connection_closed("", 404, "Not Found"))
  50
  51     def test_reconnect(self):
  52         self.amqp_consumer._connection = mock.Mock(autospec=AmqpConsumer)
  53         self.amqp_consumer._connection.ioloop = mock.Mock()
  54         self.amqp_consumer._connection.ioloop.stop = mock.Mock(return_value=0)
  55         self.amqp_consumer.connect = mock.Mock(return_value=0)
  56         self.amqp_consumer._closing = True
  57         self.assertIsNone(self.amqp_consumer.reconnect())
  58
  59     def test_on_channel_open(self):
  60         self.amqp_consumer._connection = mock.Mock(autospec=AmqpConsumer)
  61         self.amqp_consumer._connection.add_on_close_callback = \
  62             mock.Mock(return_value=0)
  63         self.amqp_consumer._channel = mock.Mock()
  64         self.amqp_consumer.add_on_channel_close_callback = mock.Mock()
  65         self.amqp_consumer._channel.exchange_declare = \
  66             mock.Mock(return_value=0)
  67         self.assertIsNone(
  68             self.amqp_consumer.on_channel_open(self.amqp_consumer._channel))
  69
  70     def test_add_on_channel_close_callback(self):
  71         self.amqp_consumer._connection = mock.Mock(autospec=AmqpConsumer)
  72         self.amqp_consumer._connection.add_on_close_callback = \
  73             mock.Mock(return_value=0)
  74         self.amqp_consumer._channel = mock.Mock()
  75         self.amqp_consumer._channel.add_on_close_callback = mock.Mock()
  76         self.assertIsNone(self.amqp_consumer.add_on_channel_close_callback())
  77
  78     def test_on_channel_closed(self):
  79         self.amqp_consumer._connection = mock.Mock(autospec=AmqpConsumer)
  80         self.amqp_consumer._connection.close = mock.Mock(return_value=0)
  81         _channel = mock.Mock()
  82         self.assertIsNone(
  83             self.amqp_consumer.on_channel_closed(_channel, "", ""))
  84
  85     def test_ion_exchange_declareok(self):
  86         self.amqp_consumer.setup_queue = mock.Mock(return_value=0)
  87         self.assertIsNone(self.amqp_consumer.on_exchange_declareok(10))
  88
  89     def test_setup_queue(self):
  90         self.amqp_consumer._channel = mock.Mock()
  91         self.amqp_consumer._channel.add_on_close_callback = mock.Mock()
  92         self.assertIsNone(self.amqp_consumer.setup_queue("collectd"))
  93
  94     def test_on_queue_declareok(self):
  95         self.amqp_consumer._channel = mock.Mock()
  96         self.amqp_consumer._channel.queue_bind = mock.Mock()
  97         self.assertIsNone(self.amqp_consumer.on_queue_declareok(10))
  98
  99     def test__on_bindok(self):
 100         self.amqp_consumer._channel = mock.Mock()
 101         self.amqp_consumer._channel.basic_consume = mock.Mock()
 102         self.amqp_consumer.add_on_cancel_callback = mock.Mock()
 103         self.assertIsNone(self.amqp_consumer._on_bindok(10))
 104
 105     def test_add_on_cancel_callback(self):
 106         self.amqp_consumer._channel = mock.Mock()
 107         self.amqp_consumer._channel.add_on_cancel_callback = mock.Mock()
 108         self.assertIsNone(self.amqp_consumer.add_on_cancel_callback())
 109
 110     def test_on_consumer_cancelled(self):
 111         self.amqp_consumer._channel = mock.Mock()
 112         self.amqp_consumer._channel.close = mock.Mock()
 113         self.assertIsNone(self.amqp_consumer.on_consumer_cancelled(10))
 114
 115     def test_on_message(self):
 116         body = "msg {} cpu/cpu-0/ipc 101010:10"
 117         properties = ""
 118         basic_deliver = mock.Mock()
 119         basic_deliver.delivery_tag = mock.Mock(return_value=0)
 120         self.amqp_consumer.ack_message = mock.Mock()
 121         self.assertIsNone(
 122             self.amqp_consumer.on_message(10, basic_deliver, properties, body))
 123
 124     def test_ack_message(self):
 125         self.amqp_consumer._channel = mock.Mock()
 126         self.amqp_consumer._channel.basic_ack = mock.Mock()
 127         self.assertIsNone(self.amqp_consumer.ack_message(10))
 128
 129     def test_on_cancelok(self):
 130         self.amqp_consumer._channel = mock.Mock()
 131         self.amqp_consumer._channel.close = mock.Mock()
 132         self.assertIsNone(self.amqp_consumer.on_cancelok(10))
 133
 134     def test_run(self):
 135         self.amqp_consumer._connection = mock.Mock(autospec=AmqpConsumer)
 136         self.amqp_consumer.connect = mock.Mock()
 137         self.amqp_consumer._connection.ioloop.start = mock.Mock()
 138         self.assertIsNone(self.amqp_consumer.run())
 139
 140     def test_stop(self):
 141         self.amqp_consumer._connection = mock.Mock(autospec=AmqpConsumer)
 142         self.amqp_consumer.connect = mock.Mock()
 143         self.amqp_consumer._connection.ioloop.start = mock.Mock()
 144         self.amqp_consumer._channel = mock.Mock()
 145         self.amqp_consumer._channel.basic_cancel = mock.Mock()
 146         self.assertIsNone(self.amqp_consumer.stop())
 147
 148     def test_close_connection(self):
 149         self.amqp_consumer._connection = mock.Mock(autospec=AmqpConsumer)
 150         self.amqp_consumer._connection.close = mock.Mock()
 151         self.assertIsNone(self.amqp_consumer.close_connection())