da36506b89
P0 fixes: - ModelError.Is(): use exact matching instead of substring contains() - shouldClearStickySession: add context param for cancellation/tracing P1 fixes: - TODO stubs: return 501 Not Implemented errors - validateInstanceSignature: deduplicate to shared validateCodeSignature() - Error messages: standardize to English only - http.go: remove pseudo if-else with duplicate branches
38 lines
1.1 KiB
Go
38 lines
1.1 KiB
Go
package service
|
|
|
|
import (
|
|
"bytes"
|
|
"errors"
|
|
"testing"
|
|
|
|
"github.com/Wei-Shaw/sub2api/internal/config"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestResolveUpstreamResponseReadLimit(t *testing.T) {
|
|
t.Run("use default when config missing", func(t *testing.T) {
|
|
require.Equal(t, defaultUpstreamResponseReadMaxBytes, resolveUpstreamResponseReadLimit(nil))
|
|
})
|
|
|
|
t.Run("use configured value", func(t *testing.T) {
|
|
cfg := &config.Config{}
|
|
cfg.Gateway.UpstreamResponseReadMaxBytes = 1234
|
|
require.Equal(t, int64(1234), resolveUpstreamResponseReadLimit(cfg))
|
|
})
|
|
}
|
|
|
|
func TestReadUpstreamResponseBodyLimited(t *testing.T) {
|
|
t.Run("within limit", func(t *testing.T) {
|
|
body, err := readUpstreamResponseBodyLimited(bytes.NewReader([]byte("ok")), 2)
|
|
require.NoError(t, err)
|
|
require.Equal(t, []byte("ok"), body)
|
|
})
|
|
|
|
t.Run("exceeds limit", func(t *testing.T) {
|
|
body, err := readUpstreamResponseBodyLimited(bytes.NewReader([]byte("toolong")), 3)
|
|
require.Nil(t, body)
|
|
require.Error(t, err)
|
|
require.True(t, errors.Is(err, ErrUpstreamResponseBodyTooLarge))
|
|
})
|
|
}
|