Merged in feature/docresult (pull request #105)

Document Result Endpoint * testing * cleantesting * progress * query * lint * readme * dockerclient * api * passtest * tests * test
2025-03-17 18:14:15 +00:00
parent f7c41c4ef3
commit 555b6d420b
105 changed files with 3276 additions and 2484 deletions
@@ -0,0 +1,59 @@
+package queryapitest
+
+import (
+	"context"
+	"fmt"
+	"testing"
+	"time"
+
+	queryapi "queryorchestration/pkg/queryAPI"
+
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+)
+
+func CreateClientWithSync(t testing.TB, ctx context.Context, client queryapi.ClientWithResponsesInterface) *queryapi.DocClient {
+	t.Helper()
+
+	clientCreateRes, err := client.CreateClientWithResponse(ctx, queryapi.ClientCreate{
+		Name: "example_name",
+		Id:   "ID",
+	})
+	require.NoError(t, err)
+
+	canSync := true
+	_, err = client.UpdateClientWithResponse(ctx, clientCreateRes.JSON201.Id, queryapi.ClientUpdate{
+		CanSync: &canSync,
+	})
+	require.NoError(t, err)
+
+	clientRes, err := client.GetClientWithResponse(ctx, clientCreateRes.JSON201.Id)
+	require.NoError(t, err)
+
+	return clientRes.JSON200
+}
+
+func WaitForClientStatus(t testing.TB, ctx context.Context, service queryapi.ClientWithResponsesInterface, id string, status queryapi.ClientStatus) {
+	t.Helper()
+
+	timeout := time.After(30 * time.Second)
+	ticker := time.NewTicker(500 * time.Millisecond)
+	defer ticker.Stop()
+
+	for {
+		select {
+		case <-timeout:
+			require.NoError(t, fmt.Errorf("Timeout waiting for client status to become %s", status))
+		case <-ticker.C:
+			jRes, err := service.GetStatusByClientIdWithResponse(ctx, id)
+			if err != nil {
+				assert.NoError(t, err)
+			}
+
+			if jRes.JSON200.Status == status {
+				assert.Equal(t, status, jRes.JSON200.Status)
+				return
+			}
+		}
+	}
+}