import unittest
import subprocess
from unittest.mock import patch
from shared.cli_wrapper import CLIWrapper


class TestCLIWrapper(unittest.TestCase):

    @patch('subprocess.run')
    def test_execute_command_success(self, mock_run):
        mock_run.return_value.stdout = "Command output"
        mock_run.return_value.returncode = 0

        output, return_code = CLIWrapper.execute_command("test command")

        self.assertEqual(output, "Command output")
        self.assertEqual(return_code, 0)
        mock_run.assert_called_once_with("test command", shell=True, check=True, text=True, capture_output=True)

    @patch('subprocess.run')
    def test_execute_command_failure(self, mock_run):
        mock_run.side_effect = subprocess.CalledProcessError(1, "test command", output="Error output")

        output, return_code = CLIWrapper.execute_command("test command")

        self.assertEqual(output, "Error output")
        self.assertEqual(return_code, 1)
        mock_run.assert_called_once_with("test command", shell=True, check=True, text=True, capture_output=True)


if __name__ == '__main__':
    unittest.main()