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.assertEqual(None, 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.assertEqual(None,
  46                          self.amqp_consumer.on_connection_closed("", 404,
  47                                                                  "Not Found"))
  48         self.amqp_consumer._closing = False
  49         self.assertEqual(None,
  50                          self.amqp_consumer.on_connection_closed("", 404,
  51                                                                  "Not Found"))
  52
  53     def test_reconnect(self):
  54         self.amqp_consumer._connection = mock.Mock(autospec=AmqpConsumer)
  55         self.amqp_consumer._connection.ioloop = mock.Mock()
  56         self.amqp_consumer._connection.ioloop.stop = mock.Mock(return_value=0)
  57         self.amqp_consumer.connect = mock.Mock(return_value=0)
  58         self.amqp_consumer._closing = True
  59         self.assertEqual(None, self.amqp_consumer.reconnect())
  60
  61     def test_on_channel_open(self):
  62         self.amqp_consumer._connection = mock.Mock(autospec=AmqpConsumer)
  63         self.amqp_consumer._connection.add_on_close_callback = \
  64             mock.Mock(return_value=0)
  65         self.amqp_consumer._channel = mock.Mock()
  66         self.amqp_consumer.add_on_channel_close_callback = mock.Mock()
  67         self.amqp_consumer._channel.exchange_declare = \
  68             mock.Mock(return_value=0)
  69         self.assertEqual(None,
  70                          self.amqp_consumer.on_channel_open(
  71                              self.amqp_consumer._channel))
  72
  73     def test_add_on_channel_close_callback(self):
  74         self.amqp_consumer._connection = mock.Mock(autospec=AmqpConsumer)
  75         self.amqp_consumer._connection.add_on_close_callback = \
  76             mock.Mock(return_value=0)
  77         self.amqp_consumer._channel = mock.Mock()
  78         self.amqp_consumer._channel.add_on_close_callback = mock.Mock()
  79         self.assertEqual(None,
  80                          self.amqp_consumer.add_on_channel_close_callback())
  81
  82     def test_on_channel_closed(self):
  83         self.amqp_consumer._connection = mock.Mock(autospec=AmqpConsumer)
  84         self.amqp_consumer._connection.close = mock.Mock(return_value=0)
  85         _channel = mock.Mock()
  86         self.assertEqual(None,
  87                          self.amqp_consumer.on_channel_closed(_channel,
  88                                                               "", ""))
  89
  90     def test_ion_exchange_declareok(self):
  91         self.amqp_consumer.setup_queue = mock.Mock(return_value=0)
  92         self.assertEqual(None, self.amqp_consumer.on_exchange_declareok(10))
  93
  94     def test_setup_queue(self):
  95         self.amqp_consumer._channel = mock.Mock()
  96         self.amqp_consumer._channel.add_on_close_callback = mock.Mock()
  97         self.assertEqual(None, self.amqp_consumer.setup_queue("collectd"))
  98
  99     def test_on_queue_declareok(self):
 100         self.amqp_consumer._channel = mock.Mock()
 101         self.amqp_consumer._channel.queue_bind = mock.Mock()
 102         self.assertEqual(None, self.amqp_consumer.on_queue_declareok(10))
 103
 104     def test__on_bindok(self):
 105         self.amqp_consumer._channel = mock.Mock()
 106         self.amqp_consumer._channel.basic_consume = mock.Mock()
 107         self.amqp_consumer.add_on_cancel_callback = mock.Mock()
 108         self.assertEqual(None, self.amqp_consumer._on_bindok(10))
 109
 110     def test_add_on_cancel_callback(self):
 111         self.amqp_consumer._channel = mock.Mock()
 112         self.amqp_consumer._channel.add_on_cancel_callback = mock.Mock()
 113         self.assertEqual(None, self.amqp_consumer.add_on_cancel_callback())
 114
 115     def test_on_consumer_cancelled(self):
 116         self.amqp_consumer._channel = mock.Mock()
 117         self.amqp_consumer._channel.close = mock.Mock()
 118         self.assertEqual(None, self.amqp_consumer.on_consumer_cancelled(10))
 119
 120     def test_on_message(self):
 121         body = "msg {} cpu/cpu-0/ipc 101010:10"
 122         properties = ""
 123         basic_deliver = mock.Mock()
 124         basic_deliver.delivery_tag = mock.Mock(return_value=0)
 125         self.amqp_consumer.ack_message = mock.Mock()
 126         self.assertEqual(None,
 127                          self.amqp_consumer.on_message(10, basic_deliver,
 128                                                        properties, body))
 129
 130     def test_ack_message(self):
 131         self.amqp_consumer._channel = mock.Mock()
 132         self.amqp_consumer._channel.basic_ack = mock.Mock()
 133         self.assertEqual(None, self.amqp_consumer.ack_message(10))
 134
 135     def test_on_cancelok(self):
 136         self.amqp_consumer._channel = mock.Mock()
 137         self.amqp_consumer._channel.close = mock.Mock()
 138         self.assertEqual(None, self.amqp_consumer.on_cancelok(10))
 139
 140     def test_run(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.assertEqual(None, self.amqp_consumer.run())
 145
 146     def test_stop(self):
 147         self.amqp_consumer._connection = mock.Mock(autospec=AmqpConsumer)
 148         self.amqp_consumer.connect = mock.Mock()
 149         self.amqp_consumer._connection.ioloop.start = mock.Mock()
 150         self.amqp_consumer._channel = mock.Mock()
 151         self.amqp_consumer._channel.basic_cancel = mock.Mock()
 152         self.assertEqual(None, self.amqp_consumer.stop())
 153
 154     def test_close_connection(self):
 155         self.amqp_consumer._connection = mock.Mock(autospec=AmqpConsumer)
 156         self.amqp_consumer._connection.close = mock.Mock()
 157         self.assertEqual(None, self.amqp_consumer.close_connection())