query-orchestration/test/queryService/client_test.go

package endtoend

import (
	"context"
	"queryorchestration/internal/test"
	queryservice "queryorchestration/pkg/queryService"
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestClient(t *testing.T) {
	ctx := context.Background()

	c, cleanup := test.CreateServiceNetwork(t, ctx, &test.ServiceNetworkConfig{
		Name: test.QueryService,
		Env: map[string]string{
			"JOB_SYNC_URL": "/i/am/here",
		},
	})
	defer cleanup()

	client, err := queryservice.NewClientWithResponses(c.URI)
	assert.NoError(t, err)

	idRes, err := client.CreateClientWithResponse(ctx, queryservice.ClientCreate{
		Name: "example_name",
	})
	assert.NoError(t, err)
	assert.NotNil(t, idRes)
	assert.NotNil(t, idRes.JSON201)
	assert.NotNil(t, idRes.JSON201.Id)
	id := idRes.JSON201.Id

	clientRes, err := client.GetClientWithResponse(ctx, id)
	assert.NoError(t, err)
	assert.Equal(t, id, clientRes.JSON200.Id)
	assert.Equal(t, "example_name", clientRes.JSON200.Name)
	assert.False(t, clientRes.JSON200.CanSync)

	updateName := "update_name"
	updateCanSync := true
	updateRes, err := client.UpdateClientWithResponse(ctx, id, queryservice.ClientUpdate{
		Name:    &updateName,
		CanSync: &updateCanSync,
	})
	assert.NoError(t, err)
	assert.NotNil(t, updateRes)

	clientRes, err = client.GetClientWithResponse(ctx, id)
	assert.NoError(t, err)
	assert.Equal(t, id, clientRes.JSON200.Id)
	assert.Equal(t, updateName, clientRes.JSON200.Name)
	assert.True(t, clientRes.JSON200.CanSync)
}