class FrameManagerTest(unittest.TestCase):

    # Arrange
    def setUp(self):
        program = Program('Excel')
        pageHolder = PageHolder(program)
        self.frame1 = Frame(0, 9, -1)
        self.frame2 = Frame(1, 9, -1)
        self.frame3 = Frame(2, 9, -1)
        self.frame4 = Frame(3, 9, -1)
        self.frames = [self.frame1, self.frame2, self.frame3, self.frame4]
        self.hdd = HDD(50)
        self.frame_manager = FrameManager(self.frames, self.hdd)
        self.page_creator = PageCreator()
        self.pcb = PCB(0, 30, pageHolder)
        self.page_creator.create(self.pcb, 5)

    def test_whenIAssignAPCBTheFirstTime_ThenFrameManagerUsesTheFirstFrame(self):
        # Pages
        first_frame = self.frame_manager.get_frames()[0]
        self.frame_manager.map_page_to_frame(self.pcb)
        first_pcb_page = self.pcb.get_page_assigned_by_number(0)
        self.assertEquals(first_frame.get_page(), first_pcb_page)

    def test_whenTheFirstPCBPageIsUsedAndPCBIsAssigned_ThenFrameManagerUsesTheSecondFrame(self):
        # Pages
        first_pcb_page = self.pcb.get_page_assigned_by_number(0)
        second_pcb_page =self.pcb.get_page_assigned_by_number(1)

        self.frame_manager.map_page_to_frame(self.pcb)
        first_pcb_page.set_used()

        self.frame_manager.map_page_to_frame(self.pcb)
        second_frame = self.frame_manager.get_frames()[1]
        self.assertEquals(second_frame.get_page(), second_pcb_page)

    def test_whenAllFramesAreUsedAndPCBAssignsOneMorePage_ThenTheYoungestFrameGetsEmpty(self):
        # Pages
        first_pcb_page = self.pcb.get_page_assigned_by_number(0)
        second_pcb_page = self.pcb.get_page_assigned_by_number(1)
        assert (second_pcb_page != None)
        third_pcb_page = self.pcb.get_page_assigned_by_number(2)
        forth_pcb_page = self.pcb.get_page_assigned_by_number(3)
        fifth_pcb_page = self.pcb.get_page_assigned_by_number(4)

        self.frame_manager.map_page_to_frame(self.pcb)
        first_pcb_page.set_used()
        self.frame_manager.map_page_to_frame(self.pcb)
        second_pcb_page.set_used()
        self.frame_manager.map_page_to_frame(self.pcb)
        third_pcb_page.set_used()
        self.frame_manager.map_page_to_frame(self.pcb)
        forth_pcb_page.set_used()

        forth_frame = self.frame_manager.get_frames()[3]

        self.frame_manager.map_page_to_frame(self.pcb)
        self.assertEqual(forth_frame.get_page(), fifth_pcb_page)
class PageCreatorTest(unittest.TestCase):

    # Arrange
    def setUp(self):
        self.page_creator = PageCreator()
        self.program = Program("Duke Nukem",20)
        id = 0
        cantidadDeInstrucciones = 20
        self.pcb = PCB(id, cantidadDeInstrucciones,PageHolder(self.program))

    def test_whenICreatePagesForPCB_ThenTheInfoHolderHasThemAsItShould(self):
        self.page_creator.create(self.pcb, 5)
        self.assertEqual(len(self.pcb.get_memory_policy_for_pcb().get_pages()), 4)

    def test_whenICreatePagesForPCB_ThenTheInfoTheyHaveIsCorrect(self):
        self.page_creator.create(self.pcb, 5)

        page1 = self.pcb.get_page_assigned_by_number(0)
        page2 = self.pcb.get_page_assigned_by_number(1)
        page3 = self.pcb.get_page_assigned_by_number(2)
        page4 = self.pcb.get_page_assigned_by_number(3)

        self.assertEqual(page1.get_starting_index(), 0)
        self.assertEqual(page1.get_ending_index(), 4)
        self.assertEqual(page1.get_amount_of_instructions(), 5)

        self.assertEqual(page2.get_starting_index(), 5)
        self.assertEqual(page2.get_ending_index(), 9)
        self.assertEqual(page2.get_amount_of_instructions(), 5)

        self.assertEqual(page3.get_starting_index(), 10)
        self.assertEqual(page3.get_ending_index(), 14)
        self.assertEqual(page3.get_amount_of_instructions(), 5)

        self.assertEqual(page4.get_starting_index(), 15)
        self.assertEqual(page4.get_ending_index(), 19)
        self.assertEqual(page4.get_amount_of_instructions(), 5)