mirror of
https://github.com/c9s/bbgo.git
synced 2024-11-22 14:55:16 +00:00
58 lines
1.6 KiB
Go
58 lines
1.6 KiB
Go
package dynamic
|
|
|
|
import (
|
|
"encoding/json"
|
|
"reflect"
|
|
"testing"
|
|
|
|
"github.com/c9s/bbgo/pkg/fixedpoint"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
type Strategy struct {
|
|
Field1 fixedpoint.Value `json:"field1" modifiable:"true"`
|
|
Field2 float64 `json:"field2"`
|
|
field3 float64 `json:"field3" modifiable:"true"`
|
|
Field4 *fixedpoint.Value `json:"field4" modifiable:"true"`
|
|
}
|
|
|
|
func TestGetModifiableFields(t *testing.T) {
|
|
s := Strategy{}
|
|
val := reflect.ValueOf(s)
|
|
GetModifiableFields(val, func(tagName, name string) {
|
|
assert.NotEqual(t, tagName, "field2")
|
|
assert.NotEqual(t, name, "Field2")
|
|
assert.NotEqual(t, tagName, "field3")
|
|
assert.NotEqual(t, name, "Field3")
|
|
|
|
})
|
|
}
|
|
|
|
func TestGetModifiableField(t *testing.T) {
|
|
// val must be get from pointer.Elem(), otherwise the fields will be unaddressable
|
|
s := &Strategy{Field1: fixedpoint.NewFromInt(1)}
|
|
val := reflect.ValueOf(s).Elem()
|
|
_, ok := GetModifiableField(val, "Field1")
|
|
assert.True(t, ok)
|
|
_, ok = GetModifiableField(val, "Field2")
|
|
assert.False(t, ok)
|
|
_, ok = GetModifiableField(val, "Field3")
|
|
assert.False(t, ok)
|
|
_, ok = GetModifiableField(val, "Random")
|
|
assert.False(t, ok)
|
|
field, ok := GetModifiableField(val, "Field1")
|
|
assert.True(t, ok)
|
|
x := reflect.New(field.Type())
|
|
xi := x.Interface()
|
|
assert.NoError(t, json.Unmarshal([]byte("\"3.1415%\""), &xi))
|
|
assert.True(t, field.CanAddr())
|
|
field.Set(x.Elem())
|
|
assert.Equal(t, s.Field1.String(), "0.031415")
|
|
field, _ = GetModifiableField(val, "Field4")
|
|
x = reflect.New(field.Type())
|
|
xi = x.Interface()
|
|
assert.NoError(t, json.Unmarshal([]byte("311"), &xi))
|
|
field.Set(x.Elem())
|
|
assert.Equal(t, s.Field4.String(), "311")
|
|
}
|