Moved setting unmarshaller for signing done message to separate function

pkg/protocol/announcer/announcer.go

@@ -71,6 +71,8 @@ type Announcer struct {
 	membershipValidator *group.MembershipValidator
 }
 
+// RegisterUnmarshaller initializes the given broadcast channel to be able to
+// handle announcement messages by registering the required unmarshaller.
 func RegisterUnmarshaller(channel net.BroadcastChannel) {
 	channel.SetUnmarshaler(func() net.TaggedUnmarshaler {
 		return &announcementMessage{}

pkg/tbtc/node.go

@@ -241,8 +241,7 @@ func (n *node) getSigningExecutor(
 		return nil, false, fmt.Errorf("failed to get broadcast channel: [%v]", err)
 	}
 
-	signing.RegisterUnmarshallers(broadcastChannel)
-	announcer.RegisterUnmarshaller(broadcastChannel)
+	registerSigningUnmarshallers(broadcastChannel)
 
 	membershipValidator := group.NewMembershipValidator(
 		executorLogger,
@@ -601,3 +600,14 @@ func withCancelOnBlock(
 
 	return blockCtx, cancelBlockCtx
 }
+
+// registerSigningUnmarshallers initializes the given broadcast channel to be
+// able to perform the signing protocol interactions by registering all the
+// required protocol unmarshallers.
+func registerSigningUnmarshallers(channel net.BroadcastChannel) {
+	signing.RegisterUnmarshallers(channel)
+	announcer.RegisterUnmarshaller(channel)
+	channel.SetUnmarshaler(func() net.TaggedUnmarshaler {
+		return &signingDoneMessage{}
+	})
+}

pkg/tbtc/signing_done.go

@@ -62,10 +62,6 @@ func newSigningDoneCheck(
 	broadcastChannel net.BroadcastChannel,
 	membershipValidator *group.MembershipValidator,
 ) *signingDoneCheck {
-	broadcastChannel.SetUnmarshaler(func() net.TaggedUnmarshaler {
-		return &signingDoneMessage{}
-	})
-
 	return &signingDoneCheck{
 		groupSize:           groupSize,
 		broadcastChannel:    broadcastChannel,