package queryapi_test import ( "net/http" "testing" queryapi "queryorchestration/api/queryAPI" "queryorchestration/internal/database/repository" "queryorchestration/internal/document" "queryorchestration/internal/serviceconfig" "queryorchestration/internal/serviceconfig/queue/clientsync" "queryorchestration/internal/test" "github.com/stretchr/testify/assert" "github.com/stretchr/testify/require" ) func TestListDocumentsByClientId(t *testing.T) { t.Parallel() cfg := &struct { serviceconfig.BaseConfig clientsync.ClientSyncConfig }{} net := test.GetNetwork(t) test.CreateDB(t, cfg, net, &test.CreateDatabaseConfig{}) cons := queryapi.NewControllers(&queryapi.Services{ Document: document.New(cfg), }) 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 := &struct { serviceconfig.BaseConfig clientsync.ClientSyncConfig }{} net := test.GetNetwork(t) test.CreateDB(t, cfg, net, &test.CreateDatabaseConfig{}) cons := queryapi.NewControllers(&queryapi.Services{ Document: document.New(cfg), }) 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{}{}, }) }