def __init__(self, config=None, handlers=[], name=None, configfile=None): super(BaseCollector, self).__init__(config, handlers, name, configfile) self.cpu_collector = CPUCollector(config=self.config, configfile=self.configfile, handlers=self.handlers) self.memory_collector = MemoryCollector(config=self.config, configfile=self.configfile, handlers=self.handlers) self.loadavg_collector = LoadAverageCollector( config=self.config, configfile=self.configfile, handlers=self.handlers) self.network_collector = NetworkCollector(config=self.config, configfile=self.configfile, handlers=self.handlers) self.diskusage_collector = DiskUsageCollector( config=self.config, configfile=self.configfile, handlers=self.handlers) self.diskspace_collector = DiskSpaceCollector( config=self.config, configfile=self.configfile, handlers=self.handlers) self.vmstat_collector = VMStatCollector(config=self.config, configfile=self.configfile, handlers=self.handlers)
class BaseCollector(diamond.collector.Collector): def __init__(self, config=None, handlers=[], name=None, configfile=None): super(BaseCollector, self).__init__(config, handlers, name, configfile) self.cpu_collector = CPUCollector(config=self.config, configfile=self.configfile, handlers=self.handlers) self.memory_collector = MemoryCollector(config=self.config, configfile=self.configfile, handlers=self.handlers) self.loadavg_collector = LoadAverageCollector(config=self.config, configfile=self.configfile, handlers=self.handlers) self.network_collector = NetworkCollector(config=self.config, configfile=self.configfile, handlers=self.handlers) self.diskusage_collector = DiskUsageCollector(config=self.config, configfile=self.configfile, handlers=self.handlers) self.diskspace_collector = DiskSpaceCollector(config=self.config, configfile=self.configfile, handlers=self.handlers) self.vmstat_collector = VMStatCollector(config=self.config, configfile=self.configfile, handlers=self.handlers) def get_default_config_help(self): config_help = super(BaseCollector, self).get_default_config_help() config_help.update({ 'simple': 'run simple mode on of its sub collectors', }) return config_help def get_default_config(self): config = super(BaseCollector, self).get_default_config() return config def collect(self): self.cpu_collector.collect() self.memory_collector.collect() self.loadavg_collector.collect() self.network_collector.collect() self.diskusage_collector.collect() self.diskspace_collector.collect() self.vmstat_collector.collect() return True
class BaseCollector(diamond.collector.Collector): def __init__(self, config=None, handlers=[], name=None, configfile=None): super(BaseCollector, self).__init__(config, handlers, name, configfile) self.cpu_collector = CPUCollector(config=self.config, configfile=self.configfile, handlers=self.handlers) self.memory_collector = MemoryCollector(config=self.config, configfile=self.configfile, handlers=self.handlers) self.loadavg_collector = LoadAverageCollector( config=self.config, configfile=self.configfile, handlers=self.handlers) self.network_collector = NetworkCollector(config=self.config, configfile=self.configfile, handlers=self.handlers) self.diskusage_collector = DiskUsageCollector( config=self.config, configfile=self.configfile, handlers=self.handlers) self.diskspace_collector = DiskSpaceCollector( config=self.config, configfile=self.configfile, handlers=self.handlers) self.vmstat_collector = VMStatCollector(config=self.config, configfile=self.configfile, handlers=self.handlers) def get_default_config_help(self): config_help = super(BaseCollector, self).get_default_config_help() config_help.update({ 'simple': 'run simple mode on of its sub collectors', }) return config_help def get_default_config(self): config = super(BaseCollector, self).get_default_config() return config def collect(self): self.cpu_collector.collect() self.memory_collector.collect() self.loadavg_collector.collect() self.network_collector.collect() self.diskusage_collector.collect() self.diskspace_collector.collect() self.vmstat_collector.collect() return True
def __init__(self, config=None, handlers=[], name=None, configfile=None): super(BaseCollector, self).__init__(config, handlers, name, configfile) self.cpu_collector = CPUCollector(config=self.config, configfile=self.configfile, handlers=self.handlers) self.memory_collector = MemoryCollector(config=self.config, configfile=self.configfile, handlers=self.handlers) self.loadavg_collector = LoadAverageCollector(config=self.config, configfile=self.configfile, handlers=self.handlers) self.network_collector = NetworkCollector(config=self.config, configfile=self.configfile, handlers=self.handlers) self.diskusage_collector = DiskUsageCollector(config=self.config, configfile=self.configfile, handlers=self.handlers) self.diskspace_collector = DiskSpaceCollector(config=self.config, configfile=self.configfile, handlers=self.handlers) self.vmstat_collector = VMStatCollector(config=self.config, configfile=self.configfile, handlers=self.handlers)
class TestVMStatCollector(CollectorTestCase): def setUp(self): config = get_collector_config('VMStatCollector', {'interval': 10}) self.collector = VMStatCollector(config, None) @patch('__builtin__.open') @patch('os.access', Mock(return_value=True)) @patch.object(Collector, 'publish') def test_should_open_proc_vmstat(self, publish_mock, open_mock): open_mock.return_value = StringIO('') self.collector.collect() open_mock.assert_called_once_with('/proc/vmstat') @patch.object(Collector, 'publish') def test_should_work_with_real_data(self, publish_mock): VMStatCollector.PROC = self.getFixturePath('proc_vmstat_1') self.collector.collect() self.assertPublishedMany(publish_mock, {}) VMStatCollector.PROC = self.getFixturePath('proc_vmstat_2') self.collector.collect() metrics = { 'pgpgin': 0.0, 'pgpgout': 9.2, 'pswpin': 0.0, 'pswpout': 0.0, } self.setDocExample(collector=self.collector.__class__.__name__, metrics=metrics, defaultpath=self.collector.config['path']) self.assertPublishedMany(publish_mock, metrics)
class TestVMStatCollector(CollectorTestCase): def setUp(self): config = get_collector_config('VMStatCollector', { 'interval': 10 }) self.collector = VMStatCollector(config, None) @patch('__builtin__.open') @patch('os.access', Mock(return_value=True)) @patch.object(Collector, 'publish') def test_should_open_proc_vmstat(self, publish_mock, open_mock): open_mock.return_value = StringIO('') self.collector.collect() open_mock.assert_called_once_with('/proc/vmstat') @patch.object(Collector, 'publish') def test_should_work_with_real_data(self, publish_mock): VMStatCollector.PROC = self.getFixturePath('proc_vmstat_1') self.collector.collect() self.assertPublishedMany(publish_mock, {}) VMStatCollector.PROC = self.getFixturePath('proc_vmstat_2') self.collector.collect() self.assertPublishedMany(publish_mock, { 'pgpgin' : 0.0, 'pgpgout' : 9.2, 'pswpin' : 0.0, 'pswpout' : 0.0, })
class TestVMStatCollector(CollectorTestCase): def setUp(self): config = get_collector_config("VMStatCollector", {"interval": 10}) self.collector = VMStatCollector(config, None) @patch("__builtin__.open") @patch("os.access", Mock(return_value=True)) @patch.object(Collector, "publish") def test_should_open_proc_vmstat(self, publish_mock, open_mock): open_mock.return_value = StringIO("") self.collector.collect() open_mock.assert_called_once_with("/proc/vmstat") @patch.object(Collector, "publish") def test_should_work_with_real_data(self, publish_mock): VMStatCollector.PROC = self.getFixturePath("proc_vmstat_1") self.collector.collect() self.assertPublishedMany(publish_mock, {}) VMStatCollector.PROC = self.getFixturePath("proc_vmstat_2") self.collector.collect() metrics = {"pgpgin": 0.0, "pgpgout": 9.2, "pswpin": 0.0, "pswpout": 0.0} self.setDocExample( collector=self.collector.__class__.__name__, metrics=metrics, defaultpath=self.collector.config["path"] ) self.assertPublishedMany(publish_mock, metrics)
class TestVMStatCollector(CollectorTestCase): def setUp(self): config = get_collector_config('VMStatCollector', { 'interval': 10 }) self.collector = VMStatCollector(config, None) def test_import(self): self.assertTrue(VMStatCollector) @patch('__builtin__.open') @patch('os.access', Mock(return_value=True)) @patch.object(Collector, 'publish') def test_should_open_proc_vmstat(self, publish_mock, open_mock): open_mock.return_value = StringIO('') self.collector.collect() open_mock.assert_called_once_with('/proc/vmstat') @patch.object(Collector, 'publish') def test_should_work_with_real_data(self, publish_mock): VMStatCollector.PROC = self.getFixturePath('proc_vmstat_1') self.collector.collect() self.assertPublishedMany(publish_mock, {}) VMStatCollector.PROC = self.getFixturePath('proc_vmstat_2') self.collector.collect() metrics = { 'pgfault': 71.1, 'pgmajfault': 0.0, 'pgpgin': 0.0, 'pgpgout': 9.2, 'pswpin': 0.0, 'pswpout': 0.0, } self.setDocExample(collector=self.collector.__class__.__name__, metrics=metrics, defaultpath=self.collector.config['path']) self.assertPublishedMany(publish_mock, metrics)
def setUp(self): config = get_collector_config('VMStatCollector', { 'interval': 10 }) self.collector = VMStatCollector(config, None)
def setUp(self): config = get_collector_config('VMStatCollector', {'interval': 10}) self.collector = VMStatCollector(config, None)
def setUp(self): config = get_collector_config("VMStatCollector", {"interval": 10}) self.collector = VMStatCollector(config, None)