def test_should_work_with_real_data(self, publish_mock):
config = get_collector_config('MemoryCgroupCollector', {
'interval': 10,
'byte_unit': 'megabyte',
'memory_path': fixtures_path
})
self.collector = MemoryCgroupCollector(config, None)
self.collector.collect()
self.assertPublishedMany(
publish_mock, {
'lxc.testcontainer.cache': 1,
'lxc.testcontainer.rss': 1,
'lxc.testcontainer.swap': 1,
'lxc.cache': 1,
'lxc.rss': 1,
'lxc.swap': 1,
'system.cache': 1,
'system.rss': 1,
'system.swap': 1,
'lxc.testcontainer.total_cache': 1,
'lxc.testcontainer.total_rss': 1,
'lxc.testcontainer.total_swap': 1,
'lxc.total_cache': 1,
'lxc.total_rss': 1,
'lxc.total_swap': 1,
'system.total_cache': 1,
'system.total_rss': 1,
'system.total_swap': 1,
})
def setUp(self):
config = get_collector_config('MemoryCgroupCollector', {
'interval': 10,
'byte_unit': 'megabyte'
})
self.collector = MemoryCgroupCollector(config, None)
def test_should_open_all_memory_stat(self, publish_mock, open_mock):
config = get_collector_config('MemoryCgroupCollector', {
'interval': 10,
'byte_unit': 'megabyte'
})
self.collector = MemoryCgroupCollector(config, None)
open_mock.side_effect = lambda x: StringIO('')
self.collector.collect()
open_mock.assert_any_call(fixtures_path +
'lxc/testcontainer/memory.stat')
open_mock.assert_any_call(fixtures_path + 'lxc/memory.stat')
open_mock.assert_any_call(fixtures_path + 'memory.stat')
def test_should_not_include_filtered_metrics(self, publish_mock):
config = get_collector_config(
'MemoryCgroupCollector', {
'interval': 10,
'byte_unit': 'megabyte',
'memory_path': fixtures_path,
'skip': 'test\w+ner',
})
self.collector = MemoryCgroupCollector(config, None)
self.collector.collect()
should_be_published = {
'lxc.cache': 1,
'lxc.rss': 1,
'lxc.swap': 1,
'system.cache': 1,
'system.rss': 1,
'system.swap': 1,
'lxc.total_cache': 1,
'lxc.total_rss': 1,
'lxc.total_swap': 1,
'system.total_cache': 1,
'system.total_rss': 1,
'system.total_swap': 1,
}
should_not_be_published = {
'lxc.testcontainer.cache': 1,
'lxc.testcontainer.rss': 1,
'lxc.testcontainer.swap': 1,
'lxc.testcontainer.total_cache': 1,
'lxc.testcontainer.total_rss': 1,
'lxc.testcontainer.total_swap': 1,
}
[
self.assertPublished(publish_mock, k, v)
for k, v in should_be_published.iteritems()
]
[
self.assertUnpublished(publish_mock, k, v)
for k, v in should_not_be_published.iteritems()
]
