bug: still stend messages even if done

Author: ross-pure

keygen.go

@@ -316,10 +316,6 @@ func transitionToStep2(state *DKGState, info *DKGCatchUpInfo, ownIndex uint16, i
 
 	state.Step = 2
 
-	if len(state.Shares) == len(state.Commitments) {
-		return true, computeOutputs(state, indices, ownIndex), nil, nil
-	}
-
 	if info != nil {
 		info.IndexSubset = make([]uint16, 0, len(state.Commitments))
 		for index := range state.Commitments {
@@ -329,6 +325,10 @@ func transitionToStep2(state *DKGState, info *DKGCatchUpInfo, ownIndex uint16, i
 		sort.Slice(info.IndexSubset, func(i, j int) bool { return info.IndexSubset[i] < info.IndexSubset[j] })
 	}
 
+	if len(state.Shares) == len(state.Commitments) {
+		return true, computeOutputs(state, indices, ownIndex), shareMessages, nil
+	}
+
 	return false, DKGOutput{}, shareMessages, nil
 }
 

keygen_test.go

@@ -74,7 +74,7 @@ var _ = Describe("DKG", func() {
 			checkOutputs(outputs, n, t, indices)
 		})
 
-		FIt("should allow a node to catch up if it was offline", func() {
+		It("should allow a node to catch up if it was offline", func() {
 			n := 10
 			t := 5