Files
gocryptotrader/exchanges/margin/margin_test.go
Ryan O'Hara-Reid e99adca86f encoding/json: Add custom JSON package with build tag support for Sonic (#1623)
* tag optional sonic and allow full library conversion

* Add workflow and disallow arm and darwin usage

* Add basic hotswap benchmark

* linter: fix

* use bash

* linter: fix?

* Fix whoopsie, add to make file, also add mention in features list.

* test enforcement

* actually read documentation see if this works

* linter: fix

* linter: fix

* sonic: bump tagged version

* encoding/json: drop build tag arch and os filters

* encoding/json: consolidate tests

* encoding/json: log build tag usage

* rm superfluous builds

* glorious/nits: add template change and regen docs

* glorious/nits: update commentary on nolint directive

* glorious/nits: rm init func and log results in main.go

* Test to actually pull flag in

* linter: fix

* thrasher: nits

* gk: nits 4 goflags goooooooooo!

* gk: nits rn

* make sonic default json implementation

* screen 386

* linter: fix

* Add commentary

* glorious: nits Makefile not working

* gk: nits

* gk: nits whoops

* whoopsirino

* mention 32bit systems won't be sonic

* gk: super-duper nit of extremes

---------

Co-authored-by: Ryan O'Hara-Reid <ryan.oharareid@thrasher.io>
2025-02-20 16:05:55 +11:00

101 lines
2.9 KiB
Go

package margin
import (
"strings"
"testing"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
"github.com/thrasher-corp/gocryptotrader/encoding/json"
)
func TestValid(t *testing.T) {
t.Parallel()
require.True(t, Isolated.Valid())
require.True(t, Multi.Valid())
require.True(t, NoMargin.Valid())
require.True(t, SpotIsolated.Valid())
require.False(t, Unset.Valid())
require.False(t, Unknown.Valid())
require.False(t, Type(137).Valid())
}
func TestUnmarshalJSON(t *testing.T) {
t.Parallel()
for name, tc := range map[string]struct {
in string
want Type
err error
}{
"isolated": {`{"margin":"isolated"}`, Isolated, nil},
"cross": {`{"margin":"cross"}`, Multi, nil},
"cash": {`{"margin":"cash"}`, NoMargin, nil},
"spotIsolated": {`{"margin":"spot_isolated"}`, SpotIsolated, nil},
"invalid": {`{"margin":"hello moto"}`, Unknown, ErrInvalidMarginType},
"unset": {`{"margin":""}`, Unset, nil},
} {
t.Run(name, func(t *testing.T) {
t.Parallel()
var alien struct {
M Type `json:"margin"`
}
err := json.Unmarshal([]byte(tc.in), &alien)
assert.ErrorIs(t, err, tc.err)
assert.Equal(t, tc.want, alien.M)
})
}
}
func TestString(t *testing.T) {
t.Parallel()
assert.Equal(t, unknownStr, Unknown.String())
assert.Equal(t, isolatedStr, Isolated.String())
assert.Equal(t, multiStr, Multi.String())
assert.Equal(t, unsetStr, Unset.String())
assert.Equal(t, spotIsolatedStr, SpotIsolated.String())
assert.Equal(t, cashStr, NoMargin.String())
assert.Equal(t, "", Type(30).String())
}
func TestUpper(t *testing.T) {
t.Parallel()
assert.Equal(t, strings.ToUpper(unknownStr), Unknown.Upper())
assert.Equal(t, strings.ToUpper(isolatedStr), Isolated.Upper())
assert.Equal(t, strings.ToUpper(multiStr), Multi.Upper())
assert.Equal(t, strings.ToUpper(spotIsolatedStr), SpotIsolated.Upper())
assert.Equal(t, strings.ToUpper(cashStr), NoMargin.Upper())
assert.Equal(t, strings.ToUpper(unsetStr), Unset.Upper())
}
func TestIsValidString(t *testing.T) {
t.Parallel()
assert.False(t, IsValidString("lol"))
assert.True(t, IsValidString("spot_isolated"))
assert.True(t, IsValidString("cash"))
assert.True(t, IsValidString("isolated"))
assert.True(t, IsValidString("cross"))
assert.True(t, IsValidString("multi"))
assert.True(t, IsValidString(""))
assert.False(t, IsValidString("unknown"))
}
func TestStringToMarginType(t *testing.T) {
t.Parallel()
for label, v := range map[string]struct {
MarginType Type
Error error
}{
"lol": {Unknown, ErrInvalidMarginType},
"": {Unset, nil},
"cross": {Multi, nil},
"multi": {Multi, nil},
"isolated": {Isolated, nil},
"cash": {NoMargin, nil},
"spot_isolated": {SpotIsolated, nil},
} {
resp, err := StringToMarginType(label)
assert.ErrorIs(t, err, v.Error)
assert.Equal(t, v.MarginType.String(), resp.String())
}
}