query-orchestration/internal/server/api/swagger_test.go

package api

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

	queryapi "queryorchestration/api/queryAPI"

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

func TestGetSwaggerJSONFunc(t *testing.T) {
	openapi, err := queryapi.GetSwagger()
	require.NoError(t, err)

	f, err := getSwaggerJSONFunc(openapi)
	require.NoError(t, err)
	require.NotNil(t, f)

	req := httptest.NewRequest(http.MethodPost, "/", strings.NewReader("hello"))
	rec := httptest.NewRecorder()
	ctx := echo.New().NewContext(req, rec)

	err = f(ctx)
	require.NoError(t, err)
	assert.Equal(t, http.StatusOK, rec.Result().StatusCode)
}

func TestGetSwaggerYAMLFunc(t *testing.T) {
	openapi, err := queryapi.GetSwagger()
	require.NoError(t, err)

	f, err := getSwaggerYAMLFunc(openapi)
	require.NoError(t, err)
	require.NotNil(t, f)

	req := httptest.NewRequest(http.MethodPost, "/", strings.NewReader("hello"))
	rec := httptest.NewRecorder()
	ctx := echo.New().NewContext(req, rec)

	err = f(ctx)
	require.NoError(t, err)
	assert.Equal(t, http.StatusOK, rec.Result().StatusCode)
}