package integration_test

import (
	"context"
	serviceinterfaces "queryorchestration/api/serviceInterfaces"
	"queryorchestration/internal/test"
	"testing"

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

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

	conn, cleanup := test.CreateAPIWithDependencies(t, ctx, "queryService")
	defer cleanup()

	client := serviceinterfaces.NewQueryServiceClient(conn)

	idRes, err := client.Create(ctx, &serviceinterfaces.QueryCreate{
		Type:            serviceinterfaces.QueryType_QUERY_TYPE_CONTEXT_FULL,
		Config:          nil,
		RequiredQueries: []string{},
	})
	assert.Nil(t, err)
	id := idRes.GetId()
	assert.NotEmpty(t, id)

	queryRes, err := client.Get(ctx, &serviceinterfaces.IdMessage{
		Id: id,
	})
	assert.Nil(t, err)
	assert.Equal(t, id, queryRes.Id)
	assert.Equal(t, serviceinterfaces.QueryType_QUERY_TYPE_CONTEXT_FULL, queryRes.Type)
	assert.Equal(t, int32(1), queryRes.ActiveVersion)
	assert.Equal(t, int32(1), queryRes.LatestVersion)
	assert.Equal(t, "", *queryRes.Config)
	assert.Equal(t, []string(nil), queryRes.RequiredQueries)

	res, err := client.Update(ctx, &serviceinterfaces.QueryUpdate{
		Id:              id,
		Config:          nil,
		ActiveVersion:   nil,
		RequiredQueries: []string{},
	})
	assert.Nil(t, err)
	assert.NotNil(t, res)

	queryRes, err = client.Get(ctx, &serviceinterfaces.IdMessage{
		Id: id,
	})
	assert.Nil(t, err)
	assert.Equal(t, id, queryRes.Id)
	assert.Equal(t, serviceinterfaces.QueryType_QUERY_TYPE_CONTEXT_FULL, queryRes.Type)
	assert.Equal(t, int32(1), queryRes.ActiveVersion)
	assert.Equal(t, int32(1), queryRes.LatestVersion)
	assert.Equal(t, "", *queryRes.Config)
	assert.Equal(t, []string(nil), queryRes.RequiredQueries)
}