query-orchestration/api/queryAPI/export_test.go

package queryapi_test

import (
	"net/http"
	"net/http/httptest"
	"strings"
	"testing"

	queryapi "queryorchestration/api/queryAPI"

	"github.com/google/uuid"
	"github.com/labstack/echo/v4"
	"github.com/stretchr/testify/assert"
)

func TestTriggerExport(t *testing.T) {
	e := echo.New()
	req := httptest.NewRequest(http.MethodPost, "/", strings.NewReader(""))
	rec := httptest.NewRecorder()
	ctx := e.NewContext(req, rec)

	cons := queryapi.NewControllers(&queryapi.Services{})

	err := cons.TriggerExport(ctx, "clientid")
	assert.NoError(t, err)
	assert.Equal(t, http.StatusOK, rec.Code)
	assert.NotEmpty(t, rec.Body.String())
}

func TestExportState(t *testing.T) {
	e := echo.New()
	req := httptest.NewRequest(http.MethodPost, "/", strings.NewReader(""))
	rec := httptest.NewRecorder()
	ctx := e.NewContext(req, rec)

	cons := queryapi.NewControllers(&queryapi.Services{})

	id := uuid.New()

	err := cons.ExportState(ctx, id)
	assert.NoError(t, err)
	assert.Equal(t, http.StatusOK, rec.Code)
	assert.NotEmpty(t, rec.Body.String())
}