ryan
cd3db8d453
Motivation: 部分 meme 币等标的价格极小（如 0.00000059），原有格式化统一保留两位小数会将其显示为 0.00，导致推送消息中的价格信息失真、误导用户；同时 viper 加载配置时会将嵌套 map 的 key 统一转为小写，导致按大写策略编码配置的策略覆盖项无法命中，仓位倍数、杠杆等覆盖参数失效。 Changes: * 新增价格展示格式化逻辑：绝对值不小于 1 的数值保留两位小数，小于 1 的数值采用最短精确表示，避免极小价格被截断为 0.00 * 策略覆盖查找改为大小写不敏感匹配，兼容 viper 将配置 key 小写化的行为，确保策略编码以任意大小写配置均可生效 * 补充极小价格展示、配置加载解析及策略覆盖匹配的单元测试
100 lines
2.8 KiB
Go
100 lines
2.8 KiB
Go
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)
|
|
}
|
|
}
|