104 lines
2.3 KiB
Go
104 lines
2.3 KiB
Go
package client
|
|
|
|
import (
|
|
"context"
|
|
"crypto/ecdsa"
|
|
"crypto/elliptic"
|
|
"crypto/rand"
|
|
"fmt"
|
|
"testing"
|
|
|
|
apistatus "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/client/status"
|
|
"git.frostfs.info/TrueCloudLab/frostfs-sdk-go/netmap"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
/*
|
|
File contains common functionality used for client package testing.
|
|
*/
|
|
|
|
var key, _ = ecdsa.GenerateKey(elliptic.P256(), rand.Reader)
|
|
|
|
var statusErr apistatus.ServerInternal
|
|
|
|
func init() {
|
|
statusErr.SetMessage("test status error")
|
|
}
|
|
|
|
func assertStatusErr(tb testing.TB, res interface{ Status() apistatus.Status }) {
|
|
require.IsType(tb, &statusErr, res.Status())
|
|
require.Equal(tb, statusErr.Message(), res.Status().(*apistatus.ServerInternal).Message())
|
|
}
|
|
|
|
func newClient(server frostFSAPIServer) *Client {
|
|
prm := PrmInit{
|
|
Key: *key,
|
|
}
|
|
|
|
var c Client
|
|
c.Init(prm)
|
|
c.setFrostFSAPIServer(server)
|
|
|
|
return &c
|
|
}
|
|
|
|
func TestClient_DialContext(t *testing.T) {
|
|
var c Client
|
|
|
|
// try to connect to any host
|
|
prm := PrmDial{
|
|
Endpoint: "localhost:8080",
|
|
}
|
|
|
|
assert := func(ctx context.Context, errExpected error) {
|
|
// expect particular context error according to Dial docs
|
|
require.ErrorIs(t, c.Dial(ctx, prm), errExpected)
|
|
}
|
|
|
|
// create pre-abandoned context
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
cancel()
|
|
|
|
assert(ctx, context.Canceled)
|
|
|
|
// create "pre-deadlined" context
|
|
ctx, cancel = context.WithTimeout(context.Background(), 0)
|
|
defer cancel()
|
|
|
|
assert(ctx, context.DeadlineExceeded)
|
|
}
|
|
|
|
func TestClient_NetMapDialContext(t *testing.T) {
|
|
var prmInit PrmInit
|
|
var c Client
|
|
publicKey := "foo"
|
|
endpoint := "localhost:8080"
|
|
|
|
node := netmap.NodeInfo{}
|
|
node.SetPublicKey([]byte(publicKey))
|
|
node.SetExternalAddresses(endpoint)
|
|
netMap := &netmap.NetMap{}
|
|
netMap.SetNodes([]netmap.NodeInfo{node})
|
|
|
|
prmInit.NetMap = netMap
|
|
c.Init(prmInit)
|
|
|
|
prmNetMapDial := PrmNetMapDial{}
|
|
|
|
assert := func(ctx context.Context, errExpected error) {
|
|
// expect particular context error according to Dial docs
|
|
require.ErrorIs(t, c.NetMapDial(ctx, fmt.Sprintf("frostfs://%s", publicKey), prmNetMapDial), errExpected)
|
|
}
|
|
|
|
// create pre-abandoned context
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
cancel()
|
|
|
|
assert(ctx, context.Canceled)
|
|
|
|
// create "pre-deadlined" context
|
|
ctx, cancel = context.WithTimeout(context.Background(), 0)
|
|
defer cancel()
|
|
|
|
assert(ctx, context.DeadlineExceeded)
|
|
}
|