def test_main_calls_zero_disk(self): self.patch_args(False, False) disks = { factory.make_name("disk").encode("ascii"): {} for _ in range(3) } self.patch(maas_wipe, "get_disk_info").return_value = disks zero_disk = self.patch(maas_wipe, "zero_disk") mock_try = self.patch(maas_wipe, "try_secure_erase") mock_try.return_value = False maas_wipe.main() wipe_calls = [call(disk, info) for disk, info in disks.items()] self.assertThat(mock_try, MockNotCalled()) self.assertThat(zero_disk, MockCallsMatch(*wipe_calls))
def test_main_calls_wipe_quickly(self): self.patch_args(False, True) disks = { factory.make_name("disk").encode("ascii"): {} for _ in range(3) } self.patch(maas_wipe, "get_disk_info").return_value = disks wipe_quickly = self.patch(maas_wipe, "wipe_quickly") mock_try = self.patch(maas_wipe, "try_secure_erase") mock_try.return_value = False maas_wipe.main() wipe_calls = [call(disk) for disk in disks.keys()] self.assertThat(mock_try, MockNotCalled()) self.assertThat(wipe_quickly, MockCallsMatch(*wipe_calls))