package queryapitest_test

import (
	"context"
	"testing"

	queryapitest "queryorchestration/internal/test/queryAPI"
	queryapimock "queryorchestration/mocks/queryapi"
	queryapi "queryorchestration/pkg/queryAPI"

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

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

	svcClient := queryapimock.NewMockClientWithResponsesInterface(t)

	svcClient.EXPECT().CreateClientWithResponse(
		mock.Anything,
		mock.MatchedBy(func(create queryapi.ClientCreate) bool {
			return create.Name == "example_name" && create.Id == "ID"
		}),
		mock.Anything,
	).Return(&queryapi.CreateClientResponse{
		JSON201: &queryapi.ClientIDBody{
			Id: "ID",
		},
	}, nil)

	svcClient.EXPECT().UpdateClientWithResponse(
		mock.Anything,
		mock.MatchedBy(func(id string) bool {
			return id == "ID"
		}),
		mock.MatchedBy(func(create queryapi.ClientUpdate) bool {
			return *create.CanSync == true
		}),
		mock.Anything,
	).Return(&queryapi.UpdateClientResponse{}, nil)

	svcClient.EXPECT().GetClientWithResponse(
		mock.Anything,
		mock.MatchedBy(func(id string) bool {
			return id == "ID"
		}),
		mock.Anything,
	).Return(&queryapi.GetClientResponse{
		JSON200: &queryapi.DocClient{
			Id: "ID",
		},
	}, nil)

	client := queryapitest.CreateClientWithSync(t, ctx, svcClient)

	assert.NotNil(t, client)
	assert.NotNil(t, client)
	assert.EqualExportedValues(t, queryapi.DocClient{
		Id: "ID",
	}, *client)
}

func TestWaitForClientStatus(t *testing.T) {
	ctx := context.Background()
	client := queryapimock.NewMockClientWithResponsesInterface(t)

	client.EXPECT().GetStatusByClientIdWithResponse(
		mock.Anything,
		mock.MatchedBy(func(id string) bool {
			return id == "id"
		}),
	).Return(&queryapi.GetStatusByClientIdResponse{
		JSON200: &queryapi.ClientStatusBody{
			Status: queryapi.INSYNC,
		},
	}, nil)

	queryapitest.WaitForClientStatus(t, ctx, client, "id", queryapi.INSYNC)
}