"""Tests for gemini_helpers."""
import subprocess
import unittest
import unittest.mock
import gemini_helpers
class GetGeminiExecutableUnittest(unittest.TestCase):
"""Unit tests for the `get_gemini_executable` function."""
def setUp(self):
"""Sets up the mocks for the tests."""
which_patcher = unittest.mock.patch('shutil.which')
self.mock_which = which_patcher.start()
self.addCleanup(which_patcher.stop)
exists_patcher = unittest.mock.patch('pathlib.Path.exists')
self.mock_exists = exists_patcher.start()
self.addCleanup(exists_patcher.stop)
def test_get_gemini_executable_from_path(self):
"""Tests that the gemini executable is found in the PATH."""
self.mock_which.return_value = '/usr/bin/gemini'
self.assertEqual(gemini_helpers.get_gemini_executable(),
'/usr/bin/gemini')
def test_get_gemini_executable_from_fallback(self):
"""Tests finding the gemini executable in the fallback path."""
self.mock_which.return_value = None
self.mock_exists.return_value = True
self.assertEqual(gemini_helpers.get_gemini_executable(),
'/google/bin/releases/gemini-cli/tools/gemini')
def test_get_gemini_executable_not_found(self):
"""Tests the default gemini command is returned when not found."""
self.mock_which.return_value = None
self.mock_exists.return_value = False
self.assertEqual(gemini_helpers.get_gemini_executable(), 'gemini')
class GetGeminiVersionUnittest(unittest.TestCase):
"""Unit tests for the `get_gemini_version` function."""
def setUp(self):
"""Sets up the mocks for the tests."""
run_patcher = unittest.mock.patch('subprocess.run')
self.mock_run = run_patcher.start()
self.addCleanup(run_patcher.stop)
def test_get_gemini_version_succeeds(self):
"""Tests that the gemini version is correctly parsed."""
self.mock_run.return_value = unittest.mock.MagicMock(
stdout='gemini version 0.1.2\n', returncode=0)
self.assertEqual(gemini_helpers.get_gemini_version(), '0.1.2')
def test_get_gemini_version_fails_on_error(self):
"""Tests that None is returned when `gemini --version` fails."""
self.mock_run.side_effect = subprocess.CalledProcessError(1, 'gemini')
self.assertIsNone(gemini_helpers.get_gemini_version())
def test_get_gemini_version_fails_on_not_found(self):
"""Tests that None is returned when the gemini command is not found."""
self.mock_run.side_effect = FileNotFoundError
self.assertIsNone(gemini_helpers.get_gemini_version())
def test_get_gemini_version_fails_on_no_match(self):
"""Tests for None when the version is not in the output."""
self.mock_run.return_value = unittest.mock.MagicMock(
stdout='some other output\n', returncode=0)
self.assertIsNone(gemini_helpers.get_gemini_version())
if __name__ == '__main__':
unittest.main()
