more test fixes

hashicorp · May 7, 2021 · bef417c · bef417c
1 parent 4174b02
commit bef417c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/command/agent/node_endpoint_test.go b/command/agent/node_endpoint_test.go
@@ -322,7 +322,7 @@ func TestHTTP_NodeDrain(t *testing.T) {
 		require.NotNil(out.LastDrain)
 		require.False(out.LastDrain.StartedAt.Before(beforeDrain))
 		require.False(out.LastDrain.UpdatedAt.Before(out.LastDrain.StartedAt))
-		require.Contains([]string{structs.DrainStatusCanceled, structs.DrainStatusComplete}, out.LastDrain.Status)
+		require.Contains([]structs.DrainStatus{structs.DrainStatusCanceled, structs.DrainStatusComplete}, out.LastDrain.Status)
 		if out.LastDrain.Status == structs.DrainStatusComplete {
 			require.Equal(map[string]string{
 				"reason": "drain",