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))
