igt/kms_vblank: Check if hangs are allowed

Before running a hang test, check if we can inject a hang and expect
recover to work. If we can't control a hang, skip the subtest.

Signed-off-by: Chris Wilson <chris@chris-wilson.co.uk>
Reviewed-by: Ville Syrjälä <ville.syrjala@linux.intel.com>

1 files changed, 4 insertions, 0 deletions

diff --git a/tests/kms_vblank.c b/tests/kms_vblank.c
index 2bee49de..508c0fa0 100644
--- a/tests/kms_vblank.c
+++ b/tests/kms_vblank.c
@@ -442,10 +442,14 @@ static void run_subtests_for_pipe(data_t *data)
 			igt_subtest_f("pipe-%s-%s-%s-hang",
 				      kmstest_pipe_name(data->pipe),
 				      f->name, m->name) {
+				igt_hang_t hang;
+
+				hang = igt_allow_hang(data->display.drm_fd, 0, 0);
 				for_each_valid_output_on_pipe(&data->display, data->pipe, data->output) {
 					data->flags = m->flags;
 					run_test(data, f->func);
 				}
+				igt_disallow_hang(data->display.drm_fd, hang);
 			}
 		}
 	}