Merged in feature/import (pull request #157)

Feature/import

* importstart

* pp

* tests

* importtests

* 100GB

* lint

* passtests

* utc

* awsprofile

* doublequotes

* host

internal/test/queryAPI/service.go

@@ -1,14 +1,17 @@
 package queryapitest
 
 import (
+	"bytes"
 	"context"
 	"fmt"
 	"log/slog"
+	"mime/multipart"
 	"net/http"
 	"testing"
 	"time"
 
 	"queryorchestration/internal/test"
+
 	queryapi "queryorchestration/pkg/queryAPI"
 
 	"github.com/google/uuid"
@@ -17,6 +20,35 @@ import (
 	"github.com/stretchr/testify/require"
 )
 
+type File struct {
+	ClientID queryapi.ClientID
+	Filename string
+	Content  []byte
+}
+
+func CreateFile(t testing.TB, client queryapi.ClientWithResponsesInterface, file File) {
+	var buf bytes.Buffer
+	fileBody := multipart.NewWriter(&buf)
+
+	name := file.Filename
+	if name == "" {
+		name = "helloworld"
+	}
+
+	write, err := fileBody.CreateFormFile("file", name)
+	require.NoError(t, err)
+	_, err = write.Write(file.Content)
+	require.NoError(t, err)
+	err = fileBody.Close()
+	require.NoError(t, err)
+
+	body := bytes.NewReader(buf.Bytes())
+
+	uploadRes, err := client.UploadDocumentWithBodyWithResponse(t.Context(), file.ClientID, fileBody.FormDataContentType(), body)
+	require.NoError(t, err)
+	require.Equal(t, http.StatusOK, uploadRes.StatusCode())
+}
+
 func CreateClientWithSync(t testing.TB, client queryapi.ClientWithResponsesInterface) queryapi.ClientID {
 	t.Helper()