From c0f57b5c620a5d13b17aaaf6c446208a9cd3ef44 Mon Sep 17 00:00:00 2001
From: Alexei Znamensky <103110+russoz@users.noreply.github.com>
Date: Sun, 2 Feb 2025 02:17:58 +1300
Subject: [PATCH] test helper: add support for check and diff modes (#9666)

---
 tests/unit/plugins/modules/helper.py          |  8 +++-
 tests/unit/plugins/modules/test_gio_mime.yaml | 46 ++++++++++++++++---
 2 files changed, 47 insertions(+), 7 deletions(-)

diff --git a/tests/unit/plugins/modules/helper.py b/tests/unit/plugins/modules/helper.py
index 2464f280de..4b0f17b362 100644
--- a/tests/unit/plugins/modules/helper.py
+++ b/tests/unit/plugins/modules/helper.py
@@ -72,7 +72,13 @@ class Helper(object):
             """
             Run unit tests for each test case in self.test_cases
             """
-            patch_ansible_module(test_case.input)
+            args = {}
+            args.update(test_case.input)
+            if test_case.flags.get("check"):
+                args["_ansible_check_mode"] = test_case.flags.get("check")
+            if test_case.flags.get("diff"):
+                args["_ansible_diff"] = test_case.flags.get("diff")
+            patch_ansible_module(args)
             self.runner.run(mocker, capfd, test_case)
 
         self.add_func_to_test_module("test_module", _test_module)
diff --git a/tests/unit/plugins/modules/test_gio_mime.yaml b/tests/unit/plugins/modules/test_gio_mime.yaml
index 6ed5af5d49..0d98f25503 100644
--- a/tests/unit/plugins/modules/test_gio_mime.yaml
+++ b/tests/unit/plugins/modules/test_gio_mime.yaml
@@ -39,8 +39,15 @@ test_cases:
     output:
       handler: google-chrome.desktop
       changed: true
+      stdout: Module executed in check mode
+      diff:
+        before:
+          handler: null
+        after:
+          handler: google-chrome.desktop
     flags:
-      skip: test helper does not support check mode yet
+      check: true
+      diff: true
     mocks:
       run_command:
         - command: [/testbin/gio, --version]
@@ -54,11 +61,6 @@ test_cases:
           out: ''
           err: >
             No default applications for “x-scheme-handler/http”
-        - command: [/testbin/gio, mime, x-scheme-handler/http, google-chrome.desktop]
-          environ: *env-def
-          rc: 0
-          out: "Set google-chrome.desktop as the default for x-scheme-handler/http\n"
-          err: ''
   - id: test_set_handler_idempot
     input:
       handler: google-chrome.desktop
@@ -89,3 +91,35 @@ test_cases:
               google-chrome.desktop
               firefox_firefox.desktop
           err: ''
+  - id: test_set_handler_idempot_check
+    input:
+      handler: google-chrome.desktop
+      mime_type: x-scheme-handler/http
+    output:
+      handler: google-chrome.desktop
+      changed: false
+    flags:
+      check: true
+    mocks:
+      run_command:
+        - command: [/testbin/gio, --version]
+          environ: *env-def
+          rc: 0
+          out: "2.80.0\n"
+          err: ''
+        - command: [/testbin/gio, mime, x-scheme-handler/http]
+          environ: *env-def
+          rc: 0
+          out: |
+            Default application for “x-scheme-handler/https”: google-chrome.desktop
+            Registered applications:
+              brave-browser.desktop
+              firefox.desktop
+              google-chrome.desktop
+              firefox_firefox.desktop
+            Recommended applications:
+              brave-browser.desktop
+              firefox.desktop
+              google-chrome.desktop
+              firefox_firefox.desktop
+          err: ''