Clean up

internal/command/command_service_test.go

@@ -8,6 +8,7 @@ package command
 import (
 	"context"
 	"testing"
+	"time"
 
 	"github.com/nginx/agent/v3/api/grpc/mpi/v1/v1fakes"
 	"github.com/nginx/agent/v3/test/protos"
@@ -32,6 +33,27 @@ func (*FakeSubscribeClient) Recv() (*mpi.ManagementPlaneRequest, error) {
 	return nil, nil
 }
 
+func TestCommandService_NewCommandService(t *testing.T) {
+	ctx := context.Background()
+	commandServiceClient := &v1fakes.FakeCommandServiceClient{}
+
+	commandService := NewCommandService(
+		ctx,
+		commandServiceClient,
+		types.GetAgentConfig(),
+		make(chan *mpi.ManagementPlaneRequest),
+	)
+
+	defer commandService.CancelSubscription(ctx)
+
+	assert.Eventually(
+		t,
+		func() bool { return commandServiceClient.SubscribeCallCount() > 0 },
+		2*time.Second,
+		10*time.Millisecond,
+	)
+}
+
 func TestCommandService_UpdateDataPlaneStatus(t *testing.T) {
 	ctx := context.Background()