diff --git a/tests/test_sync_deployment_tools.py b/tests/test_sync_deployment_tools.py index d5925e5..c770ddb 100644 --- a/tests/test_sync_deployment_tools.py +++ b/tests/test_sync_deployment_tools.py @@ -62,6 +62,7 @@ class DeploymentToolsTest(unittest.TestCase): self.assertEqual(result.returncode, 0, result.stderr) inventory = json.loads((output / "candidate-inventory.json").read_text()) self.assertEqual(inventory["product_version"], "0.1.2") + self.assertIn("session-preparations-v1", inventory["protocols"]) self.assertTrue((output / "bin/trainlog").is_file()) self.assertTrue((output / "bin/trainlog-sync-once").is_file()) self.assertTrue((output / "bin/trainlog-syncd").is_file()) diff --git a/tools/package_sync_candidate.py b/tools/package_sync_candidate.py index eb1ba47..d1ccf22 100644 --- a/tools/package_sync_candidate.py +++ b/tools/package_sync_candidate.py @@ -106,6 +106,7 @@ def main() -> int: "generation-manifest-v1", "generation-ack-v1", "generation-archive-v1", + "session-preparations-v1", ], "entry_points": [ "bin/trainlog",