CI debug testing

Author: Sushisource

temporalio/bridge/src/worker.rs

@@ -64,6 +64,18 @@ pub struct TunerHolder {
     local_activity_slot_supplier: SlotSupplier,
 }
 
+// pub fn set_task_locals_on_tuner<'a>(py: Python<'a>, tuner: &TunerHolder) -> PyResult<()> {
+//     // TODO: All suppliers
+//     if let SlotSupplier::Custom(ref cs) = tuner.workflow_slot_supplier {
+//         Python::with_gil(|py| {
+//             let py_obj = cs.inner.as_ref(py);
+//             py_obj.call_method0("set_task_locals")?;
+//             Ok(())
+//         })?;
+//     };
+//     Ok(())
+// }
+
 #[derive(FromPyObject)]
 pub enum SlotSupplier {
     FixedSize(FixedSizeSlotSupplier),
@@ -217,6 +229,7 @@ impl<SK: SlotKind + Send + Sync> SlotSupplierTrait for CustomSlotSupplierOfType<
             .await;
             match pypermit {
                 Ok(p) => {
+                    dbg!("reserve done", SK::kind());
                     return SlotSupplierPermit::with_user_data(p);
                 }
                 Err(_) => {
@@ -274,6 +287,7 @@ impl<SK: SlotKind + Send + Sync> SlotSupplierTrait for CustomSlotSupplierOfType<
     }
 
     fn release_slot(&self, ctx: &dyn SlotReleaseContext<SlotKind = Self::SlotKind>) {
+        dbg!("release", SK::kind());
         if let Err(e) = Python::with_gil(|py| {
             let permit = ctx
                 .permit()
@@ -362,6 +376,8 @@ pub fn new_replay_worker<'a>(
 impl WorkerRef {
     fn validate<'p>(&self, py: Python<'p>) -> PyResult<&'p PyAny> {
         let worker = self.worker.as_ref().unwrap().clone();
+        // Set custom slot supplier task locals so they can run futures
+        // match worker.get_config().tuner {}
         self.runtime.future_into_py(py, async move {
             worker
                 .validate()

tests/worker/test_worker.py

@@ -422,9 +422,6 @@ def reserve_asserts(self, ctx):
         await wf1.signal(WaitOnSignalWorkflow.my_signal, "finish")
         await wf1.result()
 
-    async def releases() -> int:
-        return ss.releases
-
     assert ss.reserves == ss.releases
     # Two workflow tasks, one activity
     assert ss.used == 3