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')