fix: address release promotion review findings

1 files changed, 9 insertions, 2 deletions

diff --git a/scripts/test_changelog.py b/scripts/test_changelog.py
index 07897c25..2a24b241 100644
--- a/scripts/test_changelog.py
+++ b/scripts/test_changelog.py
@@ -9,6 +9,7 @@ from scripts.changelog import (
     ChangelogError,
     archived_releases_from_current_manifest,
     build_latest_json,
+    build_parser,
     canonicalize_manifest,
     DEFAULT_PRODUCT_ANNOUNCEMENT_PATH,
     default_release_assets,
@@ -368,7 +369,11 @@ class ChangelogScriptTests(unittest.TestCase):
             },
         )
 
-    def test_manifest_from_release_payload_uses_explicit_protocol(self) -> None:
+    def test_manifest_from_release_payload_uses_explicit_compatibility_versions(self) -> None:
+        args = build_parser().parse_args([
+            "verify-release-state", "--version", "0.1.1",
+            "--protocol", "42", "--endpoint-generation", "7",
+        ])
         manifest = manifest_from_release_payload(
             {
                 "tagName": "v0.1.1",
@@ -378,10 +383,12 @@ class ChangelogScriptTests(unittest.TestCase):
                 "assets": release_assets_with_digests(),
             },
             "0.1.1",
-            protocol=42,
+            protocol=args.protocol,
+            endpoint_generation=args.endpoint_generation,
         )
 
         self.assertEqual(manifest["protocol"], 42)
+        self.assertEqual(manifest["endpoint_generation"], 7)
 
     def test_manifest_from_release_payload_rejects_missing_digest(self) -> None:
         assets = release_assets_with_digests()