add tests

tests/pytests/unit/modules/test_win_task.py

@@ -7,6 +7,7 @@ from datetime import datetime
 import pytest
 
 import salt.modules.win_task as win_task
+from salt.exceptions import CommandExecutionError
 
 pytestmark = [
     pytest.mark.skip_unless_on_windows,
@@ -85,3 +86,93 @@ def test_edit_task_delete_after(base_task):
 
     result = win_task.info(base_task)
     assert result["settings"]["delete_after"] is False
+
+
+def test_execution_time_limit(base_task):
+    result = win_task.add_trigger(
+        base_task,
+        trigger_type="Daily",
+        trigger_enabled=True,
+        end_date=datetime.today().strftime("%Y-%m-%d"),
+        end_time="23:59:59",
+    )
+    assert result is True
+
+    result = win_task.edit_task(base_task, execution_time_limit="1 hour")
+    assert result is True
+
+    result = win_task.info(base_task)
+    assert result["settings"]["execution_time_limit"] == "1 hour"
+
+    result = win_task.edit_task(base_task, execution_time_limit=False)
+    assert result is True
+
+    result = win_task.info(base_task)
+    assert result["settings"]["execution_time_limit"] is False
+
+
+@pytest.mark.parametrize(
+    "exitcode, expect",
+    [
+        (0, "The operation completed successfully"),
+        (3221225786, "The application terminated as a result of CTRL+C"),
+        (4289449455, "Unknown Task Result: 0xffabcdef"),
+    ],
+)
+def test_run_result_code(exitcode, expect):
+    task_name = "SaltTest"
+    try:
+        result = win_task.create_task(
+            task_name,
+            user_name="System",
+            force=True,
+            action_type="Execute",
+            cmd="cmd.exe",
+            arguments=f"/c exit {exitcode}",
+        )
+        assert result is True
+
+        result = win_task.info(task_name)
+        assert result["last_run_result"] == "Task has not yet run"
+
+        result = win_task.run_wait(task_name)
+        assert result is True
+
+        result = win_task.info(task_name)
+        assert result["last_run_result"] == expect
+    finally:
+        result = win_task.delete_task(task_name)
+        assert result is True
+
+
+def test_create_task_from_xml():
+    task_name = "SaltTest"
+    task_xml = '<Task xmlns="http://schemas.microsoft.com/windows/2004/02/mit/task"><Actions><Exec><Command>cmd.exe</Command><Arguments>/c exit</Arguments></Exec></Actions></Task>'
+    try:
+        result = win_task.create_task_from_xml(
+            task_name, user_name="System", xml_text=task_xml
+        )
+        assert result is True
+
+        result = win_task.info(task_name)
+        assert result["actions"][0]["action_type"] == "Execute"
+        assert result["actions"][0]["cmd"] == "cmd.exe"
+        assert result["actions"][0]["arguments"] == "/c exit"
+
+    finally:
+        result = win_task.delete_task(task_name)
+        assert result is True
+
+
+def test_create_task_from_xml_error():
+    task_name = "SaltTest"
+    try:
+        with pytest.raises(CommandExecutionError) as excinfo:
+            result = win_task.create_task_from_xml(
+                task_name, user_name="System", xml_text="test"
+            )
+            assert result is False
+        assert "The task XML is malformed" in str(excinfo.value)
+    finally:
+        result = win_task.delete_task(task_name)
+        assert result is not True