Add test

hashicorp · Apr 26, 2018 · 16272ad · 16272ad
1 parent 20b2ad3
commit 16272ad
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/nomad/client_rpc_test.go b/nomad/client_rpc_test.go
@@ -57,6 +57,7 @@ func TestServer_removeNodeConn_differentAddrs(t *testing.T) {
 	s1.addNodeConn(ctx1)
 	s1.addNodeConn(ctx2)
 	require.Len(s1.connectedNodes(), 1)
+	require.Len(s1.nodeConns[nodeID], 2)
 
 	// Check that the value is the second conn.
 	state, ok := s1.getNodeConn(nodeID)
@@ -66,6 +67,7 @@ func TestServer_removeNodeConn_differentAddrs(t *testing.T) {
 	// Delete the first
 	s1.removeNodeConn(ctx1)
 	require.Len(s1.connectedNodes(), 1)
+	require.Len(s1.nodeConns[nodeID], 1)
 
 	// Check that the value is the second conn.
 	state, ok = s1.getNodeConn(nodeID)