package backtest import ( "encoding/csv" "os" "testing" "time" "github.com/stretchr/testify/assert" "github.com/c9s/bbgo/pkg/fixedpoint" "github.com/c9s/bbgo/pkg/types" ) func TestKLineDumper(t *testing.T) { tempDir := os.TempDir() _ = os.Mkdir(tempDir, 0755) dumper := NewKLineDumper(tempDir) t1 := time.Now() err := dumper.Record(types.KLine{ Exchange: types.ExchangeBinance, Symbol: "BTCUSDT", StartTime: types.Time(t1), EndTime: types.Time(t1.Add(time.Minute)), Interval: types.Interval1m, Open: fixedpoint.NewFromFloat(1000.0), High: fixedpoint.NewFromFloat(2000.0), Low: fixedpoint.NewFromFloat(3000.0), Close: fixedpoint.NewFromFloat(4000.0), Volume: fixedpoint.NewFromFloat(5000.0), QuoteVolume: fixedpoint.NewFromFloat(6000.0), NumberOfTrades: 10, Closed: true, }) assert.NoError(t, err) err = dumper.Close() assert.NoError(t, err) filenames := dumper.Filenames() assert.NotEmpty(t, filenames) for _, filename := range filenames { f, err := os.Open(filename) if assert.NoError(t, err) { reader := csv.NewReader(f) records, err2 := reader.Read() if assert.NoError(t, err2) { assert.NotEmptyf(t, records, "%v", records) } } } }