package jiyi

import (
	"encoding/json"
	"testing"
)

func TestIsEmptyBusinessResult(t *testing.T) {
	cases := []struct {
		name  string
		data  interface{}
		empty bool
	}{
		{"nil_map", map[string]interface{}{}, true},
		{"missing_result", map[string]interface{}{"foo": 1}, true},
		{"nil_result", map[string]interface{}{"result": nil}, true},
		{"empty_result_map", map[string]interface{}{"result": map[string]interface{}{}}, true},
		{"black_tags_zero_not_empty", map[string]interface{}{
			"result": map[string]interface{}{
				"black_list":  "0",
				"black_tag04": "0",
			},
		}, false},
		{"score_present", map[string]interface{}{
			"result": map[string]interface{}{
				"score":    "617",
				"reason":   "",
				"contents": map[string]interface{}{"TC_A001": "0"},
			},
		}, false},
		{"all_blank_strings", map[string]interface{}{
			"result": map[string]interface{}{
				"score":  "",
				"reason": "",
			},
		}, true},
	}

	for _, tc := range cases {
		t.Run(tc.name, func(t *testing.T) {
			got := isEmptyBusinessResult(tc.data)
			if got != tc.empty {
				t.Fatalf("got %v want %v", got, tc.empty)
			}
		})
	}
}

func TestUnwrapResultPayload(t *testing.T) {
	wuj := map[string]interface{}{
		"result": map[string]interface{}{
			"black_list":  "1",
			"black_tag04": "1",
		},
	}
	got := unwrapResultPayload(wuj).(map[string]interface{})
	if got["black_list"] != "1" || got["black_tag04"] != "1" {
		t.Fatalf("wuj unwrap mismatch: %+v", got)
	}
	if _, ok := got["result"]; ok {
		t.Fatal("should not keep result wrapper")
	}

	cave := map[string]interface{}{
		"result": map[string]interface{}{
			"score":  "617",
			"reason": "",
			"contents": map[string]interface{}{
				"TC_A001": "0",
			},
		},
	}
	got2 := unwrapResultPayload(cave).(map[string]interface{})
	if got2["score"] != "617" {
		t.Fatalf("cave unwrap mismatch: %+v", got2)
	}

	pano := map[string]interface{}{
		"result": map[string]interface{}{
			"apply_report_detail":    map[string]interface{}{"A22160001": "543"},
			"behavior_report_detail": nil,
			"current_report_detail":  nil,
		},
	}
	got3 := unwrapResultPayload(pano).(map[string]interface{})
	if _, ok := got3["apply_report_detail"]; !ok {
		t.Fatalf("pano unwrap mismatch: %+v", got3)
	}

	plain := map[string]interface{}{"x": "1"}
	if unwrapResultPayload(plain).(map[string]interface{})["x"] != "1" {
		t.Fatal("plain payload should keep")
	}
}

func TestStripAndUnwrapPipeline(t *testing.T) {
	raw := []byte(`{"busiCode":10,"busiMsg":"查询成功","result":{"black_list":"1","black_tag07":"1"}}`)
	var data interface{}
	if err := json.Unmarshal(raw, &data); err != nil {
		t.Fatal(err)
	}
	cleaned, err := stripBusiMetaFromData(data)
	if err != nil {
		t.Fatal(err)
	}
	out := unwrapResultPayload(cleaned).(map[string]interface{})
	if _, ok := out["busiCode"]; ok {
		t.Fatal("busiCode should be stripped")
	}
	if _, ok := out["result"]; ok {
		t.Fatal("result wrapper should be unwrapped")
	}
	if out["black_list"] != "1" || out["black_tag07"] != "1" {
		t.Fatalf("unexpected: %+v", out)
	}
}

func TestIsQueryBillingAPIKey(t *testing.T) {
	if !isQueryBillingAPIKey("jy000022") || !isQueryBillingAPIKey("jy000042") || !isQueryBillingAPIKey("jy000052") {
		t.Fatal("查空计费产品应在白名单中")
	}
	if isQueryBillingAPIKey("jy000004") {
		t.Fatal("信用司南查空不计费，不应在白名单中")
	}
	if isQueryBillingAPIKey("jy000002") || isQueryBillingAPIKey("jy000003") {
		t.Fatal("全景雷达BH查空不计费，不应在白名单中")
	}
}