class TestCPUCollector(CollectorTestCase):
def setUp(self):
config = get_collector_config('CPUCollector', {
'interval': 10
})
self.collector = CPUCollector(config, None)
@patch('__builtin__.open')
@patch.object(Collector, 'publish')
def test_should_open_proc_stat(self, publish_mock, open_mock):
open_mock.return_value = StringIO('')
self.collector.collect()
open_mock.assert_called_once_with('/proc/stat')
@patch.object(Collector, 'publish')
def test_should_work_with_synthetic_data(self, publish_mock):
with patch('__builtin__.open', return_value = StringIO(
'cpu 100 200 300 400 500 0 0 0 0 0'
)):
self.collector.collect()
self.assertPublishedMany(publish_mock, {})
with patch('__builtin__.open', return_value = StringIO(
'cpu 110 220 330 440 550 0 0 0 0 0'
)):
self.collector.collect()
self.assertPublishedMany(publish_mock, {
'total.idle' : 4.0,
'total.iowait' : 5.0,
'total.nice' : 2.0,
'total.system' : 3.0,
'total.user' : 1.0
})
@patch.object(Collector, 'publish')
def test_should_work_with_real_data(self, publish_mock):
CPUCollector.PROC = get_fixture_path('proc_stat_1')
self.collector.collect()
self.assertPublishedMany(publish_mock, {})
CPUCollector.PROC = get_fixture_path('proc_stat_2')
self.collector.collect()
self.assertPublishedMany(publish_mock, {
'total.idle' : 2440.8,
'total.iowait' : 0.2,
'total.nice' : 0.0,
'total.system' : 0.2,
'total.user' : 0.4
})
def setUp(self):
config = get_collector_config('CPUCollector', {
'interval': 10
})
self.collector = CPUCollector(config, None)
