new-api/controller/ratio_config_test.go

package controller

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

	"github.com/QuantumNous/new-api/setting/ratio_setting"
	"github.com/gin-gonic/gin"
	"github.com/stretchr/testify/require"
)

func TestGetRatioConfigRejectsRequestWhenExposureIsDisabled(t *testing.T) {
	ratio_setting.SetExposeRatioEnabled(false)
	t.Cleanup(func() { ratio_setting.SetExposeRatioEnabled(false) })
	context, _ := gin.CreateTestContext(httptest.NewRecorder())

	GetRatioConfig(context)

	require.Equal(t, http.StatusForbidden, context.Writer.Status())
}

func TestGetRatioConfigReturnsDataWhenExposureIsEnabled(t *testing.T) {
	ratio_setting.SetExposeRatioEnabled(true)
	t.Cleanup(func() { ratio_setting.SetExposeRatioEnabled(false) })
	context, _ := gin.CreateTestContext(httptest.NewRecorder())

	GetRatioConfig(context)

	require.Equal(t, http.StatusOK, context.Writer.Status())
}