fix test config

Author: caroljung-cg

tests/unit/codegen/shared/configs/test_config.py

@@ -34,29 +34,30 @@ def test_merge_configs_none_values():
 
 
 def test_merge_configs_empty_string():
-    base = {"repository": {"organization_name": "org1"}}
-    override = {"repository": {"organization_name": ""}}
+    base = {"repository": {"full_name": "org1/test-repo"}}
+    override = {"repository": {"full_name": ""}}
     result = _merge_configs(base, override)
-    assert result == {"repository": {"organization_name": "org1"}}
+    assert result == {"repository": {"full_name": "org1/test-repo"}}
 
 
 # Test _load_from_toml
 def test_load_from_toml_existing_file(temp_config_file):
     config = _load_from_toml(temp_config_file)
     assert isinstance(config, SessionConfig)
     assert config.secrets.github_token == "gh_token123"
-    assert config.repository.organization_name == "test-org"
+    assert config.repository.repo_name == "test-repo"
     assert config.feature_flags.codebase.debug is True
     assert config.feature_flags.codebase.typescript.ts_dependency_manager is True
     assert config.feature_flags.codebase.import_resolution_overrides == {"@org/pkg": "./local/path"}
 
 
+@patch.dict("os.environ", {})
 @patch("codegen.shared.configs.models.SecretsConfig.model_config", {"env_file": "nonexistent.env"})
 def test_load_from_toml_nonexistent_file():
     config = _load_from_toml(Path("nonexistent.toml"))
     assert isinstance(config, SessionConfig)
     assert config.secrets.github_token is None
-    assert config.repository.organization_name is None
+    assert config.repository.full_name is None
     assert config.feature_flags.codebase.debug is None
 
 
@@ -76,15 +77,15 @@ def test_load_from_env():
 @patch("codegen.shared.configs.models.SecretsConfig.model_config", {"env_file": None, "env_prefix": "CODEGEN_SECRETS__"})
 def test_load_with_both_configs(mock_toml, mock_env):
     # Setup mock returns
-    mock_env.return_value = SessionConfig(secrets=SecretsConfig(github_token="env_token"), feature_flags=FeatureFlagsConfig(codebase=CodebaseFeatureFlags(debug=True)))
-    mock_toml.return_value = SessionConfig(secrets={"openai_api_key": "openai_key"}, repository={"organization_name": "codegen-org"})
+    mock_env.return_value = Config(secrets=SecretsConfig(github_token="env_token"), feature_flags=FeatureFlagsConfig(codebase=CodebaseFeatureFlags(debug=True)))
+    mock_toml.return_value = Config(secrets={"openai_api_key": "openai_key"}, repository={"full_name": "codegen-org/test-repo"})
 
     config = load_session_config(CONFIG_PATH)
 
     assert isinstance(config, SessionConfig)
     assert config.secrets.github_token == "env_token"
     assert config.secrets.openai_api_key == "openai_key"
-    assert config.repository.organization_name == "codegen-org"
+    assert config.repository.full_name == "codegen-org/test-repo"
     assert config.feature_flags.codebase.debug is True
 
 

tests/unit/codegen/shared/configs/test_models.py

@@ -12,7 +12,7 @@
 @pytest.fixture
 def sample_config(tmpdir):
     codebase_flags = CodebaseFeatureFlags(debug=True, verify_graph=False)
-    return SessionConfig(file_path=f"{tmpdir}/test_config.toml", repository=RepositoryConfig(full_name="test-org", repo_name="test-repo"), feature_flags=FeatureFlagsConfig(codebase=codebase_flags))
+    return Config(repository=RepositoryConfig(full_name="test-org/test-repo", repo_name="test-repo"), feature_flags=FeatureFlagsConfig(codebase=codebase_flags))
 
 
 def test_config_initialization(tmpdir):
@@ -22,9 +22,9 @@ def test_config_initialization(tmpdir):
     assert config.secrets is not None
 
 
-def test_config_with_values(tmpdir):
-    config = SessionConfig(file_path=f"{tmpdir}/test_config.toml", repository={"full_name": "test-org", "repo_name": "test-repo"})
-    assert config.repository.full_name == "test-org"
+def test_config_with_values():
+    config = Config(repository={"full_name": "test-org/test-repo", "repo_name": "test-repo"})
+    assert config.repository.full_name == "test-org/test-repo"
     assert config.repository.repo_name == "test-repo"
 
 
@@ -39,12 +39,12 @@ def test_save_config(mock_mkdir, mock_file, sample_config):
     # Verify the content being written
     written_data = mock_file().write.call_args[0][0]
     parsed_data = toml.loads(written_data)
-    assert parsed_data["repository"]["full_name"] == "test-org"
+    assert parsed_data["repository"]["full_name"] == "test-org/test-repo"
 
 
 def test_get_config_value(sample_config):
     # Test getting a simple value
-    assert json.loads(sample_config.get("repository.full_name")) == "test-org"
+    assert json.loads(sample_config.get("repository.full_name")) == "test-org/test-repo"
 
     # Test getting a nested value
     assert json.loads(sample_config.get("feature_flags.codebase.debug")) is True
@@ -57,8 +57,8 @@ def test_set_config_value(sample_config):
     # Instead of mocking save, we'll mock the open function used within save
     with patch("builtins.open", new_callable=mock_open) as mock_file:
         # Test setting a simple string value
-        sample_config.set("repository.full_name", "new-org")
-        assert sample_config.repository.full_name == "new-org"
+        sample_config.set("repository.full_name", "new-org/test-repo")
+        assert sample_config.repository.full_name == "new-org/test-repo"
 
         # Test setting a boolean value
         sample_config.set("feature_flags.codebase.debug", "false")
@@ -83,7 +83,7 @@ def test_config_str_representation(sample_config):
     assert isinstance(config_str, str)
     # Verify it's valid JSON
     parsed = json.loads(config_str)
-    assert parsed["repository"]["full_name"] == "test-org"
+    assert parsed["repository"]["full_name"] == "test-org/test-repo"
 
 
 def test_set_config_new_override_key(sample_config):