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

Refactor mocking rand

Browse Source 
Signed-off-by: Marek Siarkowicz <siarkowicz@google.com>
This commit is contained in:
Marek Siarkowicz 2023-11-15 10:26:01 +01:00
parent 6df2a0cd97
commit b71686d1e6
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),