jay brown
123 lines
3.0 KiB
Go
123 lines
3.0 KiB
Go
package queryapi_test
|
|
|
|
import (
|
|
"bytes"
|
|
"mime/multipart"
|
|
"net/http"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"queryorchestration/internal/database/repository"
|
|
"queryorchestration/internal/test"
|
|
|
|
queryapi "queryorchestration/api/queryAPI"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestUploadDocument(t *testing.T) {
|
|
cfg := &ControllerConfig{}
|
|
test.CreateDB(t, cfg)
|
|
initializeTestConfig(t, cfg)
|
|
acfg := test.CreateAWSContainer(t, cfg)
|
|
|
|
test.SetStoreClient(t, t.Context(), cfg, acfg.ExternalEndpoint)
|
|
test.CreateBucket(t, cfg)
|
|
|
|
svc := createControllerServices(cfg)
|
|
cons := queryapi.NewControllers(svc, cfg)
|
|
|
|
err := cfg.GetDBQueries().CreateClient(t.Context(), &repository.CreateClientParams{
|
|
Clientid: "client_id",
|
|
Name: "client_name",
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
var buf bytes.Buffer
|
|
writer := multipart.NewWriter(&buf)
|
|
|
|
fileWriter, err := writer.CreateFormFile("file", filepath.Base("../../assets/original/helloWorld.pdf"))
|
|
require.NoError(t, err)
|
|
|
|
_, err = fileWriter.Write([]byte("hi"))
|
|
require.NoError(t, err)
|
|
|
|
err = writer.Close()
|
|
require.NoError(t, err)
|
|
|
|
ctx, rec := createContextWithBody(t, buf.Bytes())
|
|
|
|
ctx.Request().Header.Set("Content-Type", writer.FormDataContentType())
|
|
|
|
err = cons.UploadDocument(ctx, "client_id")
|
|
require.NoError(t, err)
|
|
assert.Equal(t, http.StatusOK, rec.Code)
|
|
}
|
|
|
|
func TestListDocumentsByClientId(t *testing.T) {
|
|
cfg := &ControllerConfig{}
|
|
test.CreateDB(t, cfg)
|
|
initializeTestConfig(t, cfg)
|
|
|
|
svc := createControllerServices(cfg)
|
|
cons := queryapi.NewControllers(svc, 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",
|
|
BatchID: nil,
|
|
})
|
|
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{
|
|
queryapi.DocumentSummary{
|
|
Id: docId,
|
|
Hash: "hash",
|
|
},
|
|
})
|
|
}
|
|
|
|
func TestGetDocument(t *testing.T) {
|
|
cfg := &ControllerConfig{}
|
|
test.CreateDB(t, cfg)
|
|
initializeTestConfig(t, cfg)
|
|
|
|
svc := createControllerServices(cfg)
|
|
cons := queryapi.NewControllers(svc, 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",
|
|
BatchID: nil,
|
|
})
|
|
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{}{},
|
|
})
|
|
}
|