Fix data tag related tests (#697)

Author: roverflow

changelogs/fragments/data_tag.yml

@@ -0,0 +1,3 @@
+---
+trivial:
+  - "Fixes unit tests execution by updating set_module_args to automatically patch module arguments using the new patch_module_args for ansible 2.19"

tests/unit/modules/utils.py

@@ -11,14 +11,36 @@
 from ansible.module_utils._text import to_bytes
 
 
+cur_context = None
+
+
 def set_module_args(args):
+    global cur_context
+
+    # Add common defaults
     if "_ansible_remote_tmp" not in args:
         args["_ansible_remote_tmp"] = "/tmp"
     if "_ansible_keep_remote_files" not in args:
         args["_ansible_keep_remote_files"] = False
 
-    args = json.dumps({"ANSIBLE_MODULE_ARGS": args})
-    basic._ANSIBLE_ARGS = to_bytes(args)
+    # Clean up any previous context manager if it exists
+    if cur_context is not None:
+        try:
+            cur_context.__exit__(None, None, None)
+        except Exception:
+            pass
+        cur_context = None
+
+    # Try to use the newer patch_module_args
+    try:
+        from ansible.module_utils.testing import patch_module_args
+
+        cur_context = patch_module_args(args)
+        cur_context.__enter__()
+    except ImportError:
+        # Fall back to original behavior for older Ansible versions
+        serialized_args = json.dumps({"ANSIBLE_MODULE_ARGS": args})
+        basic._ANSIBLE_ARGS = to_bytes(serialized_args)
 
 
 class AnsibleExitJson(Exception):