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

fix: Resolve TypeError in SSL verification tests by mocking Model class

Browse source
This commit is contained in:
Paul Gauthier (aider) 2025-03-05 18:42:03 -08:00
parent e5a85108d7
commit 85b9bdd8f4
2 changed files with 37 additions and 24 deletions
Download patch file Download diff file Expand all files Collapse all files

43
tests/basic/test_ssl_verification.py
View file

@ -19,7 +19,7 @@ class TestSSLVerification(TestCase):
os.environ.clear()
os.environ.update(self.original_env)
@patch("aider.models.model_info_manager.set_verify_ssl")
@patch("aider.models.ModelInfoManager.set_verify_ssl")
@patch("aider.llm.litellm._load_litellm")
@patch("httpx.Client")
@patch("httpx.AsyncClient")
@ -29,23 +29,30 @@ class TestSSLVerification(TestCase):
# Mock the litellm._lazy_module to avoid AttributeError
mock_load_litellm.return_value = None
mock_module = MagicMock()
with patch("aider.llm.litellm._lazy_module", mock_module):
# Run main with --no-verify-ssl flag
main(
["--no-verify-ssl", "--exit", "--yes"],
input=DummyInput(),
output=DummyOutput(),
)
# Verify model_info_manager.set_verify_ssl was called with False
mock_set_verify_ssl.assert_called_once_with(False)
# Verify httpx clients were created with verify=False
mock_client.assert_called_once_with(verify=False)
mock_async_client.assert_called_once_with(verify=False)
# Verify SSL_VERIFY environment variable was set to empty string
self.assertEqual(os.environ.get("SSL_VERIFY"), "")
# Mock Model class to avoid actual model initialization
with patch("aider.models.Model") as mock_model:
# Configure the mock to avoid the TypeError
mock_model.return_value.info = {}
mock_model.return_value.validate_environment.return_value = {"missing_keys": [], "keys_in_environment": []}
with patch("aider.llm.litellm._lazy_module", mock_module):
# Run main with --no-verify-ssl flag
main(
["--no-verify-ssl", "--exit", "--yes"],
input=DummyInput(),
output=DummyOutput(),
)
# Verify model_info_manager.set_verify_ssl was called with False
mock_set_verify_ssl.assert_called_once_with(False)
# Verify httpx clients were created with verify=False
mock_client.assert_called_once_with(verify=False)
mock_async_client.assert_called_once_with(verify=False)
# Verify SSL_VERIFY environment variable was set to empty string
self.assertEqual(os.environ.get("SSL_VERIFY"), "")
@patch("aider.models.model_info_manager.set_verify_ssl")
def test_default_ssl_verification(self, mock_set_verify_ssl):