class TestDiskSpaceCollector(CollectorTestCase): def setUp(self): config = get_collector_config('DiskSpaceCollector', { 'interval': 10 }) self.collector = DiskSpaceCollector(config, None) @patch.object(Collector, 'publish') def test_should_work_with_real_data(self, publish_mock): statvfs_mock = Mock() statvfs_mock.f_bsize = 4096 statvfs_mock.f_frsize = 4096 statvfs_mock.f_blocks = 360540255 statvfs_mock.f_bfree = 285953527 statvfs_mock.f_bavail = 267639130 statvfs_mock.f_files = 91578368 statvfs_mock.f_ffree = 91229495 statvfs_mock.f_favail = 91229495 statvfs_mock.f_flag = 4096 statvfs_mock.f_namemax = 255 with nested( patch('os.stat'), patch('os.major', return_value = 9), patch('os.minor', return_value = 0), patch('__builtin__.open', return_value = get_fixture('proc_mounts')) ): file_systems_mock = disk.get_file_systems() with nested( patch('disk.get_file_systems', return_value = file_systems_mock), patch('os.statvfs', return_value = statvfs_mock) ): self.collector.collect() self.assertPublishedMany(publish_mock, { 'root.gbytes_used' : ( 284.525, 2), 'root.gbytes_free' : (1090.826, 2), 'root.gbytes_avail' : (1020.962, 2), 'root.inodes_used' : 348873, 'root.inodes_free' : 91229495, 'root.inodes_avail' : 91229495 })
def setUp(self): config = get_collector_config('DiskSpaceCollector', { 'interval': 10 }) self.collector = DiskSpaceCollector(config, None)