Split strategy config by strategy type

store/strategy_schema_test.go

@@ -0,0 +1,78 @@
+package store
+
+import (
+	"encoding/json"
+	"testing"
+)
+
+func TestStrategyConfigMarshalSeparatesGridAndAIConfig(t *testing.T) {
+	cfg := GetDefaultStrategyConfig("zh")
+	cfg.StrategyType = "grid_trading"
+	cfg.GridConfig = &GridStrategyConfig{
+		Symbol:          "BTCUSDT",
+		GridCount:       20,
+		TotalInvestment: 200,
+		Leverage:        2,
+		UseATRBounds:    true,
+		ATRMultiplier:   2,
+		Distribution:    "uniform",
+	}
+
+	raw, err := json.Marshal(cfg)
+	if err != nil {
+		t.Fatalf("marshal grid config: %v", err)
+	}
+
+	var asMap map[string]any
+	if err := json.Unmarshal(raw, &asMap); err != nil {
+		t.Fatalf("unmarshal grid config map: %v", err)
+	}
+	if asMap["strategy_type"] != "grid_trading" {
+		t.Fatalf("expected grid strategy_type, got %v", asMap["strategy_type"])
+	}
+	if _, ok := asMap["grid_config"]; !ok {
+		t.Fatalf("expected grid_config in grid strategy JSON: %s", string(raw))
+	}
+	for _, key := range []string{"ai_config", "coin_source", "indicators", "risk_control", "prompt_sections", "custom_prompt"} {
+		if _, ok := asMap[key]; ok {
+			t.Fatalf("did not expect %s in grid strategy JSON: %s", key, string(raw))
+		}
+	}
+}
+
+func TestStrategyConfigUnmarshalLegacyFlatAIConfig(t *testing.T) {
+	raw := []byte(`{
+		"strategy_type":"ai_trading",
+		"coin_source":{"source_type":"static","static_coins":["ETHUSDT"]},
+		"indicators":{"klines":{"primary_timeframe":"15m"}},
+		"risk_control":{"max_positions":2,"min_confidence":80},
+		"prompt_sections":{"entry_standards":"trend only"},
+		"custom_prompt":"prefer ETH"
+	}`)
+
+	var cfg StrategyConfig
+	if err := json.Unmarshal(raw, &cfg); err != nil {
+		t.Fatalf("unmarshal legacy flat config: %v", err)
+	}
+	if cfg.CoinSource.SourceType != "static" || len(cfg.CoinSource.StaticCoins) != 1 || cfg.CoinSource.StaticCoins[0] != "ETHUSDT" {
+		t.Fatalf("legacy coin source was not normalized: %+v", cfg.CoinSource)
+	}
+	if cfg.Indicators.Klines.PrimaryTimeframe != "15m" {
+		t.Fatalf("legacy indicators were not normalized: %+v", cfg.Indicators.Klines)
+	}
+
+	normalized, err := json.Marshal(cfg)
+	if err != nil {
+		t.Fatalf("marshal normalized config: %v", err)
+	}
+	var asMap map[string]any
+	if err := json.Unmarshal(normalized, &asMap); err != nil {
+		t.Fatalf("unmarshal normalized map: %v", err)
+	}
+	if _, ok := asMap["ai_config"]; !ok {
+		t.Fatalf("expected ai_config after normalizing legacy config: %s", string(normalized))
+	}
+	if _, ok := asMap["coin_source"]; ok {
+		t.Fatalf("did not expect legacy coin_source at top level: %s", string(normalized))
+	}
+}