mirror of
https://github.com/c9s/bbgo.git
synced 2024-11-26 08:45:16 +00:00
120 lines
2.8 KiB
Go
120 lines
2.8 KiB
Go
package bbgo
|
|
|
|
import (
|
|
"os"
|
|
"reflect"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"github.com/c9s/bbgo/pkg/fixedpoint"
|
|
"github.com/c9s/bbgo/pkg/service"
|
|
"github.com/c9s/bbgo/pkg/types"
|
|
)
|
|
|
|
type TestStruct struct {
|
|
*Environment
|
|
*Graceful
|
|
|
|
Position *types.Position `persistence:"position"`
|
|
Integer int64 `persistence:"integer"`
|
|
Integer2 int64 `persistence:"integer2"`
|
|
Float int64 `persistence:"float"`
|
|
String string `persistence:"string"`
|
|
}
|
|
|
|
func (t *TestStruct) InstanceID() string {
|
|
return "test-struct"
|
|
}
|
|
|
|
func preparePersistentServices() []service.PersistenceService {
|
|
mem := service.NewMemoryService()
|
|
jsonDir := &service.JsonPersistenceService{Directory: "testoutput/persistence"}
|
|
pss := []service.PersistenceService{
|
|
mem,
|
|
jsonDir,
|
|
}
|
|
|
|
if _, ok := os.LookupEnv("TEST_REDIS"); ok {
|
|
redisP := service.NewRedisPersistenceService(&service.RedisPersistenceConfig{
|
|
Host: "localhost",
|
|
Port: "6379",
|
|
DB: 0,
|
|
})
|
|
pss = append(pss, redisP)
|
|
}
|
|
|
|
return pss
|
|
}
|
|
|
|
func Test_callID(t *testing.T) {
|
|
id := callID(&TestStruct{})
|
|
assert.NotEmpty(t, id)
|
|
}
|
|
|
|
func Test_loadPersistenceFields(t *testing.T) {
|
|
var pss = preparePersistentServices()
|
|
|
|
for _, ps := range pss {
|
|
psName := reflect.TypeOf(ps).Elem().String()
|
|
t.Run(psName+"/empty", func(t *testing.T) {
|
|
b := &TestStruct{}
|
|
err := loadPersistenceFields(b, "test-empty", ps)
|
|
assert.NoError(t, err)
|
|
})
|
|
|
|
t.Run(psName+"/nil", func(t *testing.T) {
|
|
var b *TestStruct = nil
|
|
err := loadPersistenceFields(b, "test-nil", ps)
|
|
assert.Equal(t, errCanNotIterateNilPointer, err)
|
|
})
|
|
}
|
|
}
|
|
|
|
func Test_storePersistenceFields(t *testing.T) {
|
|
var pss = preparePersistentServices()
|
|
|
|
var a = &TestStruct{
|
|
Integer: 1,
|
|
Integer2: 2,
|
|
Float: 3.0,
|
|
String: "foobar",
|
|
Position: types.NewPosition("BTCUSDT", "BTC", "USDT"),
|
|
}
|
|
|
|
a.Position.Base = fixedpoint.NewFromFloat(10.0)
|
|
a.Position.AverageCost = fixedpoint.NewFromFloat(3343.0)
|
|
|
|
for _, ps := range pss {
|
|
psName := reflect.TypeOf(ps).Elem().String()
|
|
t.Run("all/"+psName, func(t *testing.T) {
|
|
id := callID(a)
|
|
err := storePersistenceFields(a, id, ps)
|
|
assert.NoError(t, err)
|
|
|
|
var i int64
|
|
store := ps.NewStore("state", "test-struct", "integer")
|
|
err = store.Load(&i)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, int64(1), i)
|
|
|
|
var p *types.Position
|
|
store = ps.NewStore("state", "test-struct", "position")
|
|
err = store.Load(&p)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, fixedpoint.NewFromFloat(10.0), p.Base)
|
|
assert.Equal(t, fixedpoint.NewFromFloat(3343.0), p.AverageCost)
|
|
|
|
var b = &TestStruct{}
|
|
err = loadPersistenceFields(b, id, ps)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, a.Integer, b.Integer)
|
|
assert.Equal(t, a.Integer2, b.Integer2)
|
|
assert.Equal(t, a.Float, b.Float)
|
|
assert.Equal(t, a.String, b.String)
|
|
assert.Equal(t, a.Position, b.Position)
|
|
})
|
|
}
|
|
|
|
}
|