Fix return type in tests after refactor

hashicorp · Oct 30, 2018 · f2b0277 · f2b0277
1 parent 17344a7
commit f2b0277
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/command/agent/operator_endpoint_test.go b/command/agent/operator_endpoint_test.go
@@ -294,10 +294,10 @@ func TestOperator_SchedulerSetConfiguration(t *testing.T) {
 			},
 		}
 
-		var reply structs.SchedulerConfiguration
+		var reply structs.SchedulerConfigurationResponse
 		err = s.RPC("Operator.SchedulerGetConfiguration", &args, &reply)
 		require.Nil(err)
-		require.True(reply.PreemptionConfig.SystemSchedulerEnabled)
+		require.True(reply.SchedulerConfig.PreemptionConfig.SystemSchedulerEnabled)
 	})
 }
 
@@ -320,11 +320,11 @@ func TestOperator_SchedulerCASConfiguration(t *testing.T) {
 			},
 		}
 
-		var reply structs.SchedulerConfiguration
+		var reply structs.SchedulerConfigurationResponse
 		if err := s.RPC("Operator.SchedulerGetConfiguration", &args, &reply); err != nil {
 			t.Fatalf("err: %v", err)
 		}
-		require.True(reply.PreemptionConfig.SystemSchedulerEnabled)
+		require.True(reply.SchedulerConfig.PreemptionConfig.SystemSchedulerEnabled)
 
 		// Create a CAS request, bad index
 		{
@@ -354,6 +354,6 @@ func TestOperator_SchedulerCASConfiguration(t *testing.T) {
 		if err := s.RPC("Operator.SchedulerGetConfiguration", &args, &reply); err != nil {
 			t.Fatalf("err: %v", err)
 		}
-		require.False(reply.PreemptionConfig.SystemSchedulerEnabled)
+		require.False(reply.SchedulerConfig.PreemptionConfig.SystemSchedulerEnabled)
 	})
 }