patchy/elect
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

More robust testing, TestSplitVotes

Browse Source
This commit is contained in:
Ian Gulliver
2023-06-07 23:22:53 -07:00
parent 38f232f7aa
commit e49ffd9505
3 changed files with 118 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

77
elect_test.go
View File

@@ -10,7 +10,7 @@ import (
func TestOne(t *testing.T) {
t.Parallel()
ts := NewTestSystem(t, 1)
ts := NewTestSystem(t, 1, 1)
defer ts.Stop()
require.False(t, ts.Candidate(0).IsLeader())
@@ -21,49 +21,66 @@ func TestOne(t *testing.T) {
func TestThree(t *testing.T) {
t.Parallel()
ts := NewTestSystem(t, 3)
ts := NewTestSystem(t, 3, 3)
defer ts.Stop()
require.False(t, ts.Candidate(0).IsLeader())
require.False(t, ts.Candidate(1).IsLeader())
require.False(t, ts.Candidate(2).IsLeader())
require.Eventually(t, ts.Candidate(0).IsLeader, 20*time.Second, 100*time.Millisecond)
require.False(t, ts.Candidate(1).IsLeader())
require.False(t, ts.Candidate(2).IsLeader())
{
w := NewWaiter()
w.Async(func() { require.Eventually(t, ts.Candidate(0).IsLeader, 20*time.Second, 100*time.Millisecond) })
w.Async(func() { require.Never(t, ts.Candidate(1).IsLeader, 20*time.Second, 100*time.Millisecond) })
w.Async(func() { require.Never(t, ts.Candidate(2).IsLeader, 20*time.Second, 100*time.Millisecond) })
w.Wait()
}
}
func TestFailover(t *testing.T) {
t.Parallel()
ts := NewTestSystem(t, 3)
ts := NewTestSystem(t, 3, 3)
defer ts.Stop()
require.False(t, ts.Candidate(0).IsLeader())
require.False(t, ts.Candidate(1).IsLeader())
require.False(t, ts.Candidate(2).IsLeader())
require.Eventually(t, ts.Candidate(0).IsLeader, 20*time.Second, 100*time.Millisecond)
require.False(t, ts.Candidate(1).IsLeader())
require.False(t, ts.Candidate(2).IsLeader())
{
w := NewWaiter()
w.Async(func() { require.Eventually(t, ts.Candidate(0).IsLeader, 20*time.Second, 100*time.Millisecond) })
w.Async(func() { require.Never(t, ts.Candidate(1).IsLeader, 20*time.Second, 100*time.Millisecond) })
w.Async(func() { require.Never(t, ts.Candidate(2).IsLeader, 20*time.Second, 100*time.Millisecond) })
w.Wait()
}
ts.SetServer(1)
require.Eventually(t, func() bool { return !ts.Candidate(0).IsLeader() }, 15*time.Second, 100*time.Millisecond)
// New candidate must not get leadership before old candidate loses it
require.False(t, ts.Candidate(1).IsLeader())
require.False(t, ts.Candidate(2).IsLeader())
require.Eventually(t, ts.Candidate(1).IsLeader, 20*time.Second, 100*time.Millisecond)
require.False(t, ts.Candidate(0).IsLeader())
require.False(t, ts.Candidate(2).IsLeader())
{
w := NewWaiter()
w.Async(func() { require.Eventually(t, ts.Candidate(1).IsLeader, 20*time.Second, 100*time.Millisecond) })
w.Async(func() { require.Never(t, ts.Candidate(0).IsLeader, 20*time.Second, 100*time.Millisecond) })
w.Async(func() { require.Never(t, ts.Candidate(2).IsLeader, 20*time.Second, 100*time.Millisecond) })
w.Wait()
}
}
func TestPartialVotes(t *testing.T) {
t.Parallel()
ts := NewTestSystem(t, 3)
ts := NewTestSystem(t, 3, 3)
defer ts.Stop()
ts.Proxy(0).SetRefuse(true)
@@ -72,7 +89,37 @@ func TestPartialVotes(t *testing.T) {
require.False(t, ts.Candidate(1).IsLeader())
require.False(t, ts.Candidate(2).IsLeader())
require.Eventually(t, ts.Candidate(0).IsLeader, 20*time.Second, 100*time.Millisecond)
{
w := NewWaiter()
w.Async(func() { require.Eventually(t, ts.Candidate(0).IsLeader, 20*time.Second, 100*time.Millisecond) })
w.Async(func() { require.Never(t, ts.Candidate(1).IsLeader, 20*time.Second, 100*time.Millisecond) })
w.Async(func() { require.Never(t, ts.Candidate(2).IsLeader, 20*time.Second, 100*time.Millisecond) })
w.Wait()
}
}
func TestSplitVotes(t *testing.T) {
t.Parallel()
ts := NewTestSystem(t, 3, 3)
defer ts.Stop()
ts.SetServerForVoter(1, 1)
ts.SetServerForVoter(1, 2)
require.False(t, ts.Candidate(0).IsLeader())
require.False(t, ts.Candidate(1).IsLeader())
require.False(t, ts.Candidate(2).IsLeader())
{
w := NewWaiter()
w.Async(func() { require.Eventually(t, ts.Candidate(1).IsLeader, 20*time.Second, 100*time.Millisecond) })
w.Async(func() { require.Never(t, ts.Candidate(0).IsLeader, 20*time.Second, 100*time.Millisecond) })
w.Async(func() { require.Never(t, ts.Candidate(2).IsLeader, 20*time.Second, 100*time.Millisecond) })
w.Wait()
}
}