From 7538b3be9e59ba2a29f1db45346950bb9a9125b3 Mon Sep 17 00:00:00 2001 From: "Paul Gauthier (aider)" Date: Tue, 20 Aug 2024 10:57:44 -0700 Subject: [PATCH] feat: add test for return_coder parameter --- tests/basic/test_main.py | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) diff --git a/tests/basic/test_main.py b/tests/basic/test_main.py index 4b8921d8a..9575086eb 100644 --- a/tests/basic/test_main.py +++ b/tests/basic/test_main.py @@ -15,6 +15,7 @@ from aider.dump import dump # noqa: F401 from aider.io import InputOutput from aider.main import check_gitignore, main, setup_git from aider.utils import GitTemporaryDirectory, IgnorantTemporaryDirectory, make_repo +from aider.coders import Coder class TestMain(TestCase): @@ -555,3 +556,21 @@ class TestMain(TestCase): ) self.assertFalse(coder.add_cache_headers) + + def test_return_coder(self): + with GitTemporaryDirectory(): + result = main( + ["--exit", "--yes"], + input=DummyInput(), + output=DummyOutput(), + return_coder=True, + ) + self.assertIsInstance(result, Coder) + + result = main( + ["--exit", "--yes"], + input=DummyInput(), + output=DummyOutput(), + return_coder=False, + ) + self.assertIsNone(result)