package proxy

import (
	"strings"
	"testing"
)

func TestCapRaw(t *testing.T) {
	small := []byte("hello")
	if got := capRaw(small, 10); got != "hello" {
		t.Fatalf("未超限应原样返回，实际 %q", got)
	}
	big := []byte(strings.Repeat("x", 100))
	got := capRaw(big, 10)
	if !strings.HasPrefix(got, strings.Repeat("x", 10)) {
		t.Fatalf("应保留前 10 字节：%q", got)
	}
	if !strings.Contains(got, "truncated by openteam: 100 bytes total") {
		t.Fatalf("应标注被截断与原始大小：%q", got)
	}
}

// 两个上限都远小于流式响应可能的体积，且响应上限比请求上限宽（响应通常是更大的那份）
func TestRawCapsSane(t *testing.T) {
	if rawRequestCap <= 0 || rawResponseCap <= 0 {
		t.Fatal("上限必须为正")
	}
	if rawRequestCap > rawResponseCap {
		t.Fatalf("响应上限（%d）应不小于请求上限（%d）", rawResponseCap, rawRequestCap)
	}
}