class PortStatCollectorTestCase(CollectorTestCase):
TEST_CONFIG = {
'port': {
'something1': {
'number': 5222,
},
'something2': {
'number': 8888,
}
}
}
def setUp(self):
config = get_collector_config('PortStatCollector',
self.TEST_CONFIG)
self.collector = PortStatCollector(config, None)
def test_import(self):
self.assertTrue(PortStatCollector)
@patch('portstat.get_port_stats')
@patch.object(Collector, 'publish')
def test_collect(self, publish_mock, get_port_stats_mock):
get_port_stats_mock.return_value = {'foo': 1}
self.collector.collect()
get_port_stats_mock.assert_has_calls([call(5222), call(8888)],
any_order=True)
self.assertPublished(publish_mock, 'something1.foo', 1)
self.assertPublished(publish_mock, 'something2.foo', 1)
def setUp(self):
config = get_collector_config('PortStatCollector',
self.TEST_CONFIG)
self.collector = PortStatCollector(config, None)
