package clientsyncrunner_test import ( "fmt" "testing" clientsyncrunner "queryorchestration/api/clientSyncRunner" clientsync "queryorchestration/internal/client/sync" "queryorchestration/internal/database/repository" "queryorchestration/internal/server/runner" "queryorchestration/internal/serviceconfig/queue/documentsync" "queryorchestration/internal/test" queuemock "queryorchestration/mocks/queue" "github.com/aws/aws-sdk-go-v2/service/sqs" "github.com/stretchr/testify/assert" "github.com/stretchr/testify/mock" "github.com/stretchr/testify/require" ) type ClientSyncConfig struct { runner.BaseConfig[clientsyncrunner.Body] documentsync.DocSyncConfig } func TestQueryRunner(t *testing.T) { t.Parallel() cfg := &ClientSyncConfig{} net := test.GetNetwork(t) test.CreateDB(t, cfg, net, &test.CreateDatabaseConfig{}) mockSQS := queuemock.NewMockSQSClient(t) cfg.QueueClient = mockSQS cfg.DocumentSyncURL = "/i/am/here" svc := clientsync.New(cfg) runner := clientsyncrunner.New(&clientsyncrunner.Services{ ClientSync: svc, }) err := cfg.GetDBQueries().CreateClient(t.Context(), &repository.CreateClientParams{ Clientid: "client_id", Name: "client_name", }) require.NoError(t, err) docId, err := cfg.GetDBQueries().CreateDocument(t.Context(), &repository.CreateDocumentParams{ Clientid: "client_id", Hash: "hash", }) require.NoError(t, err) bod := clientsyncrunner.Body{ ClientID: "client_id", } mockSQS.EXPECT(). SendMessage( mock.Anything, mock.MatchedBy(func(in *sqs.SendMessageInput) bool { return *in.QueueUrl == cfg.DocumentSyncURL && *in.MessageBody == fmt.Sprintf("{\"id\":\"%s\"}", docId.String()) }), mock.Anything, ). Return(&sqs.SendMessageOutput{}, nil) assert.True(t, runner.Process(t.Context(), bod)) }