def pin_vcpu_for_perf(cls, connection, vm_name, cpu, socket="0"):
threads = ""
sys_obj = CpuSysCores(connection)
soc_cpu = sys_obj.get_core_socket()
sys_cpu = int(soc_cpu["cores_per_socket"])
cores = "%s-%s" % (soc_cpu[socket][0], soc_cpu[socket][sys_cpu - 1])
if int(soc_cpu["thread_per_core"]):
threads = "%s-%s" % (soc_cpu[socket][sys_cpu], soc_cpu[socket][-1])
cpuset = "%s,%s" % (cores, threads)
return cpuset
def _get_app_cpu(self):
if self.CORES:
return self.CORES
vnf_cfg = self.scenario_helper.vnf_cfg
sys_obj = CpuSysCores(self.ssh_helper)
self.sys_cpu = sys_obj.get_core_socket()
num_core = int(vnf_cfg["worker_threads"])
if vnf_cfg.get("lb_config", "SW") == 'HW':
num_core += self.HW_DEFAULT_CORE
else:
num_core += self.SW_DEFAULT_CORE
app_cpu = self.sys_cpu[str(self.socket)][:num_core]
return app_cpu
def test_get_core_socket(self):
with mock.patch("yardstick.ssh.SSH") as ssh:
ssh_mock = mock.Mock(autospec=ssh.SSH)
ssh_mock.execute = \
mock.Mock(return_value=(1, "cpu:1\ntest:2\n \n", ""))
ssh_mock.put = \
mock.Mock(return_value=(1, "", ""))
cpu_topo = CpuSysCores(ssh_mock)
subprocess.check_output = mock.Mock(return_value=0)
cpu_topo._get_core_details = \
mock.Mock(side_effect=[[{'Core(s) per socket': '2', 'Thread(s) per core': '1'}],
[{'physical id': '2', 'processor': '1'}]])
self.assertEqual(
{
'thread_per_core': '1',
'2': ['1'],
'cores_per_socket': '2'
}, cpu_topo.get_core_socket())
