backend/internal/pkg/jpush/vendor/payload_compat_test.go

package vendor

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

	"with_you/internal/config"
)

// TestPayloadEmptyConfigMatchesLegacy 验证：未配置任何 channel/模板时，
// Factory 产出的推送 payload 与改动前 BuildNotification 行为一致。
//
// 改动前 BuildNotification(title, body, "message", extras) 产出：
//   notification: { alert, android:{alert,title,extras:{notification_type}+extras,priority:1}, ios:{alert:{title,body},sound:default,badge:+1,extras} }
//   options: { time_to_live: 86400, apns_production }   // 无 third_party_channel
func TestPayloadEmptyConfigMatchesLegacy(t *testing.T) {
	factory := NewFactory()
	cfg := config.ChannelConfig{} // 全空配置
	params := MessageParams{
		Scene:       SceneChat,
		Title:       "张三",
		Body:        "你好",
		MessageType: "message",
		Extra: map[string]any{
			"message_id":      "m1",
			"conversation_id": "c1",
			"sender_id":       "u1",
			"category":        "chat",
		},
	}

	payload := factory.Build(params, cfg)

	// 1. TPC 必须为 nil（无 third_party_channel 下发，与改动前一致）
	if payload.TPC != nil {
		t.Fatalf("TPC should be nil when all channel config empty, got: %+v", payload.TPC)
	}

	notif := payload.Notification
	if notif == nil || notif.Android == nil || notif.IOS == nil {
		t.Fatalf("notification/android/ios must not be nil")
	}

	// 2. Android 字段对比
	if notif.Android.Alert != "你好" || notif.Android.Title != "张三" || notif.Android.Priority != 1 {
		t.Errorf("Android basic fields mismatch: %+v", notif.Android)
	}
	if notif.Android.ChannelID != "" {
		t.Errorf("Android.ChannelID = %q, want empty", notif.Android.ChannelID)
	}
	if notif.Android.Extras["notification_type"] != "message" || notif.Android.Extras["message_id"] != "m1" {
		t.Errorf("Android.Extras mismatch: %+v", notif.Android.Extras)
	}

	// 3. iOS 字段对比
	if notif.IOS.Sound != "default" || notif.IOS.Badge != "+1" {
		t.Errorf("IOS basic fields mismatch: %+v", notif.IOS)
	}
	if notif.IOS.ThreadID != "" {
		t.Errorf("IOS.ThreadID = %q, want empty", notif.IOS.ThreadID)
	}
	if notif.IOS.Extras["notification_type"] != "message" {
		t.Errorf("IOS.Extras[notification_type] mismatch: %+v", notif.IOS.Extras)
	}

	// 4. 序列化 JSON 确认厂商私有字段不出现
	jsonBytes, err := json.Marshal(notif)
	if err != nil {
		t.Fatalf("marshal: %v", err)
	}
	jsonStr := string(jsonBytes)
	for _, key := range []string{`"channel_id"`, `"thread-id"`, `"third_party_channel"`, `"mi_template"`, `"private_msg_template_id"`} {
		if strings.Contains(jsonStr, key) {
			t.Errorf("notification JSON should not contain %s when config empty, got: %s", key, jsonStr)
		}
	}
	t.Logf("notification JSON: %s", jsonStr)

	// 5. classification 必须为系统消息（1）：聊天即时消息统一归系统消息避免限速
	if payload.Classification != ClassificationSystem {
		t.Errorf("Classification = %d, want %d (system)", payload.Classification, ClassificationSystem)
	}
}

// TestClassificationByScene 验证：system 与 chat 场景的 classification 均为 1（系统消息）
// 聊天即时消息归为系统消息，避免厂商运营消息限速影响送达
func TestClassificationByScene(t *testing.T) {
	factory := NewFactory()
	cfg := config.ChannelConfig{}

	for _, scene := range []string{SceneSystem, SceneChat} {
		params := MessageParams{Scene: scene, Title: "T", Body: "B"}
		payload := factory.Build(params, cfg)
		if payload.Classification != ClassificationSystem {
			t.Errorf("scene=%s: Classification = %d, want %d", scene, payload.Classification, ClassificationSystem)
		}
	}
}

// TestPayloadWithChannelConfig 验证：配置 channel_id 后正确下发，且不影响通用通知字段
func TestPayloadWithChannelConfig(t *testing.T) {
	factory := NewFactory()
	cfg := config.ChannelConfig{
		System: "sys_channel",
		Chat:   "chat_channel",
	}
	params := MessageParams{Scene: SceneChat, Title: "T", Body: "B", MessageType: "chat_message"}

	payload := factory.Build(params, cfg)

	// 通用通知仍正常
	if payload.Notification.Android.Title != "T" {
		t.Errorf("Android.Title = %q, want T", payload.Notification.Android.Title)
	}
	// Android channel_id 跟随场景（chat）
	if payload.Notification.Android.ChannelID != "chat_channel" {
		t.Errorf("Android.ChannelID = %q, want chat_channel", payload.Notification.Android.ChannelID)
	}
	// TPC 各厂商 channel_id 下发（华为/VIVO 等回退全局）
	if payload.TPC == nil {
		t.Fatal("TPC should not be nil when channel config present")
	}
	if payload.TPC.Huawei["channel_id"] != "chat_channel" {
		t.Errorf("Huawei channel_id = %v, want chat_channel", payload.TPC.Huawei["channel_id"])
	}
	t.Logf("TPC JSON: %s", mustJSON(payload.TPC))
}

// TestPayloadXiaomiTemplateOptional 验证：仅配 channel_id 不配模板时，小米不下发模板字段
func TestPayloadXiaomiTemplateOptional(t *testing.T) {
	factory := NewFactory()
	cfg := config.ChannelConfig{
		Vendor: config.VendorChannel{
			Xiaomi: config.VendorChannelID{Chat: "mi_chat"},
		},
	}
	params := MessageParams{Scene: SceneChat, Title: "T", Body: "B"}

	payload := factory.Build(params, cfg)
	if payload.TPC == nil || payload.TPC.Xiaomi == nil {
		t.Fatal("Xiaomi TPC should exist")
	}
	if _, ok := payload.TPC.Xiaomi["mi_template_id"]; ok {
		t.Error("Xiaomi should NOT contain mi_template_id when template not configured")
	}
	if payload.TPC.Xiaomi["channel_id"] != "mi_chat" {
		t.Errorf("Xiaomi channel_id = %v, want mi_chat", payload.TPC.Xiaomi["channel_id"])
	}
}

// TestOppoCategoryNotifyLevel 验证：OPPO category/notify_level 按场景区分，且 notify_level 必须伴 category
func TestOppoCategoryNotifyLevel(t *testing.T) {
	factory := NewFactory()

	// 场景1：chat 用 IM + 16（强提醒）
	cfgChat := config.ChannelConfig{
		Vendor: config.VendorChannel{
			OPPO: config.VendorChannelID{
				Chat:               "oppo_chat",
				OppoChatCategory:   "IM",
				OppoChatNotifyLevel: 16,
			},
		},
	}
	payload := factory.Build(MessageParams{Scene: SceneChat, Title: "T", Body: "B"}, cfgChat)
	if payload.TPC == nil || payload.TPC.OPPO == nil {
		t.Fatal("OPPO TPC should exist for chat scene")
	}
	if payload.TPC.OPPO["category"] != "IM" {
		t.Errorf("OPPO chat category = %v, want IM", payload.TPC.OPPO["category"])
	}
	if payload.TPC.OPPO["notify_level"] != 16 {
		t.Errorf("OPPO chat notify_level = %v, want 16", payload.TPC.OPPO["notify_level"])
	}

	// 场景2：system 用 ACCOUNT + 2（通知栏+锁屏）
	cfgSys := config.ChannelConfig{
		Vendor: config.VendorChannel{
			OPPO: config.VendorChannelID{
				System:               "oppo_sys",
				OppoSystemCategory:   "ACCOUNT",
				OppoSystemNotifyLevel: 2,
			},
		},
	}
	payload = factory.Build(MessageParams{Scene: SceneSystem, Title: "T", Body: "B"}, cfgSys)
	if payload.TPC.OPPO["category"] != "ACCOUNT" {
		t.Errorf("OPPO system category = %v, want ACCOUNT", payload.TPC.OPPO["category"])
	}
	if payload.TPC.OPPO["notify_level"] != 2 {
		t.Errorf("OPPO system notify_level = %v, want 2", payload.TPC.OPPO["notify_level"])
	}
}

// TestOppoNotifyLevelRequiresCategory 验证：notify_level 配置但 category 未配置时，跳过 notify_level（避免 OPPO 校验失败）
func TestOppoNotifyLevelRequiresCategory(t *testing.T) {
	factory := NewFactory()
	cfg := config.ChannelConfig{
		Vendor: config.VendorChannel{
			OPPO: config.VendorChannelID{
				Chat:               "oppo_chat",
				OppoChatNotifyLevel: 16, // 只配 notify_level，不配 category
			},
		},
	}
	payload := factory.Build(MessageParams{Scene: SceneChat, Title: "T", Body: "B"}, cfg)
	if payload.TPC == nil || payload.TPC.OPPO == nil {
		t.Fatal("OPPO TPC should exist (has channel_id)")
	}
	if _, ok := payload.TPC.OPPO["notify_level"]; ok {
		t.Error("OPPO should NOT contain notify_level when category not configured")
	}
	if _, ok := payload.TPC.OPPO["category"]; ok {
		t.Error("OPPO should NOT contain category when not configured")
	}
	// channel_id 仍正常下发
	if payload.TPC.OPPO["channel_id"] != "oppo_chat" {
		t.Errorf("OPPO channel_id = %v, want oppo_chat", payload.TPC.OPPO["channel_id"])
	}
}

// TestHonorImportanceByScene 验证：荣耀 importance 按场景区分（chat=NORMAL, system=LOW）
func TestHonorImportanceByScene(t *testing.T) {
	factory := NewFactory()
	cfg := config.ChannelConfig{
		Vendor: config.VendorChannel{
			Honor: config.VendorChannelID{
				Chat:                "honor_chat",
				System:              "honor_sys",
				HonorChatImportance:   "NORMAL",
				HonorSystemImportance: "LOW",
			},
		},
	}

	// chat 场景 -> NORMAL
	payload := factory.Build(MessageParams{Scene: SceneChat, Title: "T", Body: "B"}, cfg)
	if payload.TPC == nil || payload.TPC.Honor == nil {
		t.Fatal("Honor TPC should exist for chat scene")
	}
	if payload.TPC.Honor["importance"] != "NORMAL" {
		t.Errorf("Honor chat importance = %v, want NORMAL", payload.TPC.Honor["importance"])
	}

	// system 场景 -> LOW
	payload = factory.Build(MessageParams{Scene: SceneSystem, Title: "T", Body: "B"}, cfg)
	if payload.TPC.Honor["importance"] != "LOW" {
		t.Errorf("Honor system importance = %v, want LOW", payload.TPC.Honor["importance"])
	}
}

// TestHonorImportanceOptional 验证：未配置 importance 时不下发，仅 channel_id
func TestHonorImportanceOptional(t *testing.T) {
	factory := NewFactory()
	cfg := config.ChannelConfig{
		Vendor: config.VendorChannel{
			Honor: config.VendorChannelID{Chat: "honor_chat"},
		},
	}
	payload := factory.Build(MessageParams{Scene: SceneChat, Title: "T", Body: "B"}, cfg)
	if payload.TPC == nil || payload.TPC.Honor == nil {
		t.Fatal("Honor TPC should exist (has channel_id)")
	}
	if _, ok := payload.TPC.Honor["importance"]; ok {
		t.Error("Honor should NOT contain importance when not configured")
	}
	if payload.TPC.Honor["channel_id"] != "honor_chat" {
		t.Errorf("Honor channel_id = %v, want honor_chat", payload.TPC.Honor["channel_id"])
	}
}

// TestVivoCategoryByScene 验证：vivo category 按场景区分
func TestVivoCategoryByScene(t *testing.T) {
	factory := NewFactory()
	cfg := config.ChannelConfig{
		Vendor: config.VendorChannel{
			VIVO: config.VendorChannelID{
				Chat:             "vivo_chat",
				System:           "vivo_sys",
				VivoChatCategory:   "IM",
				VivoSystemCategory: "ACCOUNT",
			},
		},
	}

	// chat 场景 -> IM
	payload := factory.Build(MessageParams{Scene: SceneChat, Title: "T", Body: "B"}, cfg)
	if payload.TPC == nil || payload.TPC.VIVO == nil {
		t.Fatal("VIVO TPC should exist for chat scene")
	}
	if payload.TPC.VIVO["category"] != "IM" {
		t.Errorf("VIVO chat category = %v, want IM", payload.TPC.VIVO["category"])
	}

	// system 场景 -> ACCOUNT
	payload = factory.Build(MessageParams{Scene: SceneSystem, Title: "T", Body: "B"}, cfg)
	if payload.TPC.VIVO["category"] != "ACCOUNT" {
		t.Errorf("VIVO system category = %v, want ACCOUNT", payload.TPC.VIVO["category"])
	}
}

// TestVivoCategoryOptional 验证：未配置 category 时不下发，仅 channel_id
func TestVivoCategoryOptional(t *testing.T) {
	factory := NewFactory()
	cfg := config.ChannelConfig{
		Vendor: config.VendorChannel{
			VIVO: config.VendorChannelID{Chat: "vivo_chat"},
		},
	}
	payload := factory.Build(MessageParams{Scene: SceneChat, Title: "T", Body: "B"}, cfg)
	if payload.TPC == nil || payload.TPC.VIVO == nil {
		t.Fatal("VIVO TPC should exist (has channel_id)")
	}
	if _, ok := payload.TPC.VIVO["category"]; ok {
		t.Error("VIVO should NOT contain category when not configured")
	}
	if payload.TPC.VIVO["channel_id"] != "vivo_chat" {
		t.Errorf("VIVO channel_id = %v, want vivo_chat", payload.TPC.VIVO["channel_id"])
	}
}

func mustJSON(v any) string {
	b, _ := json.Marshal(v)
	return string(b)
}