From 1b22c92ae83eee2abf6948b1eb73f34a49256ca5 Mon Sep 17 00:00:00 2001
From: raoufnezhad <72685312+raoufnezhad@users.noreply.github.com>
Date: Sun, 30 Mar 2025 13:12:49 +0330
Subject: [PATCH] add mock input in test_proxmox_backup_schedule.py

---
 .../modules/test_proxmox_backup_schedule.py   | 50 ++++++++++---------
 1 file changed, 27 insertions(+), 23 deletions(-)

diff --git a/tests/unit/plugins/modules/test_proxmox_backup_schedule.py b/tests/unit/plugins/modules/test_proxmox_backup_schedule.py
index 0874854a8d..34bdae38f6 100644
--- a/tests/unit/plugins/modules/test_proxmox_backup_schedule.py
+++ b/tests/unit/plugins/modules/test_proxmox_backup_schedule.py
@@ -189,31 +189,35 @@ class TestProxmoxBackupScheduleModule(ModuleTestCase):
         assert result["msg"] == "missing required arguments: api_host, api_user, state"
 
     def test_update_vmid_in_backup(self):
-        with pytest.raises(AnsibleExitJson) as exc_info:
-            set_module_args({
-                'api_host': 'proxmoxhost',
-                'api_user': 'root@pam',
-                'api_password': 'supersecret',
-                'vm_name': 'test05',
-                'backup_id': 'backup-001',
-                'state': 'present'
-            })
-            self.module.main()
-        result = exc_info.value.args[0]
-        assert result['changed'] is True
+        with patch('builtins.input', return_value='mocked input'):
+            with pytest.raises(AnsibleExitJson) as exc_info:
+                set_module_args({
+                    'api_host': 'proxmoxhost',
+                    'api_user': 'root@pam',
+                    'api_password': 'supersecret',
+                    'vm_name': 'test05',
+                    'backup_id': 'backup-001',
+                    'state': 'present'
+                })
+                self.module.main()
+
+            result = exc_info.value.args[0]
+            assert result['changed'] is True
 
     def test_delete_vmid_from_backup(self):
-        with pytest.raises(AnsibleExitJson) as exc_info:
-            set_module_args({
-                'api_host': 'proxmoxhost',
-                'api_user': 'root@pam',
-                'api_password': 'supersecret',
-                'vm_id': '101',
-                'state': 'absent'
-            })
-            self.module.main()
-        result = exc_info.value.args[0]
-        assert result['changed'] is True
+        with patch('builtins.input', return_value='mocked input'):
+            with pytest.raises(AnsibleExitJson) as exc_info:
+                set_module_args({
+                    'api_host': 'proxmoxhost',
+                    'api_user': 'root@pam',
+                    'api_password': 'supersecret',
+                    'vm_id': '101',
+                    'state': 'absent'
+                })
+                self.module.main()
+
+            result = exc_info.value.args[0]
+            assert result['changed'] is True  
 
 
 if __name__ == '__main__':