bbgo_origin/pkg/backtest/recorder_test.go

package backtest

import (
	"os"
	"testing"
	"time"

	"github.com/stretchr/testify/assert"

	"github.com/c9s/bbgo/pkg/fixedpoint"
	"github.com/c9s/bbgo/pkg/types"
)

type testStrategy struct {
	Symbol string

	Position *types.Position
}

func (s *testStrategy) ID() string         { return "my-test" }
func (s *testStrategy) InstanceID() string { return "my-test:" + s.Symbol }

func TestStateRecorder(t *testing.T) {
	tmpDir, _ := os.MkdirTemp(os.TempDir(), "bbgo")
	t.Logf("tmpDir: %s", tmpDir)

	st := &testStrategy{
		Symbol:   "BTCUSDT",
		Position: types.NewPosition("BTCUSDT", "BTC", "USDT"),
	}

	recorder := NewStateRecorder(tmpDir)
	err := recorder.Scan(st)
	assert.NoError(t, err)
	assert.Len(t, recorder.writers, 1)

	n, err := recorder.Snapshot()
	assert.NoError(t, err)
	assert.Equal(t, 1, n)

	st.Position.AddTrade(types.Trade{
		OrderID:       1,
		Exchange:      types.ExchangeBinance,
		Price:         fixedpoint.NewFromFloat(18000.0),
		Quantity:      fixedpoint.NewFromFloat(1.0),
		QuoteQuantity: fixedpoint.NewFromFloat(18000.0),
		Symbol:        "BTCUSDT",
		Side:          types.SideTypeBuy,
		IsBuyer:       true,
		IsMaker:       false,
		Time:          types.Time(time.Now()),
		Fee:           fixedpoint.NewFromFloat(0.00001),
		FeeCurrency:   "BNB",
		IsMargin:      false,
		IsFutures:     false,
		IsIsolated:    false,
	})

	n, err = recorder.Snapshot()
	assert.NoError(t, err)
	assert.Equal(t, 1, n)

	err = recorder.Close()
	assert.NoError(t, err)
}