query-orchestration/api/queryAPI/documents_test.go

package queryapi_test

import (
	"net/http"
	"testing"

	queryapi "queryorchestration/api/queryAPI"
	"queryorchestration/internal/database/repository"
	"queryorchestration/internal/test"

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

func TestListDocumentsByClientId(t *testing.T) {
	t.Parallel()
	cfg := &ControllerConfig{}
	test.CreateDB(t, cfg)

	svc := createControllerServices(cfg)
	cons := queryapi.NewControllers(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)

	ctx, rec := createContext(t)

	err = cons.ListDocumentsByClientId(ctx, "client_id")
	require.NoError(t, err)
	assert.Equal(t, http.StatusOK, rec.Code)
	assertBody(t, rec, queryapi.ListDocuments{
		{
			Id:   docId,
			Hash: "hash",
		},
	})
}

func TestGetDocument(t *testing.T) {
	t.Parallel()
	cfg := &ControllerConfig{}
	test.CreateDB(t, cfg)

	svc := createControllerServices(cfg)
	cons := queryapi.NewControllers(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)

	ctx, rec := createContext(t)

	err = cons.GetDocument(ctx, docId)
	require.NoError(t, err)
	assert.Equal(t, http.StatusOK, rec.Code)
	assertBody(t, rec, queryapi.Document{
		Id:       docId,
		ClientId: "client_id",
		Hash:     "hash",
		Fields:   map[string]interface{}{},
	})
}