Sub2API v1.0 - AI API 网关（二开初始版本，基于上游 Wei-Shaw/sub2api）

backend/internal/service/gateway_request_pricing_test.go

@@ -0,0 +1,20 @@
+//go:build unit
+
+package service
+
+import (
+	"context"
+	"testing"
+
+	"github.com/stretchr/testify/require"
+)
+
+func TestWithGatewayTokenRequestPricingMarksOnlyExplicitTokenRequests(t *testing.T) {
+	ctx, pricingAt := WithGatewayTokenRequestPricing(context.Background())
+
+	got, ok := gatewayTokenRequestPricingAtFromContext(ctx)
+	require.True(t, ok)
+	require.Equal(t, pricingAt, got)
+	require.Equal(t, pricingAt, GatewayTokenRequestPricingAtFromContext(ctx))
+	require.True(t, GatewayTokenRequestPricingAtFromContext(context.Background()).IsZero())
+}