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_gnocchi_client
.metric
.get_measures
.return_value
= [
56 tzinfo
=datetime
.timezone(datetime
.timedelta(0), "+00:00"),
68 tzinfo
=datetime
.timezone(datetime
.timedelta(0), "+00:00"),
74 build_gnocchi_client
.return_value
= mock_gnocchi_client
76 backend
= GnocchiBackend({"_id": "test_uuid"})
77 value
= backend
._collect
_instance
_metric
("cpu", "test_resource_id")
78 self
.assertEqual(value
, 1.0)
79 mock_gnocchi_client
.metric
.get_measures
.assert_called_once_with(
81 aggregation
="rate:mean",
83 resource_id
="test_resource_id",
86 @mock.patch
.object(GnocchiBackend
, "_build_neutron_client")
87 @mock.patch
.object(GnocchiBackend
, "_build_gnocchi_client")
88 def test_collect_gnocchi_non_rate_instance(self
, build_gnocchi_client
, _
):
89 mock_gnocchi_client
= mock
.Mock()
90 mock_gnocchi_client
.metric
.get_measures
.return_value
= [
98 tzinfo
=datetime
.timezone(datetime
.timedelta(0), "+00:00"),
110 tzinfo
=datetime
.timezone(datetime
.timedelta(0), "+00:00"),
116 build_gnocchi_client
.return_value
= mock_gnocchi_client
118 backend
= GnocchiBackend({"_id": "test_uuid"})
119 value
= backend
._collect
_instance
_metric
("memory.usage", "test_resource_id")
120 self
.assertEqual(value
, 128)
121 mock_gnocchi_client
.metric
.get_measures
.assert_called_once_with(
125 resource_id
="test_resource_id",
128 @mock.patch
.object(GnocchiBackend
, "_build_neutron_client")
129 @mock.patch
.object(GnocchiBackend
, "_build_gnocchi_client")
130 def test_collect_gnocchi_no_metric(self
, build_gnocchi_client
, _
):
131 mock_gnocchi_client
= mock
.Mock()
132 mock_gnocchi_client
.metric
.get_measures
.side_effect
= (
133 gnocchiclient
.exceptions
.NotFound()
135 build_gnocchi_client
.return_value
= mock_gnocchi_client
137 backend
= GnocchiBackend({"_id": "test_uuid"})
138 value
= backend
._collect
_instance
_metric
("memory.usage", "test_resource_id")
139 self
.assertIsNone(value
)
140 mock_gnocchi_client
.metric
.get_measures
.assert_called_once_with(
144 resource_id
="test_resource_id",
147 @mock.patch
.object(GnocchiBackend
, "_build_neutron_client")
148 @mock.patch
.object(GnocchiBackend
, "_build_gnocchi_client")
149 def test_collect_interface_all_metric(
150 self
, build_gnocchi_client
, build_neutron_client
152 mock_gnocchi_client
= mock
.Mock()
153 mock_gnocchi_client
.resource
.search
.return_value
= [
157 mock_gnocchi_client
.metric
.get_measures
.return_value
= [
165 tzinfo
=datetime
.timezone(datetime
.timedelta(0), "+00:00"),
177 tzinfo
=datetime
.timezone(datetime
.timedelta(0), "+00:00"),
184 build_gnocchi_client
.return_value
= mock_gnocchi_client
186 backend
= GnocchiBackend({"_id": "test_uuid"})
187 value
= backend
._collect
_interface
_all
_metric
(
188 "packets_received", "test_resource_id"
190 self
.assertEqual(value
, 0.0666140726)
191 mock_gnocchi_client
.metric
.get_measures
.assert_any_call(
192 "packets_received", resource_id
="test_id_1", limit
=1
194 mock_gnocchi_client
.metric
.get_measures
.assert_any_call(
195 "packets_received", resource_id
="test_id_2", limit
=1