package service

import (
	"context"
	"encoding/json"
	"net/http"
	"net/http/httptest"
	"testing"
	"time"
)

func TestWechatTemplateKey(t *testing.T) {
	cases := map[string]string{
		"green":   "",
		"yellow":  "inspection",
		"orange":  "inspection",
		"red":     "inspection",
		"":        "",
		"unknown": "",
	}
	for level, want := range cases {
		if got := WechatTemplateKey(level); got != want {
			t.Errorf("WechatTemplateKey(%q) = %q, want %q", level, got, want)
		}
	}
}

func TestIsAuthorized(t *testing.T) {
	auth := []string{"alarm", "inspection"}
	if !IsAuthorized(auth, "inspection") {
		t.Error("inspection 应在授权列表内")
	}
	if IsAuthorized(auth, "push") {
		t.Error("push 不应在授权列表内")
	}
	if IsAuthorized(nil, "inspection") {
		t.Error("空授权列表应返回 false")
	}
}

func TestBuildSubscribeData(t *testing.T) {
	data := BuildSubscribeData("orange", 75)
	if data["thing1"]["value"] != "橙" {
		t.Errorf("thing1 应为橙色等级，实际 %v", data["thing1"])
	}
	if data["thing2"]["value"] != "75分" {
		t.Errorf("thing2 应为 75分，实际 %v", data["thing2"])
	}
}

func TestWechatCode2Session(t *testing.T) {
	srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		if r.URL.Path != "/sns/jscode2session" {
			t.Errorf("path = %s", r.URL.Path)
		}
		q := r.URL.Query()
		if q.Get("appid") != "app1" || q.Get("secret") != "sec1" || q.Get("js_code") != "code123" {
			t.Errorf("参数不正确: %v", q)
		}
		_ = json.NewEncoder(w).Encode(map[string]string{"openid": "oAbC123", "session_key": "sk"})
	}))
	defer srv.Close()

	s := NewWechatService("app1", "sec1")
	s.baseURL = srv.URL
	openid, err := s.Code2Session(context.Background(), "code123")
	if err != nil {
		t.Fatalf("Code2Session 错误: %v", err)
	}
	if openid != "oAbC123" {
		t.Errorf("openid = %s", openid)
	}
}

func TestWechatSendSubscribe(t *testing.T) {
	srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		if r.URL.Path != "/cgi-bin/message/subscribe/send" {
			t.Errorf("path = %s", r.URL.Path)
		}
		var body map[string]any
		_ = json.NewDecoder(r.Body).Decode(&body)
		if body["touser"] != "oAbC123" || body["template_id"] != "tmpl1" {
			t.Errorf("body 不正确: %v", body)
		}
		_, _ = w.Write([]byte(`{"errcode":0,"errmsg":"ok"}`))
	}))
	defer srv.Close()

	s := NewWechatService("app1", "sec1")
	s.baseURL = srv.URL
	s.accessToken = "fake-token"
	s.tokenExpire = time.Now().Add(time.Hour)
	err := s.SendSubscribe(context.Background(), "oAbC123", "tmpl1", BuildSubscribeData("red", 88), "")
	if err != nil {
		t.Fatalf("SendSubscribe 错误: %v", err)
	}
}

func TestWechatSendSubscribeErrorCode(t *testing.T) {
	srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		_, _ = w.Write([]byte(`{"errcode":40003,"errmsg":"invalid openid"}`))
	}))
	defer srv.Close()

	s := NewWechatService("app1", "sec1")
	s.baseURL = srv.URL
	s.accessToken = "fake-token"
	s.tokenExpire = time.Now().Add(time.Hour)
	result, err := s.SendSubscribeResult(context.Background(), "bad", "tmpl1", nil, "")
	if err == nil {
		t.Error("errcode!=0 应返回错误")
	}
	if result.ErrCode != 40003 || result.ErrMsg != "invalid openid" {
		t.Errorf("response code = %+v", result)
	}
}