3 # Copyright (c) 2016-2017 Intel Corporation
5 # Licensed under the Apache License, Version 2.0 (the "License");
6 # you may not use this file except in compliance with the License.
7 # You may obtain a copy of the License at
9 # http://www.apache.org/licenses/LICENSE-2.0
11 # Unless required by applicable law or agreed to in writing, software
12 # distributed under the License is distributed on an "AS IS" BASIS,
13 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14 # See the License for the specific language governing permissions and
15 # limitations under the License.
18 # Unittest for yardstick.network_services.collector.subscriber
20 from __future__ import absolute_import
23 from yardstick.network_services.collector import subscriber
26 class CollectorTestCase(unittest.TestCase):
32 self.test_subscriber = subscriber.Collector(self.TRAFFIC_PROFILE,
35 def test_successful_init(self):
37 self.assertEqual(self.test_subscriber.traffic_profile, {})
38 self.assertEqual(self.test_subscriber.service, {})
40 def test_unsuccessful_init(self):
44 self.assertIsNone(self.test_subscriber.start())
47 self.assertIsNone(self.test_subscriber.stop())
49 def test_get_kpi(self):
51 class VnfAprrox(object):
56 def collect_kpi(self):
57 self.result = {'pkt_in_up_stream': 100,
58 'pkt_drop_up_stream': 5,
59 'pkt_in_down_stream': 50,
60 'pkt_drop_down_stream': 40}
64 result = self.test_subscriber.get_kpi(vnf)
66 self.assertEqual(result["vnf__1"]["pkt_in_up_stream"], 100)
67 self.assertEqual(result["vnf__1"]["pkt_drop_up_stream"], 5)
68 self.assertEqual(result["vnf__1"]["pkt_in_down_stream"], 50)
69 self.assertEqual(result["vnf__1"]["pkt_drop_down_stream"], 40)