def test_on_activate_was_blank(self):
'''Ensure use_whole_segment is set when the disk was initially blank'''
self.disk.was_blank = True
disk_selection.on_activate(disk_info=self.disk,
disk_select=self.screen)
self.assertTrue(self.disk.use_whole_segment)
def test_on_activate_GPT(self):
'''Ensure use_whole_segment is set if the disk was GPT labeled'''
self.disk.label = [DiskInfo.GPT]
disk_selection.on_activate(disk_info=self.disk,
disk_select=self.screen)
self.assertTrue(self.disk.use_whole_segment)
def test_on_activate_default(self):
'''Ensure that do_copy flag is set after calls to on_activate'''
disk_selection.on_activate(disk_info=self.disk,
disk_select=self.screen)
self.assertFalse(self.disk.use_whole_segment)
self.assertTrue(self.screen.do_copy)