Mirroristas/etcd
2
0
Fork 0
mirror of https://github.com/etcd-io/etcd.git synced 2024-09-27 06:25:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #16944 from serathius/refactor-mock-rand

Browse Source 
Refactor mocking rand
This commit is contained in:
Marek Siarkowicz 2023-11-16 09:28:34 +01:00 committed by GitHub
commit 0c76f50dbc
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 10 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
tests/robustness/options/cluster_options_test.go
View File

@ -22,16 +22,17 @@ import (
"go.etcd.io/etcd/tests/v3/framework/e2e"
)
func resetRand() {
internalRand = rand.New(rand.NewSource(1))
}
func init() {
resetRand()
func mockRand(source rand.Source) func() {
tmp := internalRand
internalRand = rand.New(source)
return func() {
internalRand = tmp
}
}
func TestWithClusterOptionGroups(t *testing.T) {
defer resetRand()
restore := mockRand(rand.NewSource(1))
defer restore()
tickOptions1 := ClusterOptions{WithTickMs(101), WithElectionMs(1001)}
tickOptions2 := ClusterOptions{WithTickMs(202), WithElectionMs(2002)}
tickOptions3 := ClusterOptions{WithTickMs(303), WithElectionMs(3003)}
@ -67,7 +68,8 @@ func TestWithClusterOptionGroups(t *testing.T) {
}
func TestWithOptionsSubset(t *testing.T) {
defer resetRand()
restore := mockRand(rand.NewSource(1))
defer restore()
tickOptions := ClusterOptions{WithTickMs(50), WithElectionMs(500)}
opts := ClusterOptions{
WithSnapshotCatchUpEntries(100),