From 764cdac01aff837c84cc0cb7490e5424906bcef5 Mon Sep 17 00:00:00 2001
From: Mahmood Ali <mahmood@hashicorp.com>
Date: Mon, 1 Jun 2020 10:18:59 -0400
Subject: [PATCH] test for paused workers upon leadership revocation

---
 nomad/leader_test.go | 31 +++++++++++++++++++++++++++++++
 1 file changed, 31 insertions(+)

diff --git a/nomad/leader_test.go b/nomad/leader_test.go
index 03333f639..d88c8aac4 100644
--- a/nomad/leader_test.go
+++ b/nomad/leader_test.go
@@ -1325,6 +1325,37 @@ func TestLeader_TransitionsUpdateConsistencyRead(t *testing.T) {
 	require.True(t, s1.isReadyForConsistentReads())
 }
 
+// TestLeader_PausingWorkers asserts that scheduling workers are paused
+// (and unpaused) upon leader elections (and step downs).
+func TestLeader_PausingWorkers(t *testing.T) {
+	s1, cleanupS1 := TestServer(t, func(c *Config) {
+		c.NumSchedulers = 12
+	})
+	defer cleanupS1()
+	testutil.WaitForLeader(t, s1.RPC)
+	require.Len(t, s1.workers, 12)
+
+	pausedWorkers := func() int {
+		c := 0
+		for _, w := range s1.workers {
+			w.pauseLock.Lock()
+			if w.paused {
+				c++
+			}
+			w.pauseLock.Unlock()
+		}
+		return c
+	}
+
+	// pause 3/4 of the workers
+	require.Equal(t, 9, pausedWorkers())
+
+	err := s1.revokeLeadership()
+	require.NoError(t, err)
+
+	require.Zero(t, pausedWorkers())
+}
+
 // Test doing an inplace upgrade on a server from raft protocol 2 to 3
 // This verifies that removing the server and adding it back with a uuid works
 // even if the server's address stays the same.