Reset layout when aborting HW diagnostics

This commit is contained in:
2Shirt 2023-06-17 18:56:08 -07:00
parent 9810c630f6
commit 3e10f2cb8c
Signed by: 2Shirt
GPG key ID: 152FAC923B0E132C

View file

@ -100,8 +100,7 @@ class State():
test.set_status('Aborted')
# Cleanup panes
self.ui.remove_all_info_panes()
self.ui.remove_all_worker_panes()
self.reset_layout()
def disk_safety_checks(self) -> None:
"""Check for mid-run SMART failures and failed test(s)."""
@ -185,6 +184,12 @@ class State():
test_group.test_objects.append(test_obj)
self.test_groups.append(test_group)
def reset_layout(self) -> None:
"""Reset layout to avoid flickering."""
self.ui.clear_current_pane_height()
self.ui.remove_all_info_panes()
self.ui.remove_all_worker_panes()
def save_debug_reports(self) -> None:
"""Save debug reports to disk."""
LOG.info('Saving debug reports')
@ -788,6 +793,7 @@ def run_diags(state, menu, quick_mode=False, test_mode=False) -> None:
aborted = True
state.abort_testing()
state.update_progress_file()
state.reset_layout()
break
else:
# Run safety checks after disk tests