1 files changed, 50 insertions, 8 deletions
diff --git a/core/dkg-tsig-protocol_test.go b/core/dkg-tsig-protocol_test.go
index 198f50a..f6909e8 100644
--- a/core/dkg-tsig-protocol_test.go
+++ b/core/dkg-tsig-protocol_test.go
@@ -1035,13 +1035,12 @@ func benchmarkDKGGroupPubliKey(k, n int, b *testing.B) {
 		})
 	}
 
+	mpk := gov.DKGMasterPublicKeys(round)
+	comp := gov.DKGComplaints(round)
 	b.ResetTimer()
 	for i := 0; i < b.N; i++ {
 		// DKG is fininished.
-		gpk, err := typesDKG.NewGroupPublicKey(round,
-			gov.DKGMasterPublicKeys(round), gov.DKGComplaints(round),
-			k,
-		)
+		gpk, err := typesDKG.NewGroupPublicKey(round, mpk, comp, k)
 		if err != nil {
 			panic(err)
 		}
@@ -1080,13 +1079,12 @@ func benchmarkDKGNodePubliKeys(k, n int, b *testing.B) {
 		})
 	}
 
+	mpk := gov.DKGMasterPublicKeys(round)
+	comp := gov.DKGComplaints(round)
 	b.ResetTimer()
 	for i := 0; i < b.N; i++ {
 		// DKG is fininished.
-		npks, err := typesDKG.NewNodePublicKeys(round,
-			gov.DKGMasterPublicKeys(round), gov.DKGComplaints(round),
-			k,
-		)
+		npks, err := typesDKG.NewNodePublicKeys(round, mpk, comp, k)
 		if err != nil {
 			panic(err)
 		}
@@ -1095,3 +1093,47 @@ func benchmarkDKGNodePubliKeys(k, n int, b *testing.B) {
 		}
 	}
 }
+
+func BenchmarkCalcQ4_7(b *testing.B)    { benchmarkCalcQualified(4, 7, b) }
+func BenchmarkCalcQ9_13(b *testing.B)   { benchmarkCalcQualified(9, 13, b) }
+func BenchmarkCalcQ17_24(b *testing.B)  { benchmarkCalcQualified(17, 24, b) }
+func BenchmarkCalcQ81_121(b *testing.B) { benchmarkCalcQualified(81, 121, b) }
+
+func benchmarkCalcQualified(k, n int, b *testing.B) {
+	round := uint64(1)
+	reset := uint64(0)
+	_, pubKeys, err := test.NewKeys(n)
+	if err != nil {
+		panic(err)
+	}
+	gov, err := test.NewGovernance(test.NewState(DKGDelayRound,
+		pubKeys, 100, &common.NullLogger{}, true), ConfigRoundShift)
+	if err != nil {
+		panic(err)
+	}
+
+	for _, pk := range pubKeys {
+		_, pubShare := dkg.NewPrivateKeyShares(k)
+		gov.AddDKGMasterPublicKey(&typesDKG.MasterPublicKey{
+			ProposerID:      types.NewNodeID(pk),
+			Round:           round,
+			Reset:           reset,
+			DKGID:           typesDKG.NewID(types.NewNodeID(pk)),
+			PublicKeyShares: *pubShare,
+		})
+	}
+
+	mpk := gov.DKGMasterPublicKeys(round)
+	comp := gov.DKGComplaints(round)
+	b.ResetTimer()
+	for i := 0; i < b.N; i++ {
+		// DKG is fininished.
+		_, q, err := typesDKG.CalcQualifyNodes(mpk, comp, k)
+		if err != nil {
+			panic(err)
+		}
+		if len(q) != n {
+			panic("not enough of qualify id")
+		}
+	}
+}