fix: suspend leak test accepts both failure payload shapes

Author: Yyeger

test/continuum/runtime/suspend_leak_test.exs

@@ -35,9 +35,8 @@ defmodule Continuum.Runtime.SuspendLeakTest do
     assert_received {:flow, flow}
     {:ok, run_id} = Continuum.Test.start_synchronous(flow, %{})
 
-    assert {:error, %{state: :failed, error: {:error, %Continuum.SuspendLeakError{} = error}}} =
-             Continuum.await(run_id, 1_000)
-
+    assert {:error, %{state: :failed, error: failure}} = Continuum.await(run_id, 1_000)
+    assert %Continuum.SuspendLeakError{} = error = leak_error(failure)
     assert Exception.message(error) =~ "suspend signal was swallowed"
     assert Exception.message(error) =~ "rescue"
   end
@@ -64,10 +63,17 @@ defmodule Continuum.Runtime.SuspendLeakTest do
     assert_received {:flow, flow}
     {:ok, run_id} = Continuum.Test.start_synchronous(flow, %{})
 
-    assert {:error, %{state: :failed, error: {:error, %Continuum.SuspendLeakError{}}}} =
-             Continuum.await(run_id, 1_000)
+    assert {:error, %{state: :failed, error: failure}} = Continuum.await(run_id, 1_000)
+    assert %Continuum.SuspendLeakError{} = leak_error(failure)
   end
 
+  # `await` resolves the failure from either the engine's `run_finished`
+  # broadcast ({:error, exception}) or the journal row ({kind, reason,
+  # stacktrace}), depending on which side wins the subscribe race.
+  defp leak_error({:error, %Continuum.SuspendLeakError{} = error}), do: error
+  defp leak_error({:error, %Continuum.SuspendLeakError{} = error, _stacktrace}), do: error
+  defp leak_error(other), do: other
+
   test "a catch arm that re-throws the control tuple suspends normally" do
     capture_io(:standard_error, fn ->
       defmodule RethrowFlow do