Updated signing done check related unit tests

keep-network · Aug 24, 2023 · 874c320 · 874c320
1 parent 244a769
commit 874c320
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/pkg/tbtc/signing_done_test.go b/pkg/tbtc/signing_done_test.go
@@ -14,6 +14,7 @@ import (
 	"github.com/keep-network/keep-core/internal/testutils"
 	"github.com/keep-network/keep-core/pkg/chain"
 	"github.com/keep-network/keep-core/pkg/chain/local_v1"
+	"github.com/keep-network/keep-core/pkg/net"
 	"github.com/keep-network/keep-core/pkg/net/local"
 	"github.com/keep-network/keep-core/pkg/operator"
 	"github.com/keep-network/keep-core/pkg/protocol/group"
@@ -326,6 +327,10 @@ func setupSigningDoneCheck(
 		t.Fatal(err)
 	}
 
+	broadcastChannel.SetUnmarshaler(func() net.TaggedUnmarshaler {
+		return &signingDoneMessage{}
+	})
+
 	membershipValidator := group.NewMembershipValidator(
 		&testutils.MockLogger{},
 		operators,