1 # -*- coding: utf-8 -*-
3 # Copyright 2018 Whitestack, LLC
4 # *************************************************************
6 # This file is part of OSM Monitoring module
7 # All Rights Reserved to Whitestack, LLC
9 # Licensed under the Apache License, Version 2.0 (the "License"); you may
10 # not use this file except in compliance with the License. You may obtain
11 # a copy of the License at
13 # http://www.apache.org/licenses/LICENSE-2.0
15 # Unless required by applicable law or agreed to in writing, software
16 # distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
17 # WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
18 # License for the specific language governing permissions and limitations
20 # For those usages not covered by the Apache License, Version 2.0 please
21 # contact: bdiaz@whitestack.com or glavado@whitestack.com
24 # pylint: disable=E1101
27 from unittest
import TestCase
, mock
31 from osm_mon
.collector
.vnf_collectors
.openstack
import GnocchiBackend
32 from osm_mon
.core
.config
import Config
35 class CollectorTest(TestCase
):
38 self
.config
= Config()
43 @mock.patch
.object(GnocchiBackend
, "_build_neutron_client")
44 @mock.patch
.object(GnocchiBackend
, "_build_gnocchi_client")
45 def test_collect_gnocchi_rate_instance(self
, build_gnocchi_client
, _
):
46 mock_gnocchi_client
= mock
.Mock()
47 mock_gnocchi_client
.metric
= mock
.Mock()
48 mock_vim_session
= mock
.Mock()
49 mock_gnocchi_client
.metric
.get_measures
.return_value
= [
57 tzinfo
=datetime
.timezone(datetime
.timedelta(0), "+00:00"),
69 tzinfo
=datetime
.timezone(datetime
.timedelta(0), "+00:00"),
75 build_gnocchi_client
.return_value
= mock_gnocchi_client
77 backend
= GnocchiBackend({"_id": "test_uuid"}, mock_vim_session
)
78 value
= backend
._collect
_instance
_metric
("cpu", "test_resource_id")
79 self
.assertEqual(value
, 1.0)
80 mock_gnocchi_client
.metric
.get_measures
.assert_called_once_with(
82 aggregation
="rate:mean",
84 resource_id
="test_resource_id",
87 @mock.patch
.object(GnocchiBackend
, "_build_neutron_client")
88 @mock.patch
.object(GnocchiBackend
, "_build_gnocchi_client")
89 def test_collect_gnocchi_non_rate_instance(self
, build_gnocchi_client
, _
):
90 mock_gnocchi_client
= mock
.Mock()
91 mock_vim_session
= mock
.Mock()
92 mock_gnocchi_client
.metric
.get_measures
.return_value
= [
100 tzinfo
=datetime
.timezone(datetime
.timedelta(0), "+00:00"),
112 tzinfo
=datetime
.timezone(datetime
.timedelta(0), "+00:00"),
118 build_gnocchi_client
.return_value
= mock_gnocchi_client
120 backend
= GnocchiBackend({"_id": "test_uuid"}, mock_vim_session
)
121 value
= backend
._collect
_instance
_metric
("memory.usage", "test_resource_id")
122 self
.assertEqual(value
, 128)
123 mock_gnocchi_client
.metric
.get_measures
.assert_called_once_with(
127 resource_id
="test_resource_id",
130 @mock.patch
.object(GnocchiBackend
, "_build_neutron_client")
131 @mock.patch
.object(GnocchiBackend
, "_build_gnocchi_client")
132 def test_collect_gnocchi_no_metric(self
, build_gnocchi_client
, _
):
133 mock_gnocchi_client
= mock
.Mock()
134 mock_vim_session
= mock
.Mock()
135 mock_gnocchi_client
.metric
.get_measures
.side_effect
= (
136 gnocchiclient
.exceptions
.NotFound()
138 build_gnocchi_client
.return_value
= mock_gnocchi_client
140 backend
= GnocchiBackend({"_id": "test_uuid"}, mock_vim_session
)
141 value
= backend
._collect
_instance
_metric
("memory.usage", "test_resource_id")
142 self
.assertIsNone(value
)
143 mock_gnocchi_client
.metric
.get_measures
.assert_called_once_with(
147 resource_id
="test_resource_id",
150 @mock.patch
.object(GnocchiBackend
, "_build_neutron_client")
151 @mock.patch
.object(GnocchiBackend
, "_build_gnocchi_client")
152 def test_collect_interface_all_metric(
153 self
, build_gnocchi_client
, build_neutron_client
155 mock_gnocchi_client
= mock
.Mock()
156 mock_vim_session
= mock
.Mock()
157 mock_gnocchi_client
.resource
.search
.return_value
= [
161 mock_gnocchi_client
.metric
.get_measures
.return_value
= [
169 tzinfo
=datetime
.timezone(datetime
.timedelta(0), "+00:00"),
181 tzinfo
=datetime
.timezone(datetime
.timedelta(0), "+00:00"),
188 build_gnocchi_client
.return_value
= mock_gnocchi_client
190 backend
= GnocchiBackend({"_id": "test_uuid"}, mock_vim_session
)
191 value
= backend
._collect
_interface
_all
_metric
(
192 "packets_received", "test_resource_id"
194 self
.assertEqual(value
, 0.0666140726)
195 mock_gnocchi_client
.metric
.get_measures
.assert_any_call(
196 "packets_received", resource_id
="test_id_1", limit
=1
198 mock_gnocchi_client
.metric
.get_measures
.assert_any_call(
199 "packets_received", resource_id
="test_id_2", limit
=1
202 @mock.patch
.object(GnocchiBackend
, "_build_neutron_client")
203 @mock.patch
.object(GnocchiBackend
, "_build_gnocchi_client")
204 def test_collect_instance_disk_metric(
205 self
, build_gnocchi_client
, build_neutron_client
207 mock_gnocchi_client
= mock
.Mock()
208 mock_vim_session
= mock
.Mock()
209 mock_gnocchi_client
.resource
.search
.return_value
= [
212 mock_gnocchi_client
.metric
.get_measures
.return_value
= [
220 tzinfo
=datetime
.timezone(datetime
.timedelta(0), "+00:00"),
232 tzinfo
=datetime
.timezone(datetime
.timedelta(0), "+00:00"),
239 build_gnocchi_client
.return_value
= mock_gnocchi_client
241 backend
= GnocchiBackend({"_id": "test_uuid"}, mock_vim_session
)
242 value
= backend
._collect
_instance
_disk
_metric
(
243 "disk_read_bytes", "test_resource_id"
245 self
.assertEqual(value
, 230848)
246 mock_gnocchi_client
.metric
.get_measures
.assert_any_call(
247 "disk_read_bytes", resource_id
="test_id", limit
=1