package tradesignal

import (
	"errors"
	"strings"
	"testing"

	"aiaa-notification-service/internal/config"
)

func TestConvertOpen(t *testing.T) {
	lev := 100
	c := NewConverter(map[string]config.StrategyOverride{
		"BLONG": {QuantityMultipliers: config.QuantityMultipliers{Open: 100}, Leverage: &lev},
	})
	event, data, err := c.Convert([]byte(`{
		"signalId":"s1","strategyCode":"BLONG","symbol":"BTCUSDT",
		"side":"LONG","action":"OPEN","quantity":0.01,"price":64000,
		"leverage":10,"period":"1h","eventTime":"2026-06-23T01:30:00Z"
	}`))
	if err != nil {
		t.Fatal(err)
	}
	if event != "trade.open" {
		t.Fatalf("event=%q", event)
	}
	formatted, _ := data["formatted"].(string)
	if !strings.Contains(formatted, "周期: 1h") || !strings.Contains(formatted, "开仓数量: 1.00") {
		t.Fatalf("formatted=\n%s", formatted)
	}
	if data["period"] != "1h" || data["strategyCode"] != "BLONG" {
		t.Fatalf("data=%v", data)
	}
	if data["leverage"] != float64(100) && data["leverage"] != 100 {
		t.Fatalf("leverage=%v", data["leverage"])
	}
}

// TestOverrideForLowercaseKey verifies overrides still match when viper
// lower-cases the strategy_overrides map keys during config load.
func TestOverrideForLowercaseKey(t *testing.T) {
	lev := 100
	c := NewConverter(map[string]config.StrategyOverride{
		"blong": {QuantityMultipliers: config.QuantityMultipliers{Add: 100}, Leverage: &lev},
	})
	o := c.overrideFor("BLONG")
	if o == nil {
		t.Fatal("overrideFor(BLONG)=nil, lowercase config key should match")
	}
	if got := o.QuantityMultiplierFor("ADD"); got != 100 {
		t.Fatalf("add multiplier=%v want 100", got)
	}
	if o.Leverage == nil || *o.Leverage != 100 {
		t.Fatalf("leverage=%v want 100", o.Leverage)
	}
}

func TestConvertKeepsExtraJSONFields(t *testing.T) {
	_, data, err := NewConverter(nil).Convert([]byte(`{
		"action":"OPEN","symbol":"BTCUSDT","price":63014.61,
		"totalAvgPx":63014.61,"investmentAmount":100095.24
	}`))
	if err != nil {
		t.Fatal(err)
	}
	if data["totalAvgPx"] != 63014.61 {
		t.Fatalf("totalAvgPx=%v", data["totalAvgPx"])
	}
}

func TestConvertInvalidJSON(t *testing.T) {
	_, _, err := NewConverter(nil).Convert([]byte(`{`))
	if !errors.Is(err, ErrInvalidSignal) {
		t.Fatalf("err=%v", err)
	}
}

func TestConvertMissingAction(t *testing.T) {
	_, _, err := NewConverter(nil).Convert([]byte(`{"symbol":"BTCUSDT"}`))
	if !errors.Is(err, ErrInvalidSignal) {
		t.Fatalf("err=%v", err)
	}
}

func TestConvertRawMessageWithoutAction(t *testing.T) {
	event, data, err := NewConverter(nil).Convert([]byte(`{
		"strategyCode":"PUTEJJ",
		"rawMessage":"4.7.15~2026.8.15 本周期 10万本金 期末109640，盈利9.6%"
	}`))
	if err != nil {
		t.Fatal(err)
	}
	if event != "trade.message" {
		t.Fatalf("event=%q", event)
	}
	if data["rawMessage"] != "4.7.15~2026.8.15 本周期 10万本金 期末109640，盈利9.6%" {
		t.Fatalf("data=%v", data)
	}
}