package fqapi

import (
	"bytes"
	"encoding/json"
	"strings"
	"testing"
)

// TestFilterSearchTabs 上游返回全部 tab 时,应只保留请求的 tab_type;无匹配时保留原样
func TestFilterSearchTabs(t *testing.T) {
	raw := `{"code":0,"search_tabs":[
		{"tab_type":1,"title":"综合"},
		{"tab_type":3,"title":"书籍","book_id":7123456789012345678}
	]}`

	decode := func(s string) map[string]interface{} {
		dec := json.NewDecoder(strings.NewReader(s))
		dec.UseNumber()
		var m map[string]interface{}
		if err := dec.Decode(&m); err != nil {
			t.Fatal(err)
		}
		return m
	}

	m := decode(raw)
	filterSearchTabs(m, 3)
	tabs := m["search_tabs"].([]interface{})
	if len(tabs) != 1 {
		t.Fatalf("tab_type=3 应只剩 1 个 tab, got %d", len(tabs))
	}
	// 大整数 ID 不得丢精度
	want := "7123456789012345678"
	got := tabs[0].(map[string]interface{})["book_id"].(json.Number).String()
	if got != want {
		t.Errorf("book_id 精度丢失: got %s want %s", got, want)
	}

	// 无匹配 tab 时保留原样
	m = decode(raw)
	filterSearchTabs(m, 99)
	if len(m["search_tabs"].([]interface{})) != 2 {
		t.Error("无匹配时应保留原始 search_tabs")
	}
}

// TestToJSONPreservesNumberID 重组后大整数 ID 字符串不变
func TestToJSONPreservesNumberID(t *testing.T) {
	dec := json.NewDecoder(bytes.NewReader([]byte(`{"id":7123456789012345678}`)))
	dec.UseNumber()
	var m map[string]interface{}
	if err := dec.Decode(&m); err != nil {
		t.Fatal(err)
	}
	out, err := ToJSON(m)
	if err != nil {
		t.Fatal(err)
	}
	if !strings.Contains(out, "7123456789012345678") {
		t.Errorf("重组丢失精度: %s", out)
	}
}