bbgo_origin/pkg/backtest/dumper_test.go

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)
			}

		}
	}
}