query-orchestration/api/clientSyncRunner/runner_test.go

package clientsyncrunner_test

import (
	"fmt"
	"regexp"
	"testing"

	clientsyncrunner "queryorchestration/api/clientSyncRunner"
	clientsync "queryorchestration/internal/client/sync"
	"queryorchestration/internal/database/repository"
	"queryorchestration/internal/server/runner"
	"queryorchestration/internal/serviceconfig/objectstore"
	"queryorchestration/internal/serviceconfig/queue"
	"queryorchestration/internal/serviceconfig/queue/documentsync"
	"queryorchestration/internal/test"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

type ClientSyncConfig struct {
	runner.BaseConfig[clientsyncrunner.Body]
	documentsync.DocSyncConfig
	queue.QueueConfig
	objectstore.ObjectStoreConfig
}

func TestClientSyncRunner(t *testing.T) {
	cfg := &ClientSyncConfig{}
	test.CreateDB(t, cfg)
	acfg := test.CreateAWSContainer(t, cfg)

	test.SetQueueClient(t, t.Context(), cfg, acfg.ExternalEndpoint)
	cfg.DocumentSyncURL = test.CreateQueue(t, cfg, test.DocSyncRunnerName)

	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",
	}

	assert.True(t, runner.Process(t.Context(), bod))

	test.AssertMessageBody(t, cfg, cfg.GetDocumentSyncURL(), regexp.MustCompile(fmt.Sprintf(`{"id":"%s"}`, docId.String())))
}