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) }) t.Run(psName+"/pointer-field", func(t *testing.T) { var a = &TestStruct{ Position: types.NewPosition("BTCUSDT", "BTC", "USDT"), } a.Position.Base = fixedpoint.NewFromFloat(10.0) a.Position.AverageCost = fixedpoint.NewFromFloat(3343.0) err := storePersistenceFields(a, "pointer-field-test", ps) assert.NoError(t, err) b := &TestStruct{} err = loadPersistenceFields(b, "pointer-field-test", ps) assert.NoError(t, err) assert.Equal(t, "10", a.Position.Base.String()) assert.Equal(t, "3343", a.Position.AverageCost.String()) }) } } 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) }) } }