models_list_response_limit_test.go 713 B

12345678910111213141516171819202122232425
  1. package service
  2. import (
  3. "testing"
  4. "github.com/Wei-Shaw/sub2api/internal/config"
  5. "github.com/stretchr/testify/require"
  6. )
  7. func TestResolveModelsListReadLimit(t *testing.T) {
  8. t.Run("nil config uses default", func(t *testing.T) {
  9. require.Equal(t, config.DefaultModelsListReadMaxBytes, resolveModelsListReadLimit(nil))
  10. })
  11. t.Run("configured limit wins", func(t *testing.T) {
  12. cfg := &config.Config{}
  13. cfg.Gateway.ModelsListReadMaxBytes = 16 << 20
  14. require.Equal(t, int64(16<<20), resolveModelsListReadLimit(cfg))
  15. })
  16. t.Run("non-positive limit uses default", func(t *testing.T) {
  17. cfg := &config.Config{}
  18. require.Equal(t, config.DefaultModelsListReadMaxBytes, resolveModelsListReadLimit(cfg))
  19. })
  20. }