fix: type check for on_callback_created in executor test

bchampp · bchampp · commit c9ced6534cd6 · 2025-11-12T20:25:52.000-05:00
diff --git a/tests/executor_test.py b/tests/executor_test.py
@@ -78,12 +78,12 @@ def on_step_retry_scheduled(
         }
 
     def on_callback_created(
-        self, execution_arn: str, operation_id: str, callback_id: str
+        self, execution_arn: str, operation_id: str, callback_token: CallbackToken
     ) -> None:
         """Capture callback creation events."""
         self.callback_creations[execution_arn] = {
             "operation_id": operation_id,
-            "callback_id": callback_id,
+            "callback_id": callback_token.to_str(),
         }
 
     def on_callback_completed(

Original file line number	Diff line number	Diff line change
`@@ -78,12 +78,12 @@ def on_step_retry_scheduled(`
`78`	`78`	`}`
`79`	`79`
`80`	`80`	`def on_callback_created(`
`81`		`- self, execution_arn: str, operation_id: str, callback_id: str`
	`81`	`+ self, execution_arn: str, operation_id: str, callback_token: CallbackToken`
`82`	`82`	`) -> None:`
`83`	`83`	`"""Capture callback creation events."""`
`84`	`84`	`self.callback_creations[execution_arn] = {`
`85`	`85`	`"operation_id": operation_id,`
`86`		`- "callback_id": callback_id,`
	`86`	`+ "callback_id": callback_token.to_str(),`
`87`	`87`	`}`
`88`	`88`
`89`	`89`	`def on_callback_completed(`