def testCollectMeminfoHandlerIgnoreWindows(self):
vm = mock.Mock()
vm.OS_TYPE = 'windows2019_desktop'
benchmark_spec = mock.Mock(vms=[vm])
samples = []
pkb._CollectMeminfoHandler(None, benchmark_spec, samples)
self.assertEmpty(samples)
def testCollectMeminfoHandlerDefault(self):
# must set --collect_meminfo to collect samples
vm = mock.Mock()
benchmark_spec = mock.Mock(vms=[vm])
samples = []
pkb._CollectMeminfoHandler(None, benchmark_spec, samples)
self.assertEmpty(samples)
vm.RemoteCommand.assert_not_called()
def testCollectMeminfoHandler(self):
vm = mock.Mock()
vm.RemoteCommand.return_value = 'b: 100\na: 10\nbadline', ''
vm.name = 'pkb-1234-0'
vm.OS_TYPE = 'ubuntu1804'
vm.machine_type = 'n1-standard-2'
benchmark_spec = mock.Mock(vms=[vm])
samples = []
pkb._CollectMeminfoHandler(None, benchmark_spec, samples)
expected_metadata = {
'a': 10,
'b': 100,
'meminfo_keys': 'a,b',
'meminfo_malformed': 'badline',
'meminfo_machine_type': 'n1-standard-2',
'meminfo_os_type': 'ubuntu1804',
'meminfo_vmname': 'pkb-1234-0',
}
expected_sample = sample.Sample('meminfo', 0, '', expected_metadata)
self.assertSampleListsEqualUpToTimestamp([expected_sample], samples)
vm.RemoteCommand.assert_called_with('cat /proc/meminfo')