dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-20 12:24:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

test: add test coverage for --set-env functionality

Browse source
This commit is contained in:
Paul Gauthier (aider) 2024-12-07 08:08:35 -08:00
parent 87a1469c1e
commit e10d06cb47
1 changed files with 30 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

30
tests/basic/test_main.py
View file

@ -672,6 +672,36 @@ class TestMain(TestCase):
# Verify that environment variables from pytest.ini are properly set
self.assertEqual(os.environ.get("AIDER_ANALYTICS"), "false")
def test_set_env_single(self):
# Test setting a single environment variable
with GitTemporaryDirectory():
main(["--set-env", "TEST_VAR=test_value", "--exit", "--yes"])
self.assertEqual(os.environ.get("TEST_VAR"), "test_value")
def test_set_env_multiple(self):
# Test setting multiple environment variables
with GitTemporaryDirectory():
main([
"--set-env", "TEST_VAR1=value1",
"--set-env", "TEST_VAR2=value2",
"--exit", "--yes"
])
self.assertEqual(os.environ.get("TEST_VAR1"), "value1")
self.assertEqual(os.environ.get("TEST_VAR2"), "value2")
def test_set_env_with_spaces(self):
# Test setting env var with spaces in value
with GitTemporaryDirectory():
main(["--set-env", "TEST_VAR=test value with spaces", "--exit", "--yes"])
self.assertEqual(os.environ.get("TEST_VAR"), "test value with spaces")
def test_set_env_invalid_format(self):
# Test invalid format handling
with GitTemporaryDirectory():
with self.assertRaises(SystemExit) as cm:
main(["--set-env", "INVALID_FORMAT", "--exit", "--yes"])
self.assertEqual(cm.exception.code, 1)
def test_invalid_edit_format(self):
with GitTemporaryDirectory():
with patch("aider.io.InputOutput.offer_url") as mock_offer_url: