Michael McGuinness
555b6d420b
Document Result Endpoint * testing * cleantesting * progress * query * lint * readme * dockerclient * api * passtest * tests * test
60 lines
1.4 KiB
Go
60 lines
1.4 KiB
Go
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
|
|
}
|
|
}
|
|
}
|
|
}
|