package jsonl

import (
	"bytes"
	"errors"
	"strings"
	"testing"
)

// TestEachLongLine：**回归**——超过 bufio.Scanner 默认 64KB 上限的行必须能读出来。
// 这正是那个会话消失的原因（一条 212KB 的 assistant 消息）。
func TestEachLongLine(t *testing.T) {
	long := strings.Repeat("x", 300<<10) // 300KB
	in := "{\"a\":1}\n" + "{\"b\":\"" + long + "\"}\n" + "{\"c\":3}"
	var lines []int
	if err := Each(strings.NewReader(in), func(line []byte) { lines = append(lines, len(line)) }); err != nil {
		t.Fatalf("读取失败: %v", err)
	}
	if len(lines) != 3 {
		t.Fatalf("行数 = %d, want 3（最后一行无换行也要算）", len(lines))
	}
	if lines[1] < 300<<10 {
		t.Errorf("超长行被截断了: %d", lines[1])
	}
}

// TestEachSkipsEmptyAndCRLF：空行跳过；CRLF 行尾要去掉。
func TestEachSkipsEmpty(t *testing.T) {
	var got []string
	in := "\n{\"a\":1}\r\n\n\n{\"b\":2}\n"
	if err := Each(strings.NewReader(in), func(line []byte) { got = append(got, string(line)) }); err != nil {
		t.Fatal(err)
	}
	if len(got) != 2 || got[0] != "{\"a\":1}" || got[1] != "{\"b\":2}" {
		t.Errorf("实际: %q", got)
	}
}

// errReader 模拟读失败。
type errReader struct{ n int }

func (e *errReader) Read(p []byte) (int, error) {
	if e.n > 0 {
		e.n--
		copy(p, "{\"a\":1}\n")
		return 8, nil
	}
	return 0, errors.New("boom")
}

// TestEachPropagatesIOError：IO 错误要如实返回，不能当成正常结束。
func TestEachPropagatesIOError(t *testing.T) {
	err := Each(&errReader{n: 1}, func([]byte) {})
	if err == nil || !strings.Contains(err.Error(), "boom") {
		t.Errorf("应返回 IO 错误，实际 %v", err)
	}
	_ = bytes.MinRead
}