Skip to content

Commit 9ec25ca

Browse files
committed
fixes broken random value generation
1 parent 1bddf2e commit 9ec25ca

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

prepared_test.go

+7-2
Original file line numberDiff line numberDiff line change
@@ -33,6 +33,11 @@ var preparedMessageTests = []struct {
3333
}
3434

3535
func TestPreparedMessage(t *testing.T) {
36+
testRand := rand.New(rand.NewSource(99))
37+
prevMaskRand := maskRand
38+
maskRand = testRand
39+
defer func() { maskRand = prevMaskRand }()
40+
3641
for _, tt := range preparedMessageTests {
3742
var data = []byte("this is a test")
3843
var buf bytes.Buffer
@@ -43,7 +48,7 @@ func TestPreparedMessage(t *testing.T) {
4348
c.SetCompressionLevel(tt.compressionLevel)
4449

4550
// Seed random number generator for consistent frame mask.
46-
rand.Seed(1234)
51+
testRand.Seed(1234)
4752

4853
if err := c.WriteMessage(tt.messageType, data); err != nil {
4954
t.Fatal(err)
@@ -59,7 +64,7 @@ func TestPreparedMessage(t *testing.T) {
5964
copy(data, "hello world")
6065

6166
// Seed random number generator for consistent frame mask.
62-
rand.Seed(1234)
67+
testRand.Seed(1234)
6368

6469
buf.Reset()
6570
if err := c.WritePreparedMessage(pm); err != nil {

0 commit comments

Comments
 (0)