Mock config env variables (#3621)

Author: shubhamofbce

tests/unit/test_config.py

@@ -227,6 +227,7 @@ def test_env_overrides_compat_toml(monkeypatch, default_config, temp_toml_file):
     monkeypatch.setenv('WORKSPACE_BASE', 'UNDEFINED')
     monkeypatch.setenv('SANDBOX_TIMEOUT', '1000')
     monkeypatch.setenv('SANDBOX_USER_ID', '1002')
+    monkeypatch.delenv('LLM_MODEL', raising=False)
 
     load_from_toml(default_config, temp_toml_file)
 
@@ -279,6 +280,7 @@ def test_env_overrides_sandbox_toml(monkeypatch, default_config, temp_toml_file)
     monkeypatch.setenv('WORKSPACE_BASE', 'UNDEFINED')
     monkeypatch.setenv('SANDBOX_TIMEOUT', '1000')
     monkeypatch.setenv('SANDBOX_USER_ID', '1002')
+    monkeypatch.delenv('LLM_MODEL', raising=False)
 
     load_from_toml(default_config, temp_toml_file)
 
@@ -305,7 +307,7 @@ def test_env_overrides_sandbox_toml(monkeypatch, default_config, temp_toml_file)
     assert default_config.workspace_mount_path == os.getcwd() + '/UNDEFINED'
 
 
-def test_sandbox_config_from_toml(default_config, temp_toml_file):
+def test_sandbox_config_from_toml(monkeypatch, default_config, temp_toml_file):
     # Test loading configuration from a new-style TOML file
     with open(temp_toml_file, 'w', encoding='utf-8') as toml_file:
         toml_file.write(
@@ -322,7 +324,7 @@ def test_sandbox_config_from_toml(default_config, temp_toml_file):
 user_id = 1001
 """
         )
-
+    monkeypatch.setattr(os, 'environ', {})
     load_from_toml(default_config, temp_toml_file)
     load_from_env(default_config, os.environ)
     finalize_config(default_config)