Skip to content

Commit 255f8c9

Browse files
committed
refactor: fix utils tests
1 parent cad9535 commit 255f8c9

File tree

5 files changed

+469
-408
lines changed

5 files changed

+469
-408
lines changed

cmd/commit_test.go

Lines changed: 10 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,6 @@ import (
99
"github.com/ethereum/go-ethereum/ethclient"
1010
"github.com/stretchr/testify/mock"
1111
"math/big"
12-
clientPkg "razor/client"
1312
"razor/core"
1413
"razor/core/types"
1514
"razor/pkg/bindings"
@@ -121,9 +120,10 @@ func TestCommit(t *testing.T) {
121120

122121
func TestHandleCommitState(t *testing.T) {
123122
var (
124-
client *ethclient.Client
125-
epoch uint32
126-
seed []byte
123+
client *ethclient.Client
124+
epoch uint32
125+
seed []byte
126+
commitParams types.CommitParams
127127
)
128128

129129
rogueValue := big.NewInt(1111)
@@ -234,7 +234,7 @@ func TestHandleCommitState(t *testing.T) {
234234
utilsMock.On("GetRogueRandomValue", mock.Anything).Return(rogueValue)
235235

236236
utils := &UtilsStruct{}
237-
got, err := utils.HandleCommitState(client, epoch, seed, &clientPkg.HttpClient{}, tt.args.rogueData)
237+
got, err := utils.HandleCommitState(client, epoch, seed, commitParams, tt.args.rogueData)
238238
if !reflect.DeepEqual(got, tt.want) {
239239
t.Errorf("Data from HandleCommitState function, got = %v, want = %v", got, tt.want)
240240
}
@@ -372,9 +372,10 @@ func TestGetSalt(t *testing.T) {
372372

373373
func BenchmarkHandleCommitState(b *testing.B) {
374374
var (
375-
client *ethclient.Client
376-
epoch uint32
377-
seed []byte
375+
client *ethclient.Client
376+
epoch uint32
377+
seed []byte
378+
commitParams types.CommitParams
378379
)
379380

380381
rogueValue := big.NewInt(1111)
@@ -399,7 +400,7 @@ func BenchmarkHandleCommitState(b *testing.B) {
399400
utilsMock.On("GetRogueRandomValue", mock.Anything).Return(rogueValue)
400401

401402
ut := &UtilsStruct{}
402-
_, err := ut.HandleCommitState(client, epoch, seed, &clientPkg.HttpClient{}, types.Rogue{IsRogue: false})
403+
_, err := ut.HandleCommitState(client, epoch, seed, commitParams, types.Rogue{IsRogue: false})
403404
if err != nil {
404405
log.Fatal(err)
405406
}

0 commit comments

Comments
 (0)