Refactor mock leader in nonce cache tests

nitronit · Dec 19, 2023 · 731103a · 731103a
1 parent 95a1cba
commit 731103a
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/signer/cosigner_nonce_cache_test.go b/signer/cosigner_nonce_cache_test.go
@@ -153,7 +153,7 @@ func TestNonceCacheDemand(t *testing.T) {
 	nonceCache := NewCosignerNonceCache(
 		cometlog.NewTMLogger(cometlog.NewSyncWriter(os.Stdout)),
 		cosigners,
-		&MockLeader{id: 1, leader: &ThresholdValidator{myCosigner: lcs[0]}},
+		&MockLeader{id: 1, leader: &MockThresholdValidator{myCosigner: lcs[0]}},
 		500*time.Millisecond,
 		100*time.Millisecond,
 		defaultNonceExpiration,
@@ -205,7 +205,7 @@ func TestNonceCacheExpiration(t *testing.T) {
 	nonceCache := NewCosignerNonceCache(
 		cometlog.NewTMLogger(cometlog.NewSyncWriter(os.Stdout)),
 		cosigners,
-		&MockLeader{id: 1, leader: &ThresholdValidator{myCosigner: lcs[0]}},
+		&MockLeader{id: 1, leader: &MockThresholdValidator{myCosigner: lcs[0]}},
 		getNoncesInterval,
 		getNoncesTimeout,
 		noncesExpiration,
@@ -411,7 +411,7 @@ func TestNonceCacheDemandSlow(t *testing.T) {
 	nonceCache := NewCosignerNonceCache(
 		cometlog.NewTMLogger(cometlog.NewSyncWriter(os.Stdout)),
 		cosigners,
-		&MockLeader{id: 1, leader: &ThresholdValidator{myCosigner: lcs[0]}},
+		&MockLeader{id: 1, leader: &MockThresholdValidator{myCosigner: lcs[0]}},
 		90*time.Millisecond,
 		100*time.Millisecond,
 		500*time.Millisecond,
@@ -448,7 +448,7 @@ func TestNonceCacheDemandSlowDefault(t *testing.T) {
 	nonceCache := NewCosignerNonceCache(
 		cometlog.NewTMLogger(cometlog.NewSyncWriter(os.Stdout)),
 		cosigners,
-		&MockLeader{id: 1, leader: &ThresholdValidator{myCosigner: lcs[0]}},
+		&MockLeader{id: 1, leader: &MockThresholdValidator{myCosigner: lcs[0]}},
 		defaultGetNoncesInterval,
 		defaultGetNoncesTimeout,
 		defaultNonceExpiration,

diff --git a/signer/threshold_validator_test.go b/signer/threshold_validator_test.go
@@ -1,4 +1,4 @@
-package signer_test
+package signer
 
 import (
 	"bytes"