forked from TrueCloudLab/neoneo-go
90 lines
2.7 KiB
Go
90 lines
2.7 KiB
Go
package options
|
|
|
|
import (
|
|
"flag"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/nspcc-dev/neo-go/pkg/config/netmode"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
"github.com/urfave/cli"
|
|
)
|
|
|
|
func TestGetNetwork(t *testing.T) {
|
|
t.Run("privnet", func(t *testing.T) {
|
|
set := flag.NewFlagSet("flagSet", flag.ExitOnError)
|
|
ctx := cli.NewContext(cli.NewApp(), set, nil)
|
|
require.Equal(t, netmode.PrivNet, GetNetwork(ctx))
|
|
})
|
|
|
|
t.Run("testnet", func(t *testing.T) {
|
|
set := flag.NewFlagSet("flagSet", flag.ExitOnError)
|
|
set.Bool("testnet", true, "")
|
|
ctx := cli.NewContext(cli.NewApp(), set, nil)
|
|
require.Equal(t, netmode.TestNet, GetNetwork(ctx))
|
|
})
|
|
|
|
t.Run("mainnet", func(t *testing.T) {
|
|
set := flag.NewFlagSet("flagSet", flag.ExitOnError)
|
|
set.Bool("mainnet", true, "")
|
|
ctx := cli.NewContext(cli.NewApp(), set, nil)
|
|
require.Equal(t, netmode.MainNet, GetNetwork(ctx))
|
|
})
|
|
}
|
|
|
|
func TestGetTimeoutContext(t *testing.T) {
|
|
t.Run("default", func(t *testing.T) {
|
|
start := time.Now()
|
|
set := flag.NewFlagSet("flagSet", flag.ExitOnError)
|
|
ctx := cli.NewContext(cli.NewApp(), set, nil)
|
|
actualCtx, _ := GetTimeoutContext(ctx)
|
|
end := time.Now()
|
|
dl, _ := actualCtx.Deadline()
|
|
require.True(t, start.Before(dl) && dl.Before(end.Add(DefaultTimeout)))
|
|
})
|
|
|
|
t.Run("set", func(t *testing.T) {
|
|
start := time.Now()
|
|
set := flag.NewFlagSet("flagSet", flag.ExitOnError)
|
|
set.Duration("timeout", time.Duration(20), "")
|
|
ctx := cli.NewContext(cli.NewApp(), set, nil)
|
|
actualCtx, _ := GetTimeoutContext(ctx)
|
|
end := time.Now()
|
|
dl, _ := actualCtx.Deadline()
|
|
require.True(t, start.Before(dl) && dl.Before(end.Add(time.Nanosecond*20)))
|
|
})
|
|
}
|
|
|
|
func TestGetRPCClient(t *testing.T) {
|
|
// need test server for proper client.Init() handling
|
|
srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) {
|
|
w.Header().Set("Content-Type", "application/json; charset=utf-8")
|
|
response := `{"id":1,"jsonrpc":"2.0","result":{"magic":42,"tcpport":20332,"wsport":20342,"nonce":2153672787,"useragent":"/NEO-GO:0.73.1-pre-273-ge381358/"}}`
|
|
|
|
_, err := w.Write([]byte(response))
|
|
if err != nil {
|
|
t.Fatalf("Error writing response: %s", err.Error())
|
|
}
|
|
}))
|
|
defer srv.Close()
|
|
t.Run("no endpoint", func(t *testing.T) {
|
|
set := flag.NewFlagSet("flagSet", flag.ExitOnError)
|
|
ctx := cli.NewContext(cli.NewApp(), set, nil)
|
|
gctx, _ := GetTimeoutContext(ctx)
|
|
_, ec := GetRPCClient(gctx, ctx)
|
|
require.Equal(t, 1, ec.ExitCode())
|
|
})
|
|
|
|
t.Run("success", func(t *testing.T) {
|
|
set := flag.NewFlagSet("flagSet", flag.ExitOnError)
|
|
set.String(RPCEndpointFlag, srv.URL, "")
|
|
ctx := cli.NewContext(cli.NewApp(), set, nil)
|
|
gctx, _ := GetTimeoutContext(ctx)
|
|
_, ec := GetRPCClient(gctx, ctx)
|
|
require.Nil(t, ec)
|
|
})
|
|
|
|
}
|