def test_get_pids(self, lib):
"Tests get_pids() method."
pids_uint = [ctypes.c_uint(pid) for pid in [1000, 1500, 3000, 5600]]
pid_array = ctypes_build_array(pids_uint)
def pqos_pid_get_pid_assoc_m(class_id, count_ref):
"Mock pqos_pid_get_pid_assoc()."
self.assertEqual(class_id, 7)
ctypes_ref_set_uint(count_ref, len(pid_array))
return ctypes.cast(pid_array, ctypes.POINTER(ctypes.c_uint))
func_mock = MagicMock(side_effect=pqos_pid_get_pid_assoc_m)
lib.pqos_pid_get_pid_assoc = func_mock
alloc = PqosAlloc()
with patch('pqos.allocation.free_memory'):
pids = alloc.get_pids(7)
lib.pqos_pid_get_pid_assoc.assert_called_once()
self.assertEqual(len(pids), 4)
self.assertEqual(pids[0], 1000)
self.assertEqual(pids[1], 1500)
self.assertEqual(pids[2], 3000)
self.assertEqual(pids[3], 5600)
def test_get_pids(self, lib):
"Tests get_pids() method."
pids_uint = [ctypes.c_uint(pid) for pid in [1000, 1500, 3000, 5600]]
pid_array = ctypes_build_array(pids_uint)
def pqos_pid_get_pid_assoc_m(class_id, count_ref):
"Mock pqos_pid_get_pid_assoc()."
self.assertEqual(class_id, 7)
ctypes_ref_set_uint(count_ref, len(pid_array))
return ctypes.cast(pid_array, ctypes.POINTER(ctypes.c_uint))
func_mock = MagicMock(side_effect=pqos_pid_get_pid_assoc_m)
lib.pqos_pid_get_pid_assoc = func_mock
alloc = PqosAlloc()
with patch('pqos.allocation.free_memory'):
pids = alloc.get_pids(7)
lib.pqos_pid_get_pid_assoc.assert_called_once()
self.assertEqual(len(pids), 4)
self.assertEqual(pids[0], 1000)
self.assertEqual(pids[1], 1500)
self.assertEqual(pids[2], 3000)
self.assertEqual(pids[3], 5600)
# Initialize PQoS library
pqos = Pqos()
pqos.init("OS")
alloc = PqosAlloc()
# Get PID of the current process
pid = os.getpid()
# Associate process with COS 1
print('Associating process %d with COS 1...' % pid)
alloc.assoc_set_pid(pid, 1)
# Get all processes associated with COS 1
print('Processes associated with COS 1:')
print(alloc.get_pids(1))
# Release association
print('Releasing association for process %d...' % pid)
alloc.release_pid([pid])
# Current process should not be associated with COS 1
print('Processes associated with COS 1:')
print(alloc.get_pids(1))
print('Is PID %d associated with COS 1?' % pid)
print('Yes' if pid in alloc.get_pids(1) else 'No')
# Finalize PQoS library
pqos.fini()