From 6f31450b90f6276eaf8c6cebe1a4974db8dc71bb Mon Sep 17 00:00:00 2001 From: Paul Gauthier Date: Thu, 27 Jun 2024 15:26:40 -0700 Subject: [PATCH] we don't want badreq errors bubbling out of Coder.run() --- aider/tests/test_coder.py | 24 ++---------------------- 1 file changed, 2 insertions(+), 22 deletions(-) diff --git a/aider/tests/test_coder.py b/aider/tests/test_coder.py index 2cd3bb9ba..ca3014ff9 100644 --- a/aider/tests/test_coder.py +++ b/aider/tests/test_coder.py @@ -1,16 +1,15 @@ import tempfile import unittest from pathlib import Path -from unittest.mock import MagicMock, patch +from unittest.mock import MagicMock import git -import openai from aider.coders import Coder from aider.dump import dump # noqa: F401 from aider.io import InputOutput from aider.models import Model -from aider.utils import ChdirTemporaryDirectory, GitTemporaryDirectory +from aider.utils import GitTemporaryDirectory class TestCoder(unittest.TestCase): @@ -330,25 +329,6 @@ class TestCoder(unittest.TestCase): # both files should still be here self.assertEqual(len(coder.abs_fnames), 2) - def test_run_with_invalid_request_error(self): - with ChdirTemporaryDirectory(): - # Mock the IO object - mock_io = MagicMock() - - # Initialize the Coder object with the mocked IO and mocked repo - coder = Coder.create(self.GPT35, None, mock_io) - - # Call the run method and assert that InvalidRequestError is raised - with self.assertRaises(openai.BadRequestError): - with patch("litellm.completion") as Mock: - Mock.side_effect = openai.BadRequestError( - message="Invalid request", - response=MagicMock(), - body=None, - ) - - coder.run(with_message="hi") - def test_new_file_edit_one_commit(self): """A new file shouldn't get pre-committed before the GPT edit commit""" with GitTemporaryDirectory():