Fix await confirmation + add protocol to ws handler

archethic-foundation · Jul 21, 2023 · d4a2693 · d4a2693
1 parent 672a479
commit d4a2693
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/lib/archethic/utils/regression/playbook.ex b/lib/archethic/utils/regression/playbook.ex
@@ -318,14 +318,17 @@ defmodule Archethic.Utils.Regression.Playbook do
     )
 
     receive do
-      %{"transactionConfirmed" => %{"nbConfirmations" => 1}} ->
+      %{"transactionConfirmed" => %{"nbConfirmations" => n}} when n > 0 ->
         :ok
 
       %{"transactionError" => %{"reason" => reason}} ->
         {:error, reason}
 
       {:error, reason} ->
         {:error, reason}
+
+      unknown_msg ->
+        Logger.warn("await_replication received an unknown message: #{inspect(unknown_msg)}")
     end
   end
 

diff --git a/lib/archethic/utils/websocket/handler.ex b/lib/archethic/utils/websocket/handler.ex
@@ -17,8 +17,9 @@ defmodule Archethic.Utils.WebSocket.SocketHandler do
     name = Keyword.get(opts, :ws_name, __MODULE__)
     host = Keyword.get(opts, :host, "localhost")
     port = Keyword.get(opts, :port, "4000")
+    protocol = Keyword.get(opts, :protocol, "ws")
 
-    ws_url = "ws://#{host}:#{port}/socket/websocket"
+    ws_url = "#{protocol}://#{host}:#{port}/socket/websocket"
 
     state = %{
       subscriptions: %{},