add optional timeout to send_transaction_with_await_replication

archethic-foundation · Jul 21, 2023 · 672a479 · 672a479
1 parent 905c510
commit 672a479
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/lib/archethic/utils/regression/playbook.ex b/lib/archethic/utils/regression/playbook.ex
@@ -219,7 +219,8 @@ defmodule Archethic.Utils.Regression.Playbook do
         host,
         port,
         curve \\ Crypto.default_curve(),
-        proto \\ :http
+        proto \\ :http,
+        opts \\ []
       ) do
     chain_length = get_chain_size(transaction_seed, curve, host, port, proto)
 
@@ -256,7 +257,10 @@ defmodule Archethic.Utils.Regression.Playbook do
            proto
          ) do
       {:ok, %{"status" => "pending"}} ->
-        case Task.yield(replication_attestation, 5_000) || Task.shutdown(replication_attestation) do
+        await_timeout = Keyword.get(opts, :await_timeout, 5_000)
+
+        case Task.yield(replication_attestation, await_timeout) ||
+               Task.shutdown(replication_attestation) do
           {:ok, :ok} ->
             {:ok, tx.address}