[PATCH i-g-t 3/3] tests/intel/xe_pm: Fix S4 testing

[Lucas De Marchi <lucas.demarchi@intel.com>]

The additional s4-mocs implementation crossed with the change of how S4
is tested in commit 4b767566bbc6 ("tests/intel/xe_pm: S4 to go up to
devices only").  Apply the same logic as in other tests.

Fixes: 1a77ac56d453 ("tests/intel/xe_pm: Add mocs S2-idle/S3/S4 suspend resume test")
Signed-off-by: Lucas De Marchi <lucas.demarchi@intel.com>
---
 tests/intel/xe_pm.c | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/tests/intel/xe_pm.c b/tests/intel/xe_pm.c
index 584cb8548..3cbdb6a96 100644
--- a/tests/intel/xe_pm.c
+++ b/tests/intel/xe_pm.c
@@ -655,7 +655,10 @@ static void test_mocs_suspend_resume(device_t device, int s_state)
 			close(fw_handle);
 			igt_assert(igt_wait_for_pm_status(IGT_RUNTIME_PM_STATUS_SUSPENDED));
 		} else {
-			igt_system_suspend_autoresume(s_state, SUSPEND_TEST_NONE);
+			enum igt_suspend_test test = s_state == SUSPEND_STATE_DISK ?
+				SUSPEND_TEST_DEVICES : SUSPEND_TEST_NONE;
+
+			igt_system_suspend_autoresume(s_state, test);
 		}
 		igt_assert(igt_debugfs_exists(device.fd_xe, path, O_RDONLY));
 		igt_debugfs_dump(device.fd_xe, path);
-- 
2.43.0