fix(test): use RockEnvironmentConfig instead of EnvironmentConfig in test

Author: xdlkc

rock/sdk/agent/models/job/config.py

@@ -139,16 +139,6 @@ class JobConfig(BaseModel):
     # ── Rock environment (Rock sandbox config + Harbor EnvironmentConfig, not serialized to Harbor YAML) ──
     environment: RockEnvironmentConfig = Field(default_factory=RockEnvironmentConfig)
 
-    @model_validator(mode="before")
-    @classmethod
-    def _coerce_environment(cls, values):
-        """Allow passing a plain EnvironmentConfig; upcast it to RockEnvironmentConfig."""
-        if isinstance(values, dict):
-            env = values.get("environment")
-            if isinstance(env, EnvironmentConfig) and not isinstance(env, RockEnvironmentConfig):
-                values["environment"] = RockEnvironmentConfig.model_validate(env.model_dump())
-        return values
-
     # ── Harbor native fields ──
     namespace: str | None = Field(
         default=None,

tests/unit/sdk/agent/test_oss_mirror.py

@@ -120,11 +120,13 @@ def test_namespace_at_top_level_in_yaml(self):
         from rock.sdk.agent.models.job.config import JobConfig
         from rock.sdk.agent.models.trial.config import OssMirrorConfig
 
+        from rock.sdk.agent.models.trial.config import RockEnvironmentConfig
+
         cfg = JobConfig(
             job_name="mirror-test",
             namespace="my-ns",
             experiment_id="exp-1",
-            environment=EnvironmentConfig(
+            environment=RockEnvironmentConfig(
                 oss_mirror=OssMirrorConfig(
                     enabled=True,
                     oss_bucket="test-bucket",