new-api/service/asset_compatible_test.go

package service

import (
	"context"
	"net/http"
	"testing"

	"github.com/QuantumNous/new-api/model"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

func TestCompatibleAssetAdapterBlocksPrivateBaseURL(t *testing.T) {
	baseURL := "http://127.0.0.1:8080"
	adapter := NewCompatibleAssetAdapter("compatible", []AssetOperation{AssetOperationAssetCreate})
	spec, ok := ParseAssetAction("CreateAsset")
	require.True(t, ok)

	_, assetErr := adapter.DoAssetRequest(context.Background(), &model.Channel{Key: "key", BaseURL: &baseURL}, AssetRequest{
		Action:  spec,
		Version: "2024-01-01",
		RawBody: []byte(`{}`),
	})

	require.NotNil(t, assetErr)
	assert.Equal(t, http.StatusForbidden, assetErr.HTTPStatus)
}