chore: more assertions

nginx · Oct 24, 2024 · 2a61815 · 2a61815
1 parent d47bb83
commit 2a61815
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/internal/command/command_plugin_test.go b/internal/command/command_plugin_test.go
@@ -96,7 +96,8 @@ func TestCommandPlugin_Process(t *testing.T) {
 	commandPlugin.Process(ctx, &bus.Message{Topic: bus.DataPlaneResponseTopic, Data: protos.OKDataPlaneResponse()})
 	require.Equal(t, 1, fakeCommandService.SendDataPlaneResponseCallCount())
 
-	commandPlugin.Process(ctx, &bus.Message{Topic: bus.DataplaneHealthProcessTopic, Data: protos.OKDataPlaneResponse()})
+	commandPlugin.Process(ctx, &bus.Message{Topic: bus.DataplaneHealthProcessTopic, Data: protos.GetHealthyInstanceHealth()})
+	require.Equal(t, 1, fakeCommandService.UpdateDataPlaneHealthCallCount())
 	require.Equal(t, 1, fakeCommandService.SendDataPlaneResponseCallCount())
 }