Skip to content

Commit cc42102

Browse files
committed
chore: fix test
1 parent 1f5dfae commit cc42102

File tree

2 files changed

+13
-12
lines changed

2 files changed

+13
-12
lines changed

venus-shared/testutil/value_provider_primitive.go

+13-5
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,10 @@
11
package testutil
22

33
import (
4-
crand "crypto/rand"
54
"encoding/hex"
65
"math/rand"
76
"testing"
7+
"time"
88

99
"github.com/stretchr/testify/require"
1010
)
@@ -20,7 +20,7 @@ const (
2020
defaultBytesFixedSize = 16
2121
)
2222

23-
func IntProvider(t *testing.T) int { return rand.Int() }
23+
func IntProvider(t *testing.T) int { return r.Int() }
2424

2525
func IntRangedProvider(min, max int) func(*testing.T) int {
2626
return func(t *testing.T) int {
@@ -29,14 +29,22 @@ func IntRangedProvider(min, max int) func(*testing.T) int {
2929
t.Fatalf("invalid range [%d, %d)", min, max)
3030
}
3131

32-
return min + rand.Intn(gap)
32+
return min + r.Intn(gap)
3333
}
3434
}
3535

36+
var r = rand.New(rand.NewSource(time.Now().UnixNano()))
37+
38+
func getRand() *rand.Rand {
39+
seed := time.Now().UnixNano()
40+
r = rand.New(rand.NewSource(seed))
41+
return rand.New(rand.NewSource(seed))
42+
}
43+
3644
func BytesFixedProvider(size int) func(*testing.T) []byte {
3745
return func(t *testing.T) []byte {
3846
b := make([]byte, size)
39-
_, err := crand.Read(b[:])
47+
_, err := r.Read(b[:])
4048
require.NoError(t, err)
4149
return b
4250
}
@@ -45,7 +53,7 @@ func BytesFixedProvider(size int) func(*testing.T) []byte {
4553
func BytesAtMostProvider(size int) func(*testing.T) []byte {
4654
return func(t *testing.T) []byte {
4755
b := make([]byte, rand.Intn(size))
48-
_, err := crand.Read(b[:])
56+
_, err := r.Read(b[:])
4957
require.NoError(t, err)
5058
return b
5159
}

venus-shared/testutil/value_provider_primitive_test.go

-7
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,6 @@ package testutil
22

33
import (
44
"encoding/hex"
5-
"math/rand"
65
"reflect"
76
"testing"
87
"time"
@@ -11,12 +10,6 @@ import (
1110
"github.com/stretchr/testify/require"
1211
)
1312

14-
func getRand() *rand.Rand {
15-
seed := time.Now().UnixNano()
16-
rand.New(rand.NewSource(seed))
17-
return rand.New(rand.NewSource(seed))
18-
}
19-
2013
func TestDefaultBytes(t *testing.T) {
2114
tf.UnitTest(t)
2215

0 commit comments

Comments
 (0)