frostfs-sdk-go/client/session_test.go

package client

import (
	"context"
	"testing"

	v2netmap "github.com/nspcc-dev/neofs-api-go/v2/netmap"
	"github.com/nspcc-dev/neofs-api-go/v2/rpc/client"
	"github.com/nspcc-dev/neofs-api-go/v2/session"
	neofscrypto "github.com/nspcc-dev/neofs-sdk-go/crypto"
	"github.com/nspcc-dev/neofs-sdk-go/crypto/test"
	"github.com/stretchr/testify/require"
)

type sessionAPIServer struct {
	signer  neofscrypto.Signer
	setBody func(body *session.CreateResponseBody)
}

func (m sessionAPIServer) netMapSnapshot(context.Context, v2netmap.SnapshotRequest) (*v2netmap.SnapshotResponse, error) {
	return nil, nil
}

func (m sessionAPIServer) createSession(*client.Client, *session.CreateRequest, ...client.CallOption) (*session.CreateResponse, error) {
	var body session.CreateResponseBody
	m.setBody(&body)

	var resp session.CreateResponse
	resp.SetBody(&body)

	if err := signServiceMessage(m.signer, &resp); err != nil {
		return nil, err
	}

	return &resp, nil
}

func TestClient_SessionCreate(t *testing.T) {
	ctx := context.Background()
	signer := test.RandomSignerRFC6979(t)

	var prmInit PrmInit
	prmInit.SetDefaultSigner(signer)

	var c Client
	c.Init(prmInit)

	var prmSessionCreate PrmSessionCreate
	prmSessionCreate.UseSigner(signer)
	prmSessionCreate.SetExp(1)

	t.Run("missing session id", func(t *testing.T) {
		c.setNeoFSAPIServer(&sessionAPIServer{signer: signer, setBody: func(body *session.CreateResponseBody) {
			body.SetSessionKey([]byte{1})
		}})

		result, err := c.SessionCreate(ctx, prmSessionCreate)
		require.Nil(t, result)
		require.ErrorIs(t, err, ErrMissingResponseField)
		require.Equal(t, "missing session id field in the response", err.Error())
	})

	t.Run("missing session key", func(t *testing.T) {
		c.setNeoFSAPIServer(&sessionAPIServer{signer: signer, setBody: func(body *session.CreateResponseBody) {
			body.SetID([]byte{1})
		}})

		result, err := c.SessionCreate(ctx, prmSessionCreate)
		require.Nil(t, result)
		require.ErrorIs(t, err, ErrMissingResponseField)
		require.Equal(t, "missing session key field in the response", err.Error())
	})
}