From cb6a697109979c36200b393c45f58cde62f1b81a Mon Sep 17 00:00:00 2001 From: Alex Dadgar Date: Tue, 20 Feb 2018 15:05:16 -0800 Subject: [PATCH] add a revoke leadership test --- nomad/leader_test.go | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) diff --git a/nomad/leader_test.go b/nomad/leader_test.go index 4689cbfcbd59..03252f3993f5 100644 --- a/nomad/leader_test.go +++ b/nomad/leader_test.go @@ -13,6 +13,7 @@ import ( "github.com/hashicorp/nomad/nomad/structs" "github.com/hashicorp/nomad/testutil" "github.com/stretchr/testify/assert" + "github.com/stretchr/testify/require" ) func TestLeader_LeftServer(t *testing.T) { @@ -978,3 +979,19 @@ func TestLeader_RollRaftServer(t *testing.T) { }) } } + +func TestLeader_RevokeLeadership_MultipleTimes(t *testing.T) { + s1 := testServer(t, nil) + defer s1.Shutdown() + testutil.WaitForLeader(t, s1.RPC) + + testutil.WaitForResult(func() (bool, error) { + return s1.evalBroker.Enabled(), nil + }, func(err error) { + t.Fatalf("should have finished establish leader loop") + }) + + require.Nil(t, s1.revokeLeadership()) + require.Nil(t, s1.revokeLeadership()) + require.Nil(t, s1.revokeLeadership()) +}