test(schema): cover experimental v2 bindings

Author: PsiACE

tests/test_gen_all.py

@@ -2,6 +2,7 @@
 
 from acp.schema import ReadTextFileRequest
 from scripts.gen_all import resolve_ref, schema_source_paths
+from scripts.gen_meta import generate_meta
 from scripts.gen_schema import generate_schema
 
 
@@ -13,6 +14,7 @@ def test_generated_field_descriptions_are_introspectable() -> None:
 
 def test_resolve_ref_accepts_schema_release_tags() -> None:
     assert resolve_ref("schema-v1.16.0") == "refs/tags/schema-v1.16.0"
+    assert resolve_ref("schema-v2.0.0-alpha.3") == "refs/tags/schema-v2.0.0-alpha.3"
 
 
 def test_resolve_ref_keeps_legacy_version_tags() -> None:
@@ -27,6 +29,12 @@ def test_schema_release_tags_prefer_v1_schema_layout() -> None:
     )
 
 
+def test_v2_generation_uses_v2_schema_layout() -> None:
+    assert schema_source_paths("refs/tags/schema-v2.0.0-alpha.3", 2) == (
+        ("schema/v2/schema.unstable.json", "schema/v2/meta.unstable.json"),
+    )
+
+
 def test_legacy_tags_keep_legacy_schema_layout_first() -> None:
     assert schema_source_paths("refs/tags/v0.13.6")[0] == (
         "schema/schema.unstable.json",
@@ -49,8 +57,16 @@ def test_parse_args_can_skip_format(monkeypatch) -> None:
 
 
 def test_codegen_check_is_clean_and_read_only() -> None:
-    output = Path("src/acp/schema.py")
-    before = output.read_bytes()
+    outputs = (
+        Path("src/acp/meta.py"),
+        Path("src/acp/schema.py"),
+        Path("src/acp/experimental/v2/meta.py"),
+        Path("src/acp/experimental/v2/schema.py"),
+    )
+    before = {output: output.read_bytes() for output in outputs}
 
     assert generate_schema(check=True)
-    assert output.read_bytes() == before
+    assert generate_meta(check=True)
+    assert generate_schema(check=True, protocol_version=2)
+    assert generate_meta(check=True, protocol_version=2)
+    assert {output: output.read_bytes() for output in outputs} == before

tests/test_v2_schema.py

@@ -0,0 +1,48 @@
+import pytest
+from pydantic import ValidationError
+
+from acp.experimental.v2 import PROTOCOL_VERSION
+from acp.experimental.v2.schema import (
+    AgentMessageChunk,
+    OtherSessionUpdate,
+    TextContentBlock,
+    UpdateSessionNotification,
+)
+
+
+def test_v2_models_fill_protocol_discriminators() -> None:
+    update = AgentMessageChunk(
+        message_id="message-1",
+        content=TextContentBlock(text="hello"),
+    )
+
+    assert PROTOCOL_VERSION == 2
+    assert update.model_dump(by_alias=True, exclude_none=True) == {
+        "sessionUpdate": "agent_message_chunk",
+        "messageId": "message-1",
+        "content": {"type": "text", "text": "hello"},
+    }
+
+
+def test_v2_open_union_preserves_unknown_updates() -> None:
+    notification = UpdateSessionNotification.model_validate({
+        "sessionId": "session-1",
+        "update": {"sessionUpdate": "_vendor_status", "status": "waiting"},
+    })
+
+    assert isinstance(notification.update, OtherSessionUpdate)
+    assert notification.model_dump(by_alias=True, exclude_none=True)["update"] == {
+        "sessionUpdate": "_vendor_status",
+        "status": "waiting",
+    }
+
+
+def test_v2_open_union_rejects_malformed_known_updates() -> None:
+    with pytest.raises(ValidationError):
+        UpdateSessionNotification.model_validate({
+            "sessionId": "session-1",
+            "update": {
+                "sessionUpdate": "agent_message_chunk",
+                "content": {"type": "text", "text": "hello"},
+            },
+        })