@temporalio/core-bridge 1.13.1 → 1.14.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (422) hide show
  1. package/Cargo.lock +350 -436
  2. package/Cargo.toml +14 -13
  3. package/{sdk-core/fsm/rustfsm_procmacro/LICENSE.txt → LICENSE} +5 -5
  4. package/README.md +0 -1
  5. package/bridge-macros/Cargo.toml +16 -0
  6. package/bridge-macros/src/derive_js_function.rs +126 -0
  7. package/bridge-macros/src/derive_tryfromjs.rs +138 -0
  8. package/bridge-macros/src/derive_tryintojs.rs +151 -0
  9. package/bridge-macros/src/lib.rs +42 -0
  10. package/lib/native.d.ts +23 -9
  11. package/package.json +6 -5
  12. package/releases/aarch64-apple-darwin/index.node +0 -0
  13. package/releases/aarch64-unknown-linux-gnu/index.node +0 -0
  14. package/releases/x86_64-apple-darwin/index.node +0 -0
  15. package/releases/x86_64-pc-windows-msvc/index.node +0 -0
  16. package/releases/x86_64-unknown-linux-gnu/index.node +0 -0
  17. package/sdk-core/.cargo/config.toml +78 -12
  18. package/sdk-core/.clippy.toml +1 -0
  19. package/sdk-core/.github/workflows/heavy.yml +2 -0
  20. package/sdk-core/.github/workflows/per-pr.yml +90 -21
  21. package/sdk-core/AGENTS.md +17 -19
  22. package/sdk-core/ARCHITECTURE.md +44 -48
  23. package/sdk-core/Cargo.toml +25 -7
  24. package/sdk-core/README.md +16 -15
  25. package/sdk-core/arch_docs/diagrams/TimerMachine_Coverage.puml +14 -0
  26. package/sdk-core/arch_docs/diagrams/deps.svg +102 -0
  27. package/sdk-core/arch_docs/diagrams/initial_event_history.png +0 -0
  28. package/sdk-core/arch_docs/sdks_intro.md +299 -0
  29. package/sdk-core/{client → crates/client}/Cargo.toml +15 -16
  30. package/sdk-core/{client → crates/client}/src/callback_based.rs +1 -2
  31. package/sdk-core/{client → crates/client}/src/lib.rs +594 -457
  32. package/sdk-core/{client → crates/client}/src/metrics.rs +32 -8
  33. package/sdk-core/crates/client/src/proxy.rs +209 -0
  34. package/sdk-core/{client → crates/client}/src/raw.rs +648 -328
  35. package/sdk-core/crates/client/src/replaceable.rs +253 -0
  36. package/sdk-core/crates/client/src/request_extensions.rs +40 -0
  37. package/sdk-core/{client → crates/client}/src/retry.rs +32 -22
  38. package/sdk-core/crates/client/src/worker/mod.rs +1468 -0
  39. package/sdk-core/{client → crates/client}/src/workflow_handle/mod.rs +24 -21
  40. package/sdk-core/{core-api → crates/common}/Cargo.toml +21 -12
  41. package/sdk-core/{sdk-core-protos → crates/common}/build.rs +10 -23
  42. package/sdk-core/crates/common/protos/api_cloud_upstream/VERSION +1 -0
  43. package/sdk-core/{sdk-core-protos → crates/common}/protos/api_cloud_upstream/temporal/api/cloud/account/v1/message.proto +18 -0
  44. package/sdk-core/{sdk-core-protos → crates/common}/protos/api_cloud_upstream/temporal/api/cloud/cloudservice/v1/request_response.proto +38 -11
  45. package/sdk-core/{sdk-core-protos → crates/common}/protos/api_cloud_upstream/temporal/api/cloud/cloudservice/v1/service.proto +21 -4
  46. package/sdk-core/{sdk-core-protos → crates/common}/protos/api_cloud_upstream/temporal/api/cloud/operation/v1/message.proto +6 -6
  47. package/sdk-core/{sdk-core-protos → crates/common}/protos/api_cloud_upstream/temporal/api/cloud/sink/v1/message.proto +22 -0
  48. package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/.github/workflows/create-release.yml +13 -0
  49. package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/openapi/openapiv2.json +254 -5
  50. package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/openapi/openapiv3.yaml +234 -5
  51. package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/temporal/api/common/v1/message.proto +1 -1
  52. package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/temporal/api/deployment/v1/message.proto +6 -0
  53. package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/temporal/api/namespace/v1/message.proto +6 -2
  54. package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/temporal/api/workflowservice/v1/request_response.proto +60 -2
  55. package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/temporal/api/workflowservice/v1/service.proto +30 -6
  56. package/sdk-core/{sdk-core-protos → crates/common}/protos/local/temporal/sdk/core/workflow_activation/workflow_activation.proto +2 -0
  57. package/sdk-core/{core-api → crates/common}/src/envconfig.rs +155 -56
  58. package/sdk-core/{core-api → crates/common}/src/errors.rs +8 -1
  59. package/sdk-core/{fsm/rustfsm_trait/src/lib.rs → crates/common/src/fsm_trait.rs} +1 -27
  60. package/sdk-core/{core-api → crates/common}/src/lib.rs +92 -9
  61. package/sdk-core/{test-utils/src → crates/common/src/protos}/canned_histories.rs +5 -5
  62. package/sdk-core/{sdk-core-protos/src → crates/common/src/protos}/history_builder.rs +3 -3
  63. package/sdk-core/{sdk-core-protos/src → crates/common/src/protos}/history_info.rs +2 -2
  64. package/sdk-core/{sdk-core-protos/src/lib.rs → crates/common/src/protos/mod.rs} +43 -26
  65. package/sdk-core/crates/common/src/protos/test_utils.rs +89 -0
  66. package/sdk-core/{sdk-core-protos/src → crates/common/src/protos}/utilities.rs +14 -5
  67. package/sdk-core/{core-api → crates/common}/src/telemetry/metrics.rs +449 -51
  68. package/sdk-core/{core-api → crates/common}/src/telemetry.rs +16 -1
  69. package/sdk-core/{core-api → crates/common}/src/worker.rs +112 -18
  70. package/sdk-core/crates/common/tests/worker_task_types_test.rs +129 -0
  71. package/sdk-core/crates/macros/Cargo.toml +23 -0
  72. package/sdk-core/{fsm/rustfsm_procmacro → crates/macros}/src/lib.rs +12 -12
  73. package/sdk-core/{sdk → crates/sdk}/Cargo.toml +18 -12
  74. package/sdk-core/{sdk → crates/sdk}/src/activity_context.rs +9 -7
  75. package/sdk-core/{sdk → crates/sdk}/src/app_data.rs +1 -1
  76. package/sdk-core/{sdk → crates/sdk}/src/interceptors.rs +2 -5
  77. package/sdk-core/{sdk → crates/sdk}/src/lib.rs +43 -42
  78. package/sdk-core/{sdk → crates/sdk}/src/workflow_context/options.rs +12 -3
  79. package/sdk-core/{sdk → crates/sdk}/src/workflow_context.rs +21 -19
  80. package/sdk-core/{sdk → crates/sdk}/src/workflow_future.rs +2 -2
  81. package/sdk-core/crates/sdk-core/Cargo.toml +222 -0
  82. package/sdk-core/{core/benches/workflow_replay.rs → crates/sdk-core/benches/workflow_replay_bench.rs} +15 -10
  83. package/sdk-core/crates/sdk-core/machine_coverage/ActivityMachine_Coverage.puml +32 -0
  84. package/sdk-core/crates/sdk-core/machine_coverage/CancelExternalMachine_Coverage.puml +9 -0
  85. package/sdk-core/crates/sdk-core/machine_coverage/CancelWorkflowMachine_Coverage.puml +6 -0
  86. package/sdk-core/crates/sdk-core/machine_coverage/ChildWorkflowMachine_Coverage.puml +27 -0
  87. package/sdk-core/crates/sdk-core/machine_coverage/CompleteWorkflowMachine_Coverage.puml +6 -0
  88. package/sdk-core/crates/sdk-core/machine_coverage/ContinueAsNewWorkflowMachine_Coverage.puml +6 -0
  89. package/sdk-core/crates/sdk-core/machine_coverage/FailWorkflowMachine_Coverage.puml +6 -0
  90. package/sdk-core/crates/sdk-core/machine_coverage/LocalActivityMachine_Coverage.puml +23 -0
  91. package/sdk-core/crates/sdk-core/machine_coverage/ModifyWorkflowPropertiesMachine_Coverage.puml +5 -0
  92. package/sdk-core/crates/sdk-core/machine_coverage/PatchMachine_Coverage.puml +8 -0
  93. package/sdk-core/crates/sdk-core/machine_coverage/SignalExternalMachine_Coverage.puml +12 -0
  94. package/sdk-core/crates/sdk-core/machine_coverage/TimerMachine_Coverage.puml +13 -0
  95. package/sdk-core/crates/sdk-core/machine_coverage/UpdateMachine_Coverage.puml +19 -0
  96. package/sdk-core/crates/sdk-core/machine_coverage/UpsertSearchAttributesMachine_Coverage.puml +5 -0
  97. package/sdk-core/crates/sdk-core/machine_coverage/WorkflowTaskMachine_Coverage.puml +11 -0
  98. package/sdk-core/{core → crates/sdk-core}/src/abstractions.rs +62 -6
  99. package/sdk-core/crates/sdk-core/src/antithesis.rs +60 -0
  100. package/sdk-core/{core → crates/sdk-core}/src/core_tests/activity_tasks.rs +85 -250
  101. package/sdk-core/{core → crates/sdk-core}/src/core_tests/mod.rs +13 -16
  102. package/sdk-core/{core → crates/sdk-core}/src/core_tests/queries.rs +25 -24
  103. package/sdk-core/{core → crates/sdk-core}/src/core_tests/replay_flag.rs +11 -68
  104. package/sdk-core/{core → crates/sdk-core}/src/core_tests/updates.rs +21 -20
  105. package/sdk-core/{core → crates/sdk-core}/src/core_tests/workers.rs +242 -16
  106. package/sdk-core/{core → crates/sdk-core}/src/core_tests/workflow_cancels.rs +10 -7
  107. package/sdk-core/{core → crates/sdk-core}/src/core_tests/workflow_tasks.rs +150 -326
  108. package/sdk-core/{core → crates/sdk-core}/src/debug_client.rs +1 -1
  109. package/sdk-core/{core → crates/sdk-core}/src/ephemeral_server/mod.rs +18 -6
  110. package/sdk-core/{test-utils → crates/sdk-core}/src/histfetch.rs +9 -4
  111. package/sdk-core/{core → crates/sdk-core}/src/internal_flags.rs +15 -4
  112. package/sdk-core/{core → crates/sdk-core}/src/lib.rs +128 -72
  113. package/sdk-core/{core → crates/sdk-core}/src/pollers/mod.rs +13 -11
  114. package/sdk-core/{core → crates/sdk-core}/src/pollers/poll_buffer.rs +53 -18
  115. package/sdk-core/{core → crates/sdk-core}/src/protosext/mod.rs +14 -6
  116. package/sdk-core/{core → crates/sdk-core}/src/protosext/protocol_messages.rs +5 -12
  117. package/sdk-core/{core → crates/sdk-core}/src/replay/mod.rs +14 -11
  118. package/sdk-core/crates/sdk-core/src/retry_logic.rs +390 -0
  119. package/sdk-core/{core → crates/sdk-core}/src/telemetry/log_export.rs +2 -2
  120. package/sdk-core/{core → crates/sdk-core}/src/telemetry/metrics.rs +80 -33
  121. package/sdk-core/{core → crates/sdk-core}/src/telemetry/mod.rs +12 -6
  122. package/sdk-core/{core → crates/sdk-core}/src/telemetry/otel.rs +1 -1
  123. package/sdk-core/{core → crates/sdk-core}/src/telemetry/prometheus_meter.rs +14 -14
  124. package/sdk-core/{core → crates/sdk-core}/src/telemetry/prometheus_server.rs +2 -2
  125. package/sdk-core/{core/src/test_help/mod.rs → crates/sdk-core/src/test_help/integ_helpers.rs} +284 -329
  126. package/sdk-core/crates/sdk-core/src/test_help/mod.rs +13 -0
  127. package/sdk-core/crates/sdk-core/src/test_help/unit_helpers.rs +220 -0
  128. package/sdk-core/{core → crates/sdk-core}/src/worker/activities/activity_heartbeat_manager.rs +44 -8
  129. package/sdk-core/{core → crates/sdk-core}/src/worker/activities/local_activities.rs +33 -31
  130. package/sdk-core/{core → crates/sdk-core}/src/worker/activities.rs +31 -15
  131. package/sdk-core/{core → crates/sdk-core}/src/worker/client/mocks.rs +28 -15
  132. package/sdk-core/{core → crates/sdk-core}/src/worker/client.rs +275 -145
  133. package/sdk-core/crates/sdk-core/src/worker/heartbeat.rs +246 -0
  134. package/sdk-core/crates/sdk-core/src/worker/mod.rs +1462 -0
  135. package/sdk-core/{core → crates/sdk-core}/src/worker/nexus.rs +24 -12
  136. package/sdk-core/{core → crates/sdk-core}/src/worker/slot_provider.rs +18 -10
  137. package/sdk-core/{core → crates/sdk-core}/src/worker/tuner/fixed_size.rs +5 -1
  138. package/sdk-core/{core → crates/sdk-core}/src/worker/tuner/resource_based.rs +453 -57
  139. package/sdk-core/{core → crates/sdk-core}/src/worker/tuner.rs +179 -3
  140. package/sdk-core/{core → crates/sdk-core}/src/worker/workflow/driven_workflow.rs +1 -1
  141. package/sdk-core/{core → crates/sdk-core}/src/worker/workflow/history_update.rs +5 -267
  142. package/sdk-core/{core → crates/sdk-core}/src/worker/workflow/machines/activity_state_machine.rs +147 -90
  143. package/sdk-core/{core → crates/sdk-core}/src/worker/workflow/machines/cancel_external_state_machine.rs +8 -91
  144. package/sdk-core/{core → crates/sdk-core}/src/worker/workflow/machines/cancel_nexus_op_state_machine.rs +14 -12
  145. package/sdk-core/crates/sdk-core/src/worker/workflow/machines/cancel_workflow_state_machine.rs +97 -0
  146. package/sdk-core/{core → crates/sdk-core}/src/worker/workflow/machines/child_workflow_state_machine.rs +32 -229
  147. package/sdk-core/{core → crates/sdk-core}/src/worker/workflow/machines/complete_workflow_state_machine.rs +13 -8
  148. package/sdk-core/{core → crates/sdk-core}/src/worker/workflow/machines/continue_as_new_workflow_state_machine.rs +6 -50
  149. package/sdk-core/{core → crates/sdk-core}/src/worker/workflow/machines/fail_workflow_state_machine.rs +6 -7
  150. package/sdk-core/{core → crates/sdk-core}/src/worker/workflow/machines/local_activity_state_machine.rs +39 -584
  151. package/sdk-core/{core → crates/sdk-core}/src/worker/workflow/machines/mod.rs +22 -17
  152. package/sdk-core/{core → crates/sdk-core}/src/worker/workflow/machines/modify_workflow_properties_state_machine.rs +4 -74
  153. package/sdk-core/{core → crates/sdk-core}/src/worker/workflow/machines/nexus_operation_state_machine.rs +118 -21
  154. package/sdk-core/crates/sdk-core/src/worker/workflow/machines/patch_state_machine.rs +273 -0
  155. package/sdk-core/{core → crates/sdk-core}/src/worker/workflow/machines/signal_external_state_machine.rs +9 -151
  156. package/sdk-core/{core → crates/sdk-core}/src/worker/workflow/machines/timer_state_machine.rs +27 -132
  157. package/sdk-core/{core → crates/sdk-core}/src/worker/workflow/machines/transition_coverage.rs +1 -2
  158. package/sdk-core/{core → crates/sdk-core}/src/worker/workflow/machines/update_state_machine.rs +19 -13
  159. package/sdk-core/{core → crates/sdk-core}/src/worker/workflow/machines/upsert_search_attributes_state_machine.rs +25 -80
  160. package/sdk-core/{core → crates/sdk-core}/src/worker/workflow/machines/workflow_machines/local_acts.rs +1 -1
  161. package/sdk-core/{core → crates/sdk-core}/src/worker/workflow/machines/workflow_machines.rs +68 -72
  162. package/sdk-core/{core → crates/sdk-core}/src/worker/workflow/machines/workflow_task_state_machine.rs +13 -15
  163. package/sdk-core/{core → crates/sdk-core}/src/worker/workflow/managed_run.rs +55 -37
  164. package/sdk-core/{core → crates/sdk-core}/src/worker/workflow/mod.rs +171 -61
  165. package/sdk-core/{core → crates/sdk-core}/src/worker/workflow/run_cache.rs +10 -7
  166. package/sdk-core/{core → crates/sdk-core}/src/worker/workflow/wft_extraction.rs +4 -2
  167. package/sdk-core/{core → crates/sdk-core}/src/worker/workflow/wft_poller.rs +15 -4
  168. package/sdk-core/{core → crates/sdk-core}/src/worker/workflow/workflow_stream.rs +38 -19
  169. package/sdk-core/crates/sdk-core/tests/c_bridge_smoke_test.c +10 -0
  170. package/sdk-core/crates/sdk-core/tests/cloud_tests.rs +25 -0
  171. package/sdk-core/crates/sdk-core/tests/common/fake_grpc_server.rs +106 -0
  172. package/sdk-core/crates/sdk-core/tests/common/http_proxy.rs +134 -0
  173. package/sdk-core/{test-utils/src/lib.rs → crates/sdk-core/tests/common/mod.rs} +286 -334
  174. package/sdk-core/{test-utils/src → crates/sdk-core/tests/common}/workflows.rs +6 -4
  175. package/sdk-core/crates/sdk-core/tests/fsm_procmacro.rs +6 -0
  176. package/sdk-core/{fsm/rustfsm_procmacro/tests/trybuild → crates/sdk-core/tests/fsm_trybuild}/dupe_transitions_fail.rs +1 -3
  177. package/sdk-core/crates/sdk-core/tests/fsm_trybuild/dupe_transitions_fail.stderr +12 -0
  178. package/sdk-core/{fsm/rustfsm_procmacro/tests/trybuild → crates/sdk-core/tests/fsm_trybuild}/dynamic_dest_pass.rs +2 -4
  179. package/sdk-core/{fsm/rustfsm_procmacro/tests/trybuild → crates/sdk-core/tests/fsm_trybuild}/forgot_name_fail.rs +1 -3
  180. package/sdk-core/{fsm/rustfsm_procmacro/tests/trybuild → crates/sdk-core/tests/fsm_trybuild}/forgot_name_fail.stderr +4 -4
  181. package/sdk-core/{fsm/rustfsm_procmacro/tests/trybuild → crates/sdk-core/tests/fsm_trybuild}/handler_arg_pass.rs +2 -4
  182. package/sdk-core/{fsm/rustfsm_procmacro/tests/trybuild → crates/sdk-core/tests/fsm_trybuild}/handler_pass.rs +2 -4
  183. package/sdk-core/{fsm/rustfsm_procmacro/tests/trybuild → crates/sdk-core/tests/fsm_trybuild}/medium_complex_pass.rs +2 -4
  184. package/sdk-core/{fsm/rustfsm_procmacro/tests/trybuild → crates/sdk-core/tests/fsm_trybuild}/no_handle_conversions_require_into_fail.rs +2 -4
  185. package/sdk-core/crates/sdk-core/tests/fsm_trybuild/no_handle_conversions_require_into_fail.stderr +15 -0
  186. package/sdk-core/{fsm/rustfsm_procmacro/tests/trybuild → crates/sdk-core/tests/fsm_trybuild}/simple_pass.rs +2 -4
  187. package/sdk-core/{fsm/rustfsm_procmacro/tests/trybuild → crates/sdk-core/tests/fsm_trybuild}/struct_event_variant_fail.rs +1 -3
  188. package/sdk-core/crates/sdk-core/tests/fsm_trybuild/struct_event_variant_fail.stderr +5 -0
  189. package/sdk-core/{fsm/rustfsm_procmacro/tests/trybuild → crates/sdk-core/tests/fsm_trybuild}/tuple_more_item_event_variant_fail.rs +1 -3
  190. package/sdk-core/crates/sdk-core/tests/fsm_trybuild/tuple_more_item_event_variant_fail.stderr +5 -0
  191. package/sdk-core/{fsm/rustfsm_procmacro/tests/trybuild → crates/sdk-core/tests/fsm_trybuild}/tuple_zero_item_event_variant_fail.rs +1 -3
  192. package/sdk-core/crates/sdk-core/tests/fsm_trybuild/tuple_zero_item_event_variant_fail.stderr +5 -0
  193. package/sdk-core/{tests → crates/sdk-core/tests}/global_metric_tests.rs +22 -22
  194. package/sdk-core/{tests → crates/sdk-core/tests/heavy_tests}/fuzzy_workflow.rs +4 -4
  195. package/sdk-core/{tests → crates/sdk-core/tests}/heavy_tests.rs +25 -14
  196. package/sdk-core/{tests → crates/sdk-core/tests}/integ_tests/activity_functions.rs +1 -1
  197. package/sdk-core/{tests → crates/sdk-core/tests}/integ_tests/client_tests.rs +115 -123
  198. package/sdk-core/{tests → crates/sdk-core/tests}/integ_tests/ephemeral_server_tests.rs +19 -15
  199. package/sdk-core/{tests → crates/sdk-core/tests}/integ_tests/heartbeat_tests.rs +24 -20
  200. package/sdk-core/{tests → crates/sdk-core/tests}/integ_tests/metrics_tests.rs +241 -67
  201. package/sdk-core/crates/sdk-core/tests/integ_tests/pagination_tests.rs +273 -0
  202. package/sdk-core/crates/sdk-core/tests/integ_tests/polling_tests.rs +513 -0
  203. package/sdk-core/{tests → crates/sdk-core/tests}/integ_tests/queries_tests.rs +13 -11
  204. package/sdk-core/{tests → crates/sdk-core/tests}/integ_tests/update_tests.rs +47 -29
  205. package/sdk-core/{tests → crates/sdk-core/tests}/integ_tests/visibility_tests.rs +49 -27
  206. package/sdk-core/crates/sdk-core/tests/integ_tests/worker_heartbeat_tests.rs +1061 -0
  207. package/sdk-core/crates/sdk-core/tests/integ_tests/worker_tests.rs +920 -0
  208. package/sdk-core/{tests → crates/sdk-core/tests}/integ_tests/worker_versioning_tests.rs +58 -40
  209. package/sdk-core/{tests → crates/sdk-core/tests}/integ_tests/workflow_tests/activities.rs +268 -30
  210. package/sdk-core/{tests → crates/sdk-core/tests}/integ_tests/workflow_tests/appdata_propagation.rs +4 -4
  211. package/sdk-core/crates/sdk-core/tests/integ_tests/workflow_tests/cancel_external.rs +155 -0
  212. package/sdk-core/crates/sdk-core/tests/integ_tests/workflow_tests/cancel_wf.rs +121 -0
  213. package/sdk-core/crates/sdk-core/tests/integ_tests/workflow_tests/child_workflows.rs +717 -0
  214. package/sdk-core/crates/sdk-core/tests/integ_tests/workflow_tests/continue_as_new.rs +109 -0
  215. package/sdk-core/{core/src/core_tests → crates/sdk-core/tests/integ_tests/workflow_tests}/determinism.rs +108 -20
  216. package/sdk-core/{tests → crates/sdk-core/tests}/integ_tests/workflow_tests/eager.rs +10 -5
  217. package/sdk-core/crates/sdk-core/tests/integ_tests/workflow_tests/local_activities.rs +2884 -0
  218. package/sdk-core/crates/sdk-core/tests/integ_tests/workflow_tests/modify_wf_properties.rs +119 -0
  219. package/sdk-core/{tests → crates/sdk-core/tests}/integ_tests/workflow_tests/nexus.rs +174 -46
  220. package/sdk-core/crates/sdk-core/tests/integ_tests/workflow_tests/patches.rs +740 -0
  221. package/sdk-core/{tests → crates/sdk-core/tests}/integ_tests/workflow_tests/replay.rs +78 -27
  222. package/sdk-core/{tests → crates/sdk-core/tests}/integ_tests/workflow_tests/resets.rs +46 -28
  223. package/sdk-core/crates/sdk-core/tests/integ_tests/workflow_tests/signals.rs +294 -0
  224. package/sdk-core/{tests → crates/sdk-core/tests}/integ_tests/workflow_tests/stickyness.rs +9 -8
  225. package/sdk-core/crates/sdk-core/tests/integ_tests/workflow_tests/timers.rs +249 -0
  226. package/sdk-core/{tests → crates/sdk-core/tests}/integ_tests/workflow_tests/upsert_search_attrs.rs +72 -7
  227. package/sdk-core/{tests → crates/sdk-core/tests}/integ_tests/workflow_tests.rs +97 -33
  228. package/sdk-core/{tests → crates/sdk-core/tests}/main.rs +40 -25
  229. package/sdk-core/{tests → crates/sdk-core/tests}/manual_tests.rs +25 -15
  230. package/sdk-core/{tests → crates/sdk-core/tests}/runner.rs +23 -41
  231. package/sdk-core/{tests → crates/sdk-core/tests}/shared_tests/mod.rs +11 -6
  232. package/sdk-core/{tests → crates/sdk-core/tests}/shared_tests/priority.rs +6 -6
  233. package/sdk-core/crates/sdk-core-c-bridge/Cargo.toml +59 -0
  234. package/sdk-core/{core-c-bridge → crates/sdk-core-c-bridge}/include/temporal-sdk-core-c-bridge.h +215 -44
  235. package/sdk-core/{core-c-bridge → crates/sdk-core-c-bridge}/src/client.rs +515 -214
  236. package/sdk-core/crates/sdk-core-c-bridge/src/envconfig.rs +314 -0
  237. package/sdk-core/{core-c-bridge → crates/sdk-core-c-bridge}/src/lib.rs +2 -1
  238. package/sdk-core/{core-c-bridge → crates/sdk-core-c-bridge}/src/metric.rs +1 -1
  239. package/sdk-core/{core-c-bridge → crates/sdk-core-c-bridge}/src/random.rs +4 -4
  240. package/sdk-core/{core-c-bridge → crates/sdk-core-c-bridge}/src/runtime.rs +42 -28
  241. package/sdk-core/{core-c-bridge → crates/sdk-core-c-bridge}/src/testing.rs +2 -5
  242. package/sdk-core/{core-c-bridge → crates/sdk-core-c-bridge}/src/tests/context.rs +42 -41
  243. package/sdk-core/{core-c-bridge → crates/sdk-core-c-bridge}/src/tests/mod.rs +38 -34
  244. package/sdk-core/{core-c-bridge → crates/sdk-core-c-bridge}/src/tests/utils.rs +10 -11
  245. package/sdk-core/{core-c-bridge → crates/sdk-core-c-bridge}/src/worker.rs +427 -121
  246. package/sdk-core/docker-cgroup-tests.sh +24 -0
  247. package/sdk-core/{docker → etc/docker}/docker-compose-ci.yaml +9 -9
  248. package/sdk-core/{docker → etc/docker}/docker-compose-telem.yaml +11 -11
  249. package/sdk-core/{docker → etc/docker}/docker-compose.yaml +8 -8
  250. package/sdk-core/{integ-with-otel.sh → etc/integ-with-otel.sh} +1 -1
  251. package/sdk-core/etc/regen-depgraph.sh +2 -2
  252. package/src/client.rs +117 -49
  253. package/src/helpers/callbacks.rs +4 -4
  254. package/src/helpers/errors.rs +7 -1
  255. package/src/helpers/handles.rs +1 -0
  256. package/src/helpers/try_from_js.rs +5 -4
  257. package/src/lib.rs +3 -2
  258. package/src/logs.rs +1 -1
  259. package/src/metrics.rs +6 -3
  260. package/src/runtime.rs +41 -24
  261. package/src/testing.rs +3 -3
  262. package/src/worker.rs +77 -44
  263. package/ts/native.ts +26 -9
  264. package/LICENSE.md +0 -23
  265. package/sdk-core/arch_docs/diagrams/workflow_internals.svg +0 -1
  266. package/sdk-core/client/src/proxy.rs +0 -90
  267. package/sdk-core/client/src/worker_registry/mod.rs +0 -266
  268. package/sdk-core/core/Cargo.toml +0 -146
  269. package/sdk-core/core/src/core_tests/child_workflows.rs +0 -281
  270. package/sdk-core/core/src/core_tests/local_activities.rs +0 -1442
  271. package/sdk-core/core/src/retry_logic.rs +0 -224
  272. package/sdk-core/core/src/worker/heartbeat.rs +0 -231
  273. package/sdk-core/core/src/worker/mod.rs +0 -980
  274. package/sdk-core/core/src/worker/workflow/machines/cancel_workflow_state_machine.rs +0 -165
  275. package/sdk-core/core/src/worker/workflow/machines/patch_state_machine.rs +0 -801
  276. package/sdk-core/core-c-bridge/Cargo.toml +0 -51
  277. package/sdk-core/etc/deps.svg +0 -162
  278. package/sdk-core/fsm/Cargo.toml +0 -21
  279. package/sdk-core/fsm/README.md +0 -3
  280. package/sdk-core/fsm/rustfsm_procmacro/Cargo.toml +0 -27
  281. package/sdk-core/fsm/rustfsm_procmacro/tests/progress.rs +0 -8
  282. package/sdk-core/fsm/rustfsm_procmacro/tests/trybuild/dupe_transitions_fail.stderr +0 -12
  283. package/sdk-core/fsm/rustfsm_procmacro/tests/trybuild/no_handle_conversions_require_into_fail.stderr +0 -15
  284. package/sdk-core/fsm/rustfsm_procmacro/tests/trybuild/struct_event_variant_fail.stderr +0 -5
  285. package/sdk-core/fsm/rustfsm_procmacro/tests/trybuild/tuple_more_item_event_variant_fail.stderr +0 -5
  286. package/sdk-core/fsm/rustfsm_procmacro/tests/trybuild/tuple_zero_item_event_variant_fail.stderr +0 -5
  287. package/sdk-core/fsm/rustfsm_trait/Cargo.toml +0 -14
  288. package/sdk-core/fsm/rustfsm_trait/LICENSE.txt +0 -21
  289. package/sdk-core/fsm/src/lib.rs +0 -2
  290. package/sdk-core/sdk-core-protos/Cargo.toml +0 -37
  291. package/sdk-core/sdk-core-protos/protos/api_cloud_upstream/VERSION +0 -1
  292. package/sdk-core/test-utils/Cargo.toml +0 -38
  293. package/sdk-core/test-utils/src/interceptors.rs +0 -46
  294. package/sdk-core/tests/cloud_tests.rs +0 -23
  295. package/sdk-core/tests/integ_tests/polling_tests.rs +0 -294
  296. package/sdk-core/tests/integ_tests/worker_tests.rs +0 -203
  297. package/sdk-core/tests/integ_tests/workflow_tests/cancel_external.rs +0 -74
  298. package/sdk-core/tests/integ_tests/workflow_tests/cancel_wf.rs +0 -57
  299. package/sdk-core/tests/integ_tests/workflow_tests/child_workflows.rs +0 -246
  300. package/sdk-core/tests/integ_tests/workflow_tests/continue_as_new.rs +0 -65
  301. package/sdk-core/tests/integ_tests/workflow_tests/determinism.rs +0 -85
  302. package/sdk-core/tests/integ_tests/workflow_tests/local_activities.rs +0 -908
  303. package/sdk-core/tests/integ_tests/workflow_tests/modify_wf_properties.rs +0 -51
  304. package/sdk-core/tests/integ_tests/workflow_tests/patches.rs +0 -206
  305. package/sdk-core/tests/integ_tests/workflow_tests/signals.rs +0 -164
  306. package/sdk-core/tests/integ_tests/workflow_tests/timers.rs +0 -123
  307. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_cloud_upstream/.github/workflows/build.yaml +0 -0
  308. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_cloud_upstream/.github/workflows/push-to-buf.yml +0 -0
  309. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_cloud_upstream/CODEOWNERS +0 -0
  310. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_cloud_upstream/LICENSE +0 -0
  311. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_cloud_upstream/Makefile +0 -0
  312. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_cloud_upstream/README.md +0 -0
  313. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_cloud_upstream/buf.gen.yaml +0 -0
  314. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_cloud_upstream/buf.lock +0 -0
  315. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_cloud_upstream/buf.yaml +0 -0
  316. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_cloud_upstream/temporal/api/cloud/connectivityrule/v1/message.proto +0 -0
  317. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_cloud_upstream/temporal/api/cloud/identity/v1/message.proto +0 -0
  318. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_cloud_upstream/temporal/api/cloud/namespace/v1/message.proto +0 -0
  319. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_cloud_upstream/temporal/api/cloud/nexus/v1/message.proto +0 -0
  320. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_cloud_upstream/temporal/api/cloud/region/v1/message.proto +0 -0
  321. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_cloud_upstream/temporal/api/cloud/resource/v1/message.proto +0 -0
  322. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_cloud_upstream/temporal/api/cloud/usage/v1/message.proto +0 -0
  323. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/.github/CODEOWNERS +0 -0
  324. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/.github/PULL_REQUEST_TEMPLATE.md +0 -0
  325. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/.github/workflows/ci.yml +0 -0
  326. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/.github/workflows/publish-docs.yml +0 -0
  327. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/.github/workflows/push-to-buf.yml +0 -0
  328. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/.github/workflows/trigger-api-go-delete-release.yml +0 -0
  329. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/.github/workflows/trigger-api-go-publish-release.yml +0 -0
  330. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/.github/workflows/trigger-api-go-update.yml +0 -0
  331. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/LICENSE +0 -0
  332. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/Makefile +0 -0
  333. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/README.md +0 -0
  334. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/api-linter.yaml +0 -0
  335. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/buf.gen.yaml +0 -0
  336. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/buf.lock +0 -0
  337. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/buf.yaml +0 -0
  338. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/google/api/annotations.proto +0 -0
  339. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/google/api/http.proto +0 -0
  340. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/google/protobuf/any.proto +0 -0
  341. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/google/protobuf/descriptor.proto +0 -0
  342. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/google/protobuf/duration.proto +0 -0
  343. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/google/protobuf/empty.proto +0 -0
  344. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/google/protobuf/struct.proto +0 -0
  345. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/google/protobuf/timestamp.proto +0 -0
  346. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/google/protobuf/wrappers.proto +0 -0
  347. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/openapi/payload_description.txt +0 -0
  348. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/temporal/api/activity/v1/message.proto +0 -0
  349. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/temporal/api/batch/v1/message.proto +0 -0
  350. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/temporal/api/command/v1/message.proto +0 -0
  351. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/temporal/api/enums/v1/batch_operation.proto +0 -0
  352. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/temporal/api/enums/v1/command_type.proto +0 -0
  353. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/temporal/api/enums/v1/common.proto +0 -0
  354. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/temporal/api/enums/v1/deployment.proto +0 -0
  355. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/temporal/api/enums/v1/event_type.proto +0 -0
  356. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/temporal/api/enums/v1/failed_cause.proto +0 -0
  357. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/temporal/api/enums/v1/namespace.proto +0 -0
  358. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/temporal/api/enums/v1/nexus.proto +0 -0
  359. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/temporal/api/enums/v1/query.proto +0 -0
  360. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/temporal/api/enums/v1/reset.proto +0 -0
  361. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/temporal/api/enums/v1/schedule.proto +0 -0
  362. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/temporal/api/enums/v1/task_queue.proto +0 -0
  363. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/temporal/api/enums/v1/update.proto +0 -0
  364. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/temporal/api/enums/v1/workflow.proto +0 -0
  365. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/temporal/api/errordetails/v1/message.proto +0 -0
  366. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/temporal/api/export/v1/message.proto +0 -0
  367. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/temporal/api/failure/v1/message.proto +0 -0
  368. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/temporal/api/filter/v1/message.proto +0 -0
  369. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/temporal/api/history/v1/message.proto +0 -0
  370. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/temporal/api/nexus/v1/message.proto +0 -0
  371. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/temporal/api/operatorservice/v1/request_response.proto +0 -0
  372. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/temporal/api/operatorservice/v1/service.proto +0 -0
  373. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/temporal/api/protocol/v1/message.proto +0 -0
  374. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/temporal/api/query/v1/message.proto +0 -0
  375. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/temporal/api/replication/v1/message.proto +0 -0
  376. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/temporal/api/rules/v1/message.proto +0 -0
  377. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/temporal/api/schedule/v1/message.proto +0 -0
  378. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/temporal/api/sdk/v1/enhanced_stack_trace.proto +0 -0
  379. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/temporal/api/sdk/v1/task_complete_metadata.proto +0 -0
  380. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/temporal/api/sdk/v1/user_metadata.proto +0 -0
  381. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/temporal/api/sdk/v1/worker_config.proto +0 -0
  382. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/temporal/api/sdk/v1/workflow_metadata.proto +0 -0
  383. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/temporal/api/taskqueue/v1/message.proto +0 -0
  384. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/temporal/api/update/v1/message.proto +0 -0
  385. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/temporal/api/version/v1/message.proto +0 -0
  386. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/temporal/api/worker/v1/message.proto +0 -0
  387. /package/sdk-core/{sdk-core-protos → crates/common}/protos/api_upstream/temporal/api/workflow/v1/message.proto +0 -0
  388. /package/sdk-core/{sdk-core-protos → crates/common}/protos/google/rpc/status.proto +0 -0
  389. /package/sdk-core/{sdk-core-protos → crates/common}/protos/grpc/health/v1/health.proto +0 -0
  390. /package/sdk-core/{sdk-core-protos → crates/common}/protos/local/temporal/sdk/core/activity_result/activity_result.proto +0 -0
  391. /package/sdk-core/{sdk-core-protos → crates/common}/protos/local/temporal/sdk/core/activity_task/activity_task.proto +0 -0
  392. /package/sdk-core/{sdk-core-protos → crates/common}/protos/local/temporal/sdk/core/child_workflow/child_workflow.proto +0 -0
  393. /package/sdk-core/{sdk-core-protos → crates/common}/protos/local/temporal/sdk/core/common/common.proto +0 -0
  394. /package/sdk-core/{sdk-core-protos → crates/common}/protos/local/temporal/sdk/core/core_interface.proto +0 -0
  395. /package/sdk-core/{sdk-core-protos → crates/common}/protos/local/temporal/sdk/core/external_data/external_data.proto +0 -0
  396. /package/sdk-core/{sdk-core-protos → crates/common}/protos/local/temporal/sdk/core/nexus/nexus.proto +0 -0
  397. /package/sdk-core/{sdk-core-protos → crates/common}/protos/local/temporal/sdk/core/workflow_commands/workflow_commands.proto +0 -0
  398. /package/sdk-core/{sdk-core-protos → crates/common}/protos/local/temporal/sdk/core/workflow_completion/workflow_completion.proto +0 -0
  399. /package/sdk-core/{sdk-core-protos → crates/common}/protos/testsrv_upstream/Makefile +0 -0
  400. /package/sdk-core/{sdk-core-protos → crates/common}/protos/testsrv_upstream/api-linter.yaml +0 -0
  401. /package/sdk-core/{sdk-core-protos → crates/common}/protos/testsrv_upstream/buf.yaml +0 -0
  402. /package/sdk-core/{sdk-core-protos → crates/common}/protos/testsrv_upstream/temporal/api/testservice/v1/request_response.proto +0 -0
  403. /package/sdk-core/{sdk-core-protos → crates/common}/protos/testsrv_upstream/temporal/api/testservice/v1/service.proto +0 -0
  404. /package/sdk-core/{sdk-core-protos/src → crates/common/src/protos}/constants.rs +0 -0
  405. /package/sdk-core/{sdk-core-protos/src → crates/common/src/protos}/task_token.rs +0 -0
  406. /package/sdk-core/{fsm → crates/macros}/LICENSE.txt +0 -0
  407. /package/sdk-core/{core → crates/sdk-core}/src/abstractions/take_cell.rs +0 -0
  408. /package/sdk-core/{core → crates/sdk-core}/src/worker/slot_supplier.rs +0 -0
  409. /package/sdk-core/{histories → crates/sdk-core/tests/histories}/ends_empty_wft_complete.bin +0 -0
  410. /package/sdk-core/{histories → crates/sdk-core/tests/histories}/evict_while_la_running_no_interference-16_history.bin +0 -0
  411. /package/sdk-core/{histories → crates/sdk-core/tests/histories}/evict_while_la_running_no_interference-23_history.bin +0 -0
  412. /package/sdk-core/{histories → crates/sdk-core/tests/histories}/evict_while_la_running_no_interference-85_history.bin +0 -0
  413. /package/sdk-core/{histories → crates/sdk-core/tests/histories}/fail_wf_task.bin +0 -0
  414. /package/sdk-core/{histories → crates/sdk-core/tests/histories}/long_local_activity_with_update-0_history.bin +0 -0
  415. /package/sdk-core/{histories → crates/sdk-core/tests/histories}/long_local_activity_with_update-1_history.bin +0 -0
  416. /package/sdk-core/{histories → crates/sdk-core/tests/histories}/long_local_activity_with_update-2_history.bin +0 -0
  417. /package/sdk-core/{histories → crates/sdk-core/tests/histories}/long_local_activity_with_update-3_history.bin +0 -0
  418. /package/sdk-core/{histories → crates/sdk-core/tests/histories}/old_change_marker_format.bin +0 -0
  419. /package/sdk-core/{histories → crates/sdk-core/tests/histories}/timer_workflow_history.bin +0 -0
  420. /package/sdk-core/{tests → crates/sdk-core/tests}/integ_tests/workflow_tests/priority.rs +0 -0
  421. /package/sdk-core/{core-c-bridge → crates/sdk-core-c-bridge}/build.rs +0 -0
  422. /package/sdk-core/{cargo-tokio-console.sh → etc/cargo-tokio-console.sh} +0 -0
@@ -0,0 +1,13 @@
1
+ @startuml
2
+ Created -[#blue]-> StartCommandCreated: Schedule
3
+ StartCommandCreated -[#blue]-> StartCommandCreated: CommandStartTimer
4
+ StartCommandCreated -[#blue]-> StartCommandRecorded: TimerStarted
5
+ StartCommandCreated -[#blue]-> Canceled: Cancel
6
+ StartCommandRecorded -[#blue]-> Fired: TimerFired
7
+ StartCommandRecorded -[#blue]-> CancelTimerCommandCreated: Cancel
8
+ CancelTimerCommandCreated --> CancelTimerCommandCreated: Cancel
9
+ CancelTimerCommandCreated -[#blue]-> CancelTimerCommandSent: CommandCancelTimer
10
+ CancelTimerCommandSent --> Canceled: TimerCanceled
11
+ Canceled -[#blue]-> Canceled: Cancel
12
+ Fired -[#blue]-> Fired: Cancel
13
+ @enduml
@@ -0,0 +1,19 @@
1
+ @startuml
2
+ RequestInitiated --> Accepted: Accept
3
+ RequestInitiated --> Rejected: Reject
4
+ Accepted --> CompletedImmediately: Complete
5
+ Accepted --> CompletedImmediately: Reject
6
+ Accepted -[#blue]-> AcceptCommandCreated: CommandProtocolMessage
7
+ AcceptCommandCreated -[#blue]-> AcceptCommandRecorded: WorkflowExecutionUpdateAccepted
8
+ AcceptCommandCreated --> CompletedImmediatelyAcceptCreated: Complete
9
+ AcceptCommandCreated --> CompletedImmediatelyAcceptCreated: Reject
10
+ AcceptCommandRecorded --> Completed: Complete
11
+ AcceptCommandRecorded --> Completed: Reject
12
+ Completed -[#blue]-> CompletedCommandCreated: CommandProtocolMessage
13
+ CompletedCommandCreated -[#blue]-> CompletedCommandRecorded: WorkflowExecutionUpdateCompleted
14
+ CompletedImmediately -[#blue]-> CompletedImmediatelyAcceptCreated: CommandProtocolMessage
15
+ CompletedImmediatelyAcceptCreated -[#blue]-> CompletedImmediatelyCompleteCreated: CommandProtocolMessage
16
+ CompletedImmediatelyCompleteCreated -[#blue]-> CompletedCommandCreated: WorkflowExecutionUpdateAccepted
17
+ Rejected --> [*]
18
+ CompletedCommandRecorded --> [*]
19
+ @enduml
@@ -0,0 +1,5 @@
1
+ @startuml
2
+ Created -[#blue]-> CommandIssued: CommandScheduled
3
+ CommandIssued -[#blue]-> Done: CommandRecorded
4
+ Done --> [*]
5
+ @enduml
@@ -0,0 +1,11 @@
1
+ @startuml
2
+ Created -[#blue]-> Scheduled: WorkflowTaskScheduled
3
+ Scheduled -[#blue]-> Started: WorkflowTaskStarted
4
+ Scheduled --> TimedOut: WorkflowTaskTimedOut
5
+ Started -[#blue]-> Completed: WorkflowTaskCompleted
6
+ Started -[#blue]-> Failed: WorkflowTaskFailed
7
+ Started -[#blue]-> TimedOut: WorkflowTaskTimedOut
8
+ TimedOut --> [*]
9
+ Completed --> [*]
10
+ Failed --> [*]
11
+ @enduml
@@ -10,7 +10,7 @@ use std::{
10
10
  atomic::{AtomicBool, AtomicUsize, Ordering},
11
11
  },
12
12
  };
13
- use temporal_sdk_core_api::{
13
+ use temporalio_common::{
14
14
  telemetry::metrics::TemporalMeter,
15
15
  worker::{
16
16
  SlotKind, SlotMarkUsedContext, SlotReleaseContext, SlotReservationContext, SlotSupplier,
@@ -25,6 +25,7 @@ use tokio_util::sync::CancellationToken;
25
25
  #[derive(Clone)]
26
26
  pub(crate) struct MeteredPermitDealer<SK: SlotKind> {
27
27
  supplier: Arc<dyn SlotSupplier<SlotKind = SK> + Send + Sync>,
28
+ slot_supplier_kind: SlotSupplierKind,
28
29
  /// The number of permit owners who have acquired a permit, but are not yet meaningfully using
29
30
  /// that permit. This is useful for giving a more semantically accurate count of used task
30
31
  /// slots, since we typically wait for a permit first before polling, but that slot isn't used
@@ -54,6 +55,35 @@ pub(crate) struct PermitDealerContextData {
54
55
  pub(crate) worker_deployment_version: Option<WorkerDeploymentVersion>,
55
56
  }
56
57
 
58
+ #[derive(Clone, Debug, PartialEq, Eq)]
59
+ pub(crate) enum SlotSupplierKind {
60
+ Fixed,
61
+ ResourceBased,
62
+ Custom(String),
63
+ }
64
+
65
+ impl SlotSupplierKind {
66
+ fn from_label(label: &str) -> Self {
67
+ if label == "Fixed" {
68
+ SlotSupplierKind::Fixed
69
+ } else if label == "ResourceBased" {
70
+ SlotSupplierKind::ResourceBased
71
+ } else {
72
+ SlotSupplierKind::Custom(label.to_string())
73
+ }
74
+ }
75
+ }
76
+
77
+ impl std::fmt::Display for SlotSupplierKind {
78
+ fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
79
+ match self {
80
+ SlotSupplierKind::Fixed => f.write_str("Fixed"),
81
+ SlotSupplierKind::ResourceBased => f.write_str("ResourceBased"),
82
+ SlotSupplierKind::Custom(name) => f.write_str(name.as_str()),
83
+ }
84
+ }
85
+ }
86
+
57
87
  impl<SK> MeteredPermitDealer<SK>
58
88
  where
59
89
  SK: SlotKind + 'static,
@@ -65,8 +95,11 @@ where
65
95
  context_data: Arc<PermitDealerContextData>,
66
96
  meter: Option<TemporalMeter>,
67
97
  ) -> Self {
98
+ let supplier_kind_label = supplier.slot_supplier_kind();
99
+ let slot_supplier_kind = SlotSupplierKind::from_label(supplier_kind_label.as_ref());
68
100
  Self {
69
101
  supplier,
102
+ slot_supplier_kind,
70
103
  unused_claimants: Arc::new(AtomicUsize::new(0)),
71
104
  extant_permits: watch::channel(0),
72
105
  metrics_ctx,
@@ -81,6 +114,10 @@ where
81
114
  self.supplier.available_slots()
82
115
  }
83
116
 
117
+ pub(crate) fn slot_supplier_kind(&self) -> &SlotSupplierKind {
118
+ &self.slot_supplier_kind
119
+ }
120
+
84
121
  #[cfg(test)]
85
122
  pub(crate) fn unused_permits(&self) -> Option<usize> {
86
123
  self.available_permits()
@@ -386,10 +423,23 @@ impl<SK: SlotKind> OwnedMeteredSemPermit<SK> {
386
423
  pub(crate) struct UsedMeteredSemPermit<SK: SlotKind>(#[allow(dead_code)] OwnedMeteredSemPermit<SK>);
387
424
 
388
425
  macro_rules! dbg_panic {
389
- ($($arg:tt)*) => {
390
- error!($($arg)*);
391
- debug_assert!(false, $($arg)*);
392
- };
426
+ ($($arg:tt)*) => {{
427
+ let message = format!($($arg)*);
428
+ error!("{}", message);
429
+
430
+ #[cfg(feature = "antithesis_assertions")]
431
+ crate::antithesis::assert_always!(
432
+ false,
433
+ "dbg_panic invariant triggered",
434
+ ::serde_json::json!({
435
+ "message": message,
436
+ "file": file!(),
437
+ "line": line!(),
438
+ "module": module_path!(),
439
+ })
440
+ );
441
+ debug_assert!(false, "{}", message);
442
+ }};
393
443
  }
394
444
  pub(crate) use dbg_panic;
395
445
 
@@ -427,7 +477,7 @@ pub(crate) mod tests {
427
477
  use super::*;
428
478
  use crate::{advance_fut, worker::tuner::FixedSizeSlotSupplier};
429
479
  use futures_util::FutureExt;
430
- use temporal_sdk_core_api::worker::WorkflowSlotKind;
480
+ use temporalio_common::worker::WorkflowSlotKind;
431
481
 
432
482
  pub(crate) fn fixed_size_permit_dealer<SK: SlotKind + Send + Sync + 'static>(
433
483
  size: usize,
@@ -492,4 +542,10 @@ pub(crate) mod tests {
492
542
  // Now it'll proceed
493
543
  acquire_fut.await;
494
544
  }
545
+
546
+ #[test]
547
+ fn captures_slot_supplier_kind() {
548
+ let dealer = fixed_size_permit_dealer::<WorkflowSlotKind>(1);
549
+ assert_eq!(*dealer.slot_supplier_kind(), SlotSupplierKind::Fixed);
550
+ }
495
551
  }
@@ -0,0 +1,60 @@
1
+ //! Antithesis SDK integration for invariant testing.
2
+ //!
3
+ //! This module provides assertion macros that integrate with the Antithesis
4
+ //! testing platform to detect invariant violations during fuzz testing.
5
+
6
+ use std::sync::OnceLock;
7
+
8
+ /// Ensure Antithesis is initialized exactly once.
9
+ pub(crate) fn ensure_init() {
10
+ static INIT: OnceLock<()> = OnceLock::new();
11
+ INIT.get_or_init(|| {
12
+ ::antithesis_sdk::antithesis_init();
13
+ });
14
+ }
15
+
16
+ /// Assert that a condition is always true during Antithesis fuzz testing.
17
+ /// Use `false` as the condition to log an invariant violation.
18
+ macro_rules! assert_always {
19
+ ($condition:expr, $message:literal, $details:expr) => {{
20
+ $crate::antithesis::ensure_init();
21
+ let details: ::serde_json::Value = $details;
22
+ ::antithesis_sdk::assert_always!($condition, $message, &details);
23
+ }};
24
+ ($condition:expr, $message:literal) => {{
25
+ $crate::antithesis::ensure_init();
26
+ ::antithesis_sdk::assert_always!($condition, $message);
27
+ }};
28
+ }
29
+
30
+ /// Assert that a condition is sometimes true during Antithesis fuzz testing.
31
+ /// This checks that the condition occurs at least once across the entire test session.
32
+ macro_rules! assert_sometimes {
33
+ ($condition:expr, $message:literal, $details:expr) => {{
34
+ $crate::antithesis::ensure_init();
35
+ let details: ::serde_json::Value = $details;
36
+ ::antithesis_sdk::assert_sometimes!($condition, $message, &details);
37
+ }};
38
+ ($condition:expr, $message:literal) => {{
39
+ $crate::antithesis::ensure_init();
40
+ ::antithesis_sdk::assert_sometimes!($condition, $message);
41
+ }};
42
+ }
43
+
44
+ /// Assert that a code location is unreachable during Antithesis fuzz testing.
45
+ /// Use this for code paths that should never be reached (bugs, invariant violations).
46
+ macro_rules! assert_unreachable {
47
+ ($message:literal, $details:expr) => {{
48
+ $crate::antithesis::ensure_init();
49
+ let details: ::serde_json::Value = $details;
50
+ ::antithesis_sdk::assert_unreachable!($message, &details);
51
+ }};
52
+ ($message:literal) => {{
53
+ $crate::antithesis::ensure_init();
54
+ ::antithesis_sdk::assert_unreachable!($message);
55
+ }};
56
+ }
57
+
58
+ pub(crate) use assert_always;
59
+ pub(crate) use assert_sometimes;
60
+ pub(crate) use assert_unreachable;
@@ -1,10 +1,10 @@
1
1
  use crate::{
2
2
  ActivityHeartbeat, Worker, advance_fut, job_assert, prost_dur,
3
3
  test_help::{
4
- MockPollCfg, MockWorkerInputs, MocksHolder, QueueResponse, ResponseType, TEST_Q, WorkerExt,
5
- WorkflowCachingPolicy, build_fake_worker, build_mock_pollers, canned_histories,
6
- gen_assert_and_reply, mock_manual_poller, mock_poller, mock_poller_from_resps,
7
- mock_sdk_cfg, mock_worker, poll_and_reply, single_hist_mock_sg, test_worker_cfg,
4
+ MockPollCfg, MockWorkerInputs, MocksHolder, QueueResponse, WorkerExt,
5
+ WorkflowCachingPolicy, build_fake_worker, build_mock_pollers, fanout_tasks,
6
+ gen_assert_and_reply, mock_manual_poller, mock_poller, mock_worker, poll_and_reply,
7
+ single_hist_mock_sg, test_worker_cfg,
8
8
  },
9
9
  worker::client::mocks::{mock_manual_worker_client, mock_worker_client},
10
10
  };
@@ -19,47 +19,43 @@ use std::{
19
19
  Arc,
20
20
  atomic::{AtomicBool, AtomicUsize, Ordering},
21
21
  },
22
- time::Duration,
22
+ time::{Duration, Instant},
23
23
  };
24
- use temporal_client::WorkflowOptions;
25
- use temporal_sdk::{ActivityOptions, WfContext};
26
- use temporal_sdk_core_api::{
24
+ use temporalio_common::{
27
25
  Worker as WorkerTrait,
28
- errors::{CompleteActivityError, PollError},
29
- worker::PollerBehavior,
30
- };
31
- use temporal_sdk_core_protos::{
32
- DEFAULT_ACTIVITY_TYPE, DEFAULT_WORKFLOW_TYPE, TestHistoryBuilder,
33
- coresdk::{
34
- ActivityTaskCompletion,
35
- activity_result::{
36
- ActivityExecutionResult, ActivityResolution, Success, activity_execution_result,
37
- activity_resolution,
38
- },
39
- activity_task::{ActivityCancelReason, ActivityTask, Cancel, activity_task},
40
- workflow_activation::{ResolveActivity, WorkflowActivationJob, workflow_activation_job},
41
- workflow_commands::{
42
- ActivityCancellationType, CompleteWorkflowExecution, RequestCancelActivity,
43
- ScheduleActivity,
44
- },
45
- workflow_completion::WorkflowActivationCompletion,
46
- },
47
- temporal::api::{
48
- command::v1::{ScheduleActivityTaskCommandAttributes, command::Attributes},
49
- enums::v1::{CommandType, EventType},
50
- history::v1::{
51
- ActivityTaskScheduledEventAttributes, history_event::Attributes as EventAttributes,
26
+ errors::CompleteActivityError,
27
+ protos::{
28
+ TestHistoryBuilder, canned_histories,
29
+ coresdk::{
30
+ ActivityTaskCompletion,
31
+ activity_result::{
32
+ ActivityExecutionResult, ActivityResolution, Success, activity_execution_result,
33
+ activity_resolution,
34
+ },
35
+ activity_task::{ActivityCancelReason, ActivityTask, Cancel, activity_task},
36
+ workflow_activation::{
37
+ ResolveActivity, WorkflowActivationJob, workflow_activation_job,
38
+ },
39
+ workflow_commands::{
40
+ ActivityCancellationType, CompleteWorkflowExecution, RequestCancelActivity,
41
+ ScheduleActivity,
42
+ },
43
+ workflow_completion::WorkflowActivationCompletion,
52
44
  },
53
- sdk::v1::UserMetadata,
54
- workflowservice::v1::{
55
- PollActivityTaskQueueResponse, RecordActivityTaskHeartbeatResponse,
56
- RespondActivityTaskCanceledResponse, RespondActivityTaskCompletedResponse,
57
- RespondActivityTaskFailedResponse, RespondWorkflowTaskCompletedResponse,
45
+ temporal::api::{
46
+ command::v1::{ScheduleActivityTaskCommandAttributes, command::Attributes},
47
+ enums::v1::EventType,
48
+ workflowservice::v1::{
49
+ PollActivityTaskQueueResponse, RecordActivityTaskHeartbeatResponse,
50
+ RespondActivityTaskCanceledResponse, RespondActivityTaskCompletedResponse,
51
+ RespondActivityTaskFailedResponse, RespondWorkflowTaskCompletedResponse,
52
+ },
58
53
  },
54
+ test_utils::start_timer_cmd,
59
55
  },
56
+ worker::{PollerBehavior, WorkerTaskTypes},
60
57
  };
61
- use temporal_sdk_core_test_utils::{TestWorker, fanout_tasks, start_timer_cmd};
62
- use tokio::{join, sync::Barrier, time::sleep};
58
+ use tokio::{join, time::sleep};
63
59
  use tokio_util::sync::CancellationToken;
64
60
 
65
61
  fn three_tasks() -> VecDeque<PollActivityTaskQueueResponse> {
@@ -642,6 +638,47 @@ async fn max_tq_acts_set_passed_to_poll_properly() {
642
638
  worker.poll_activity_task().await.unwrap();
643
639
  }
644
640
 
641
+ #[tokio::test]
642
+ async fn max_worker_acts_per_second_respected() {
643
+ let mut mock_client = mock_worker_client();
644
+ mock_client
645
+ .expect_poll_activity_task()
646
+ .returning(move |_, _| {
647
+ Ok(PollActivityTaskQueueResponse {
648
+ task_token: vec![1],
649
+ activity_id: "some-id".to_string(),
650
+ ..Default::default()
651
+ })
652
+ });
653
+ mock_client
654
+ .expect_complete_activity_task()
655
+ .returning(|_, _| Ok(RespondActivityTaskCompletedResponse::default()));
656
+
657
+ let cfg = test_worker_cfg()
658
+ .activity_task_poller_behavior(PollerBehavior::SimpleMaximum(1_usize))
659
+ .max_outstanding_activities(10_usize)
660
+ .max_worker_activities_per_second(1.0)
661
+ .build()
662
+ .unwrap();
663
+ let worker = Worker::new_test(cfg, mock_client);
664
+ let start = Instant::now();
665
+ let mut received = 0;
666
+ while start.elapsed().as_millis() < 900 {
667
+ let at = worker.poll_activity_task().await.unwrap();
668
+ received += 1;
669
+ worker
670
+ .complete_activity_task(ActivityTaskCompletion {
671
+ task_token: at.task_token,
672
+ result: Some(ActivityExecutionResult::ok("hi".into())),
673
+ })
674
+ .await
675
+ .unwrap();
676
+ }
677
+ // Two will be allowed because of the initial request. Without ratelimit in effect, this number
678
+ // would be comically high due to the mocks responding very fast.
679
+ assert_eq!(received, 2);
680
+ }
681
+
645
682
  #[rstest::rstest]
646
683
  #[tokio::test]
647
684
  async fn no_eager_activities_requested_when_worker_options_disable_it(
@@ -688,7 +725,7 @@ async fn no_eager_activities_requested_when_worker_options_disable_it(
688
725
  mock.worker_cfg(|wc| {
689
726
  wc.max_cached_workflows = 2;
690
727
  if reason == "no_remote" {
691
- wc.no_remote_activities = true;
728
+ wc.task_types = WorkerTaskTypes::workflow_only();
692
729
  } else {
693
730
  wc.max_task_queue_activities_per_second = Some(1.0);
694
731
  }
@@ -701,7 +738,7 @@ async fn no_eager_activities_requested_when_worker_options_disable_it(
701
738
  ScheduleActivity {
702
739
  seq: 1,
703
740
  activity_id: "act_id".to_string(),
704
- task_queue: TEST_Q.to_string(),
741
+ task_queue: core.get_config().task_queue.clone(),
705
742
  cancellation_type: ActivityCancellationType::TryCancel as i32,
706
743
  ..Default::default()
707
744
  }
@@ -781,11 +818,14 @@ async fn activity_tasks_from_completion_are_delivered() {
781
818
  mock.expect_complete_activity_task()
782
819
  .times(3)
783
820
  .returning(|_, _| Ok(RespondActivityTaskCompletedResponse::default()));
784
- let mut mock = single_hist_mock_sg(wfid, t, [1], mock, true);
785
821
  let act_tasks: Vec<QueueResponse<PollActivityTaskQueueResponse>> = vec![];
786
- mock.set_act_poller(mock_poller_from_resps(act_tasks));
822
+ let mut mh = MockPollCfg::from_resp_batches(wfid, t, [1], mock);
823
+ mh.enforce_correct_number_of_polls = true;
824
+ mh.activity_responses = Some(act_tasks);
825
+ let mut mock = build_mock_pollers(mh);
787
826
  mock.worker_cfg(|wc| wc.max_cached_workflows = 2);
788
827
  let core = mock_worker(mock);
828
+ let task_queue = core.get_config().task_queue.clone();
789
829
 
790
830
  // Test start
791
831
  let wf_task = core.poll_workflow_activation().await.unwrap();
@@ -794,7 +834,7 @@ async fn activity_tasks_from_completion_are_delivered() {
794
834
  ScheduleActivity {
795
835
  seq,
796
836
  activity_id: format!("act_id_{seq}_same_queue"),
797
- task_queue: TEST_Q.to_string(),
837
+ task_queue: task_queue.clone(),
798
838
  cancellation_type: ActivityCancellationType::TryCancel as i32,
799
839
  ..Default::default()
800
840
  }
@@ -805,7 +845,7 @@ async fn activity_tasks_from_completion_are_delivered() {
805
845
  ScheduleActivity {
806
846
  seq: 4,
807
847
  activity_id: "act_id_same_queue_not_eager".to_string(),
808
- task_queue: TEST_Q.to_string(),
848
+ task_queue: task_queue.clone(),
809
849
  cancellation_type: ActivityCancellationType::TryCancel as i32,
810
850
  ..Default::default()
811
851
  }
@@ -848,160 +888,6 @@ async fn activity_tasks_from_completion_are_delivered() {
848
888
  assert_eq!(num_eager_requested.load(Ordering::Relaxed), 3);
849
889
  }
850
890
 
851
- #[tokio::test]
852
- async fn activity_tasks_from_completion_reserve_slots() {
853
- let wf_id = "fake_wf_id";
854
- let mut t = TestHistoryBuilder::default();
855
- t.add_by_type(EventType::WorkflowExecutionStarted);
856
- t.add_full_wf_task();
857
- let schedid = t.add(EventAttributes::ActivityTaskScheduledEventAttributes(
858
- ActivityTaskScheduledEventAttributes {
859
- activity_id: "1".to_string(),
860
- activity_type: Some("act1".into()),
861
- ..Default::default()
862
- },
863
- ));
864
- let startid = t.add_activity_task_started(schedid);
865
- t.add_activity_task_completed(schedid, startid, b"hi".into());
866
- t.add_full_wf_task();
867
- let schedid = t.add(EventAttributes::ActivityTaskScheduledEventAttributes(
868
- ActivityTaskScheduledEventAttributes {
869
- activity_id: "2".to_string(),
870
- activity_type: Some("act2".into()),
871
- ..Default::default()
872
- },
873
- ));
874
- let startid = t.add_activity_task_started(schedid);
875
- t.add_activity_task_completed(schedid, startid, b"hi".into());
876
- t.add_full_wf_task();
877
- t.add_workflow_execution_completed();
878
-
879
- let mut mock = mock_worker_client();
880
- // Set up two tasks to be returned via normal activity polling
881
- let act_tasks = VecDeque::from(vec![
882
- PollActivityTaskQueueResponse {
883
- task_token: vec![1],
884
- activity_id: "act1".to_string(),
885
- ..Default::default()
886
- }
887
- .into(),
888
- PollActivityTaskQueueResponse {
889
- task_token: vec![2],
890
- activity_id: "act2".to_string(),
891
- ..Default::default()
892
- }
893
- .into(),
894
- ]);
895
- mock.expect_complete_activity_task()
896
- .times(2)
897
- .returning(|_, _| Ok(RespondActivityTaskCompletedResponse::default()));
898
- let barr: &'static Barrier = Box::leak(Box::new(Barrier::new(2)));
899
- let mut mh = MockPollCfg::from_resp_batches(
900
- wf_id,
901
- t,
902
- [
903
- ResponseType::ToTaskNum(1),
904
- // We don't want the second task to be delivered until *after* the activity tasks
905
- // have been completed, so that the second activity schedule will have slots available
906
- ResponseType::UntilResolved(
907
- async {
908
- barr.wait().await;
909
- barr.wait().await;
910
- }
911
- .boxed(),
912
- 2,
913
- ),
914
- ResponseType::AllHistory,
915
- ],
916
- mock,
917
- );
918
- mh.completion_mock_fn = Some(Box::new(|wftc| {
919
- // Make sure when we see the completion with the schedule act command that it does
920
- // not have the eager execution flag set the first time, and does the second.
921
- if let Some(Attributes::ScheduleActivityTaskCommandAttributes(attrs)) = wftc
922
- .commands
923
- .first()
924
- .and_then(|cmd| cmd.attributes.as_ref())
925
- {
926
- if attrs.activity_id == "1" {
927
- assert!(!attrs.request_eager_execution);
928
- } else {
929
- assert!(attrs.request_eager_execution);
930
- }
931
- }
932
- Ok(Default::default())
933
- }));
934
- let mut mock = build_mock_pollers(mh);
935
- mock.worker_cfg(|cfg| {
936
- cfg.max_cached_workflows = 2;
937
- cfg.max_outstanding_activities = Some(2);
938
- });
939
- mock.set_act_poller(mock_poller_from_resps(act_tasks));
940
- let core = Arc::new(mock_worker(mock));
941
- let mut worker = TestWorker::new(core.clone(), TEST_Q.to_string());
942
-
943
- // First poll for activities twice, occupying both slots
944
- let at1 = core.poll_activity_task().await.unwrap();
945
- let at2 = core.poll_activity_task().await.unwrap();
946
- let workflow_complete_token = CancellationToken::new();
947
- let workflow_complete_token_clone = workflow_complete_token.clone();
948
-
949
- worker.register_wf(DEFAULT_WORKFLOW_TYPE, move |ctx: WfContext| {
950
- let complete_token = workflow_complete_token.clone();
951
- async move {
952
- ctx.activity(ActivityOptions {
953
- activity_type: "act1".to_string(),
954
- ..Default::default()
955
- })
956
- .await;
957
- ctx.activity(ActivityOptions {
958
- activity_type: "act2".to_string(),
959
- ..Default::default()
960
- })
961
- .await;
962
- complete_token.cancel();
963
- Ok(().into())
964
- }
965
- });
966
-
967
- worker
968
- .submit_wf(
969
- wf_id.to_owned(),
970
- DEFAULT_WORKFLOW_TYPE,
971
- vec![],
972
- WorkflowOptions::default(),
973
- )
974
- .await
975
- .unwrap();
976
- let act_completer = async {
977
- barr.wait().await;
978
- core.complete_activity_task(ActivityTaskCompletion {
979
- task_token: at1.task_token,
980
- result: Some(ActivityExecutionResult::ok("hi".into())),
981
- })
982
- .await
983
- .unwrap();
984
- core.complete_activity_task(ActivityTaskCompletion {
985
- task_token: at2.task_token,
986
- result: Some(ActivityExecutionResult::ok("hi".into())),
987
- })
988
- .await
989
- .unwrap();
990
- barr.wait().await;
991
- // Wait for workflow to complete in order for all eager activities to be requested before shutting down.
992
- // After shutdown, no eager activities slots can be allocated.
993
- workflow_complete_token_clone.cancelled().await;
994
- core.initiate_shutdown();
995
- // Even though this test requests eager activity tasks, none are returned in poll responses.
996
- let err = core.poll_activity_task().await.unwrap_err();
997
- assert_matches!(err, PollError::ShutDown);
998
- };
999
- // This wf poll should *not* set the flag that it wants tasks back since both slots are
1000
- // occupied
1001
- let run_fut = async { worker.run_until_done().await.unwrap() };
1002
- join!(run_fut, act_completer);
1003
- }
1004
-
1005
891
  #[tokio::test]
1006
892
  async fn retryable_net_error_exhaustion_is_nonfatal() {
1007
893
  let mut mock_client = mock_worker_client();
@@ -1198,57 +1084,6 @@ async fn activities_must_be_flushed_to_server_on_shutdown(#[values(true, false)]
1198
1084
  join!(shutdown_task, complete_task);
1199
1085
  }
1200
1086
 
1201
- #[tokio::test]
1202
- async fn pass_activity_summary_to_metadata() {
1203
- let t = canned_histories::single_activity("1");
1204
- let mut mock_cfg = MockPollCfg::from_hist_builder(t);
1205
- let wf_id = mock_cfg.hists[0].wf_id.clone();
1206
- let wf_type = DEFAULT_WORKFLOW_TYPE;
1207
- let expected_user_metadata = Some(UserMetadata {
1208
- summary: Some(b"activity summary".into()),
1209
- details: None,
1210
- });
1211
- mock_cfg.completion_asserts_from_expectations(|mut asserts| {
1212
- asserts
1213
- .then(move |wft| {
1214
- assert_eq!(wft.commands.len(), 1);
1215
- assert_eq!(
1216
- wft.commands[0].command_type(),
1217
- CommandType::ScheduleActivityTask
1218
- );
1219
- assert_eq!(wft.commands[0].user_metadata, expected_user_metadata)
1220
- })
1221
- .then(move |wft| {
1222
- assert_eq!(wft.commands.len(), 1);
1223
- assert_eq!(
1224
- wft.commands[0].command_type(),
1225
- CommandType::CompleteWorkflowExecution
1226
- );
1227
- });
1228
- });
1229
-
1230
- let mut worker = mock_sdk_cfg(mock_cfg, |_| {});
1231
- worker.register_wf(wf_type, |ctx: WfContext| async move {
1232
- ctx.activity(ActivityOptions {
1233
- activity_type: DEFAULT_ACTIVITY_TYPE.to_string(),
1234
- summary: Some("activity summary".to_string()),
1235
- ..Default::default()
1236
- })
1237
- .await;
1238
- Ok(().into())
1239
- });
1240
- worker
1241
- .submit_wf(
1242
- wf_id.to_owned(),
1243
- wf_type.to_owned(),
1244
- vec![],
1245
- WorkflowOptions::default(),
1246
- )
1247
- .await
1248
- .unwrap();
1249
- worker.run_until_done().await.unwrap();
1250
- }
1251
-
1252
1087
  #[tokio::test]
1253
1088
  async fn heartbeat_response_can_be_paused() {
1254
1089
  let mut mock_client = mock_worker_client();