[i-g-t 29/43] tests/kms_force_connector_basic: Add XE support

[Bhanuprakash Modem]

Add XE driver support for kms tests.

Signed-off-by: Bhanuprakash Modem <bhanuprakash.modem at intel.com>
---
 tests/kms_force_connector_basic.c | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/tests/kms_force_connector_basic.c b/tests/kms_force_connector_basic.c
index 4ff1efe3b..8c7e449a5 100644
--- a/tests/kms_force_connector_basic.c
+++ b/tests/kms_force_connector_basic.c
@@ -23,6 +23,7 @@
  */
 
 #include "igt.h"
+#include "xe/xe_query.h"
 
 IGT_TEST_DESCRIPTION("Check the debugfs force connector/edid features work"
 		     " correctly.");
@@ -364,6 +365,9 @@ igt_main_args("", long_opts, help_str, opt_handler, NULL)
 end:
 		/* Reacquire status after clearing any previous overrides. */
 		connector = drmModeGetConnector(drm_fd, connector_id);
+
+		if(is_xe_device(drm_fd))
+			xe_device_get(drm_fd);
 	}
 
 	igt_describe("Test to detect forced load.");
@@ -378,6 +382,8 @@ end:
 
 	igt_fixture {
 		drmModeFreeConnector(connector);
+		if(is_xe_device(drm_fd))
+			xe_device_put(drm_fd);
 		close(drm_fd);
 
 		reset_connectors();
-- 
2.40.0