class TestKernel(unittest.TestCase):
def setUp(self):
self.memory = Memory(50)
self.scheduler = Scheduler()
self.scheduler.set_as_fifo()
self.a_kernel = Kernel(None)
self.pcb_table = PCBTable()
self.a_kernel.set_pcb_table(self.pcb_table)
interruption_manager = InterruptionHandler(self.a_kernel.cpu)
self.a_kernel.set_interruption_handler(interruption_manager)
load_in_interruption_manager = Handle_Loaders()
load_in_interruption_manager.load_handlers(self, interruption_manager)
def write_program(self,program,memory):
pos = 0
for instruction in program.obtain_instructions():
memory.put(pos,instruction)
def load_a_instruction_in_a_program(self):
program = Program("SIN-IO")
instruction = Instruction("Texto")
program.addInstruction(instruction)
program.addInstruction(instruction)
self.write_program(program,self.memory)
self.setup_load_of_a_program_in_memory(2,program,2)
def load_a_io_instruction_in_a_program(self):
program = Program("IO")
self.a_kernel.set_scheduler(self.scheduler)
instruction = InstructionIO()
program.addInstruction(instruction)
program.addInstruction(instruction)
self.write_program(program, self.memory)
self.setup_load_of_a_program_in_memory(2,program,2)
def setup_load_of_a_program_in_memory(self, amount_instructions, program, pcb_id):
block_holder = BlockHolder(program)
block_holder.set_representation([0,1])
pcb = PCB(amount_instructions, pcb_id, block_holder)
self.pcb_table.add(pcb)
self.scheduler.policy.add_pcb(pcb)
memory_admin = ToyMemoryAdmin(self.memory)
self.a_kernel.cpu.set_actual_pcb(pcb)
self.a_kernel.cpu.set_memory_manager(memory_admin)
def test_given_pcb_when_cpu_complete_instruction_cycle_then_increments_pc(self):
'''
Compare the initial state of PCB's PC with final state
'''
self.load_a_instruction_in_a_program()
self.assertEqual(0, self.a_kernel.cpu.actual_pcb.get_pc())
self.a_kernel.cpu.complete_instruction_cycle()
self.assertEqual(1, self.a_kernel.cpu.actual_pcb.get_pc())
def test_given_pcb_when_cpu_complete_instruction_cycle_then_IO_Manager_captures_interruption_and_increments_pc(self):
self.load_a_io_instruction_in_a_program()
self.assertEqual(0, self.a_kernel.cpu.actual_pcb.get_pc())
self.a_kernel.cpu.complete_instruction_cycle()
self.assertEqual(1, self.a_kernel.cpu.actual_pcb.get_pc())
class TestIOInterruption(unittest.TestCase):
def setUp(self):
self.kernel = Kernel(None)
self.kernel.to_user_mode()
self.kernel.scheduler = Scheduler(None)
self.kernel.scheduler.set_as_fifo()
self.interruption_manager = InterruptionHandler(self.kernel.cpu)
self.kernel.set_interruption_handler(self.interruption_manager)
load_in_interruption_manager = Handle_Loaders()
load_in_interruption_manager.load_handlers(self, self.interruption_manager)
self.memory = ToyMemory()
self.memory_manager = ToyMemoryAdmin(self.memory)
self.kernel.set_memory_manager(self.memory_manager)
def two_programs_in_ready_queue(self):
instruction_io = InstructionIO()
instruction = Instruction("text")
hold1 = [1,2,3,4]
hold2 = [5,6,7,8]
self.block_holder1 = BlockHolder(self.a_program_with_instruction(instruction_io))
self.block_holder1.set_representation(hold1)
self.pcb1 = PCB(1, 1, self.block_holder1)
self.block_holder2 = BlockHolder(self.a_program_with_instruction(instruction))
self.block_holder2.set_representation(hold2)
self.pcb2 = PCB(1, 2, self.block_holder2)
self.kernel.scheduler.push_to_queue(self.pcb1)
self.kernel.scheduler.push_to_queue(self.pcb2)
def a_program_with_instruction(self,instruction_io):
a_program = Program("P")
for i in range(0,5):
a_program.addInstruction(instruction_io)
self.memory.write_program(a_program)
return a_program
def test_when_a_process_is_io_then_goes_to_the_waiting_queue(self):
self.two_programs_in_ready_queue()
io_pcb = self.kernel.get_ready_queue._get()
self.kernel.cpu.set_actual_pcb(io_pcb)
without_io_pcb = self.kernel.get_ready_queue._get()
self.assertEqual(io_pcb.get_pid,self.kernel.cpu.actual_pcb.get_pid)
self.kernel.cpu.complete_instruction_cycle()
self.assertEqual(ProcessState.ProcessState.waiting, io_pcb.get_state)
self.kernel.cpu.set_actual_pcb(without_io_pcb)
self.assertEqual(without_io_pcb.get_pid,self.kernel.cpu.actual_pcb.get_pid)
without_io_pcb.set_state(ProcessState.ProcessState.running)
self.assertEqual(ProcessState.ProcessState.running, without_io_pcb.get_state)
