fix: Reset MODEL_SETTINGS between tests to prevent parameter leakage

This commit is contained in:
Paul Gauthier (aider) 2025-02-06 08:21:42 -08:00
parent 11a233da84
commit aef2b95d41

View file

@ -12,6 +12,17 @@ from aider.models import (
class TestModels(unittest.TestCase):
def setUp(self):
"""Reset MODEL_SETTINGS before each test"""
from aider.models import MODEL_SETTINGS
self._original_settings = MODEL_SETTINGS.copy()
def tearDown(self):
"""Restore original MODEL_SETTINGS after each test"""
from aider.models import MODEL_SETTINGS
MODEL_SETTINGS.clear()
MODEL_SETTINGS.extend(self._original_settings)
def test_get_model_info_nonexistent(self):
manager = ModelInfoManager()
info = manager.get_model_info("non-existent-model")