def pqos_l3ca_get_mock(socket, cos_num, num_ca_ref, l3cas):
"Mock pqos_l3ca_get()."
self.assertEqual(socket, 1)
self.assertEqual(cos_num, 2)
cos_c = CPqosL3Ca(class_id=0, u=CPqosL3CaMask(ways_mask=0x01fe))
cos2_c = CPqosL3Ca(class_id=1, u=CPqosL3CaMask(ways_mask=0x003f))
cos_arr_c = ctypes_build_array([cos_c, cos2_c])
ctypes.memmove(l3cas, cos_arr_c, ctypes.sizeof(cos_arr_c))
ctypes_ref_set_int(num_ca_ref, len(cos_arr_c))
return 0
def pqos_mba_get_mock(socket, max_num_cos, num_cos_ref, cos_arr):
"Mock pqos_mba_get()."
self.assertEqual(socket, 1)
self.assertEqual(max_num_cos, 2)
self.assertEqual(len(cos_arr), max_num_cos)
ctypes_ref_set_int(num_cos_ref, 2)
cos_arr[0].class_id = 0
cos_arr[0].mb_max = 4000
cos_arr[0].ctrl = 1
cos_arr[1].class_id = 1
cos_arr[1].mb_max = 8000
cos_arr[1].ctrl = 1
return 0
def pqos_mba_get_mock(socket, max_num_cos, num_cos_ref, cos_arr):
"Mock pqos_mba_get()."
self.assertEqual(socket, 1)
self.assertEqual(max_num_cos, 2)
self.assertEqual(len(cos_arr), max_num_cos)
ctypes_ref_set_int(num_cos_ref, 2)
cos_arr[0].class_id = 0
cos_arr[0].mb_max = 4000
cos_arr[0].ctrl = 1
cos_arr[1].class_id = 1
cos_arr[1].mb_max = 8000
cos_arr[1].ctrl = 1
return 0
def pqos_l3ca_get_min_cbm_bits_mock(min_cbm_bits_ref):
"Mock pqos_l3ca_get_min_cbm_bits()."
ctypes_ref_set_int(min_cbm_bits_ref, 2)
return 0
def pqos_mba_ct_enabled_m(_cap_ref, supported_ref, enabled_ref):
"Mock pqos_mba_ctrl_enabled()."
ctypes_ref_set_int(supported_ref, 1)
ctypes_ref_set_int(enabled_ref, -1)
return 0
def pqos_cores_l3id_m(_p_cpu, l3_id, count_ref):
"Mock pqos_cpu_get_cores_l3id()."
self.assertEqual(l3_id, 2)
ctypes_ref_set_int(count_ref, len(cores_arr))
return ctypes.cast(cores_arr, ctypes.POINTER(ctypes.c_uint))
def pqos_get_socketid_m(_p_cpu, core, socket_ref):
"Mock pqos_cpu_get_socketid()."
self.assertEqual(core, 3)
ctypes_ref_set_int(socket_ref, 4)
return 0
def pqos_get_one_core_m(_p_cpu, socket, core_ref):
"Mock pqos_cpu_get_one_core()."
self.assertEqual(socket, 1)
ctypes_ref_set_int(core_ref, 5)
return 0
def pqos_cores_l3id_m(_p_cpu, l3_id, count_ref):
"Mock pqos_cpu_get_cores_l3id()."
self.assertEqual(l3_id, 2)
ctypes_ref_set_int(count_ref, len(cores_arr))
return ctypes.cast(cores_arr, ctypes.POINTER(ctypes.c_uint))
def pqos_mba_cos_num_mock(_p_cap, num_cos_ref):
"Mock pqos_mba_get_cos_num()."
ctypes_ref_set_int(num_cos_ref, 2)
return 0
def pqos_get_clusterid_m(_p_cpu, core, cluster_ref):
"Mock pqos_cpu_get_clusterid()."
self.assertEqual(core, 1)
ctypes_ref_set_int(cluster_ref, 0)
return 0
def pqos_get_socketid_m(_p_cpu, core, socket_ref):
"Mock pqos_cpu_get_socketid()."
self.assertEqual(core, 3)
ctypes_ref_set_int(socket_ref, 4)
return 0
def pqos_get_by_l2id_m(_p_cpu, l2_id, core_ref):
"Mock pqos_cpu_get_one_core()."
self.assertEqual(l2_id, 8)
ctypes_ref_set_int(core_ref, 15)
return 0
def pqos_get_one_core_m(_p_cpu, socket, core_ref):
"Mock pqos_cpu_get_one_core()."
self.assertEqual(socket, 1)
ctypes_ref_set_int(core_ref, 5)
return 0
def pqos_cpu_get_cores_m(_p_cpu, socket, count_ref):
"Mock pqos_cpu_get_cores()."
self.assertEqual(socket, 0)
ctypes_ref_set_int(count_ref, len(cores_arr))
return ctypes.cast(cores_arr, ctypes.POINTER(ctypes.c_uint))
def pqos_mba_cos_num_m(_cap_ref, cos_num_ref):
"Mock pqos_mba_cos_num()."
ctypes_ref_set_int(cos_num_ref, 9)
return 0
def pqos_mba_cos_num_mock(_p_cap, num_cos_ref):
"Mock pqos_mba_get_cos_num()."
ctypes_ref_set_int(num_cos_ref, 2)
return 0
def pqos_l2cdp_enabled_m(_cap_ref, supported_ref, enabled_ref):
"Mock pqos_l2ca_cdp_enabled()."
ctypes_ref_set_int(supported_ref, 0)
ctypes_ref_set_int(enabled_ref, 1)
return 0
def pqos_cpu_get_l2ids_m(_p_cpu, count_ref):
"Mock pqos_cpu_get_l2ids()."
ctypes_ref_set_int(count_ref, len(l2ids_arr))
return ctypes.cast(l2ids_arr, ctypes.POINTER(ctypes.c_uint))
def pqos_mba_ct_enabled_m(_cap_ref, supported_ref, enabled_ref):
"Mock pqos_mba_ctrl_enabled()."
ctypes_ref_set_int(supported_ref, 1)
ctypes_ref_set_int(enabled_ref, -1)
return 0
def pqos_cpu_get_cores_m(_p_cpu, socket, count_ref):
"Mock pqos_cpu_get_cores()."
self.assertEqual(socket, 0)
ctypes_ref_set_int(count_ref, len(cores_arr))
return ctypes.cast(cores_arr, ctypes.POINTER(ctypes.c_uint))
def pqos_mba_cos_num_m(_cap_ref, cos_num_ref):
"Mock pqos_mba_cos_num()."
ctypes_ref_set_int(cos_num_ref, 9)
return 0
def pqos_get_by_l2id_m(_p_cpu, l2_id, core_ref):
"Mock pqos_cpu_get_one_core()."
self.assertEqual(l2_id, 8)
ctypes_ref_set_int(core_ref, 15)
return 0
def pqos_l2cdp_enabled_m(_cap_ref, supported_ref, enabled_ref):
"Mock pqos_l2ca_cdp_enabled()."
ctypes_ref_set_int(supported_ref, 0)
ctypes_ref_set_int(enabled_ref, 1)
return 0
def pqos_get_clusterid_m(_p_cpu, core, cluster_ref):
"Mock pqos_cpu_get_clusterid()."
self.assertEqual(core, 1)
ctypes_ref_set_int(cluster_ref, 0)
return 0
def pqos_cpu_get_l2ids_m(_p_cpu, count_ref):
"Mock pqos_cpu_get_l2ids()."
ctypes_ref_set_int(count_ref, len(l2ids_arr))
return ctypes.cast(l2ids_arr, ctypes.POINTER(ctypes.c_uint))
