frostfs-sdk-go/client/object_search_test.go

package client

import (
	"errors"
	"io"
	"testing"

	"github.com/nspcc-dev/neo-go/pkg/crypto/keys"
	"github.com/nspcc-dev/neofs-api-go/v2/object"
	"github.com/nspcc-dev/neofs-api-go/v2/refs"
	signatureV2 "github.com/nspcc-dev/neofs-api-go/v2/signature"
	oid "github.com/nspcc-dev/neofs-sdk-go/object/id"
	oidtest "github.com/nspcc-dev/neofs-sdk-go/object/id/test"
	"github.com/stretchr/testify/require"
)

func TestObjectSearch(t *testing.T) {
	ids := make([]oid.ID, 20)
	for i := range ids {
		ids[i] = *oidtest.ID()
	}

	resp, setID := testListReaderResponse(t)

	buf := make([]oid.ID, 2)
	checkRead := func(t *testing.T, expected []oid.ID) {
		n, ok := resp.Read(buf)
		require.True(t, ok == (len(expected) == len(buf)), "expected no error")
		require.Equal(t, len(expected), n, "expected %d items to be read", len(expected))
		require.Equal(t, expected, buf[:len(expected)])
	}

	// nil panic
	require.Panics(t, func() { resp.Read(nil) })

	// both ID fetched
	setID(ids[:3])
	checkRead(t, ids[:2])

	// one ID cached, second fetched
	setID(ids[3:6])
	checkRead(t, ids[2:4])

	// both ID cached
	resp.ctxCall.resp = nil
	checkRead(t, ids[4:6])

	// both ID fetched in 2 requests, with empty one in the middle
	var n int
	resp.ctxCall.rResp = func() error {
		switch n {
		case 0:
			setID(ids[6:7])
		case 1:
			setID(nil)
		case 2:
			setID(ids[7:8])
		default:
			t.FailNow()
		}
		n++
		return nil
	}
	checkRead(t, ids[6:8])

	// read from tail multiple times
	resp.ctxCall.rResp = nil
	setID(ids[8:11])
	buf = buf[:1]
	checkRead(t, ids[8:9])
	checkRead(t, ids[9:10])
	checkRead(t, ids[10:11])

	// handle EOF
	buf = buf[:2]
	n = 0
	resp.ctxCall.rResp = func() error {
		if n > 0 {
			return io.EOF
		}
		n++
		setID(ids[11:12])
		return nil
	}
	checkRead(t, ids[11:12])
}

func TestObjectIterate(t *testing.T) {
	ids := make([]oid.ID, 3)
	for i := range ids {
		ids[i] = *oidtest.ID()
	}

	t.Run("iterate all sequence", func(t *testing.T) {
		resp, setID := testListReaderResponse(t)

		// Iterate over all sequence
		var n int
		resp.ctxCall.rResp = func() error {
			switch n {
			case 0:
				setID(ids[0:2])
			case 1:
				setID(nil)
			case 2:
				setID(ids[2:3])
			default:
				return io.EOF
			}
			n++
			return nil
		}

		var actual []oid.ID
		require.NoError(t, resp.Iterate(func(id oid.ID) bool {
			actual = append(actual, id)
			return false
		}))
		require.Equal(t, ids[:3], actual)
	})
	t.Run("stop by return value", func(t *testing.T) {
		resp, setID := testListReaderResponse(t)

		var actual []oid.ID
		setID(ids)
		require.NoError(t, resp.Iterate(func(id oid.ID) bool {
			actual = append(actual, id)
			return len(actual) == 2
		}))
		require.Equal(t, ids[:2], actual)
	})
	t.Run("stop after error", func(t *testing.T) {
		resp, setID := testListReaderResponse(t)
		expectedErr := errors.New("test error")

		var actual []oid.ID
		var n int
		resp.ctxCall.rResp = func() error {
			switch n {
			case 0:
				setID(ids[:2])
			default:
				return expectedErr
			}
			n++
			return nil
		}

		err := resp.Iterate(func(id oid.ID) bool {
			actual = append(actual, id)
			return false
		})
		require.True(t, errors.Is(err, expectedErr), "got: %v", err)
		require.Equal(t, ids[:2], actual)
	})
}

func testListReaderResponse(t *testing.T) (*ObjectListReader, func(id []oid.ID) *object.SearchResponse) {
	p, err := keys.NewPrivateKey()
	require.NoError(t, err)

	obj := &ObjectListReader{
		cancelCtxStream: func() {},
		ctxCall: contextCall{
			closer:    func() error { return nil },
			result:    func(v2 responseV2) {},
			statusRes: new(ResObjectSearch),
		},
		reqWritten: true,
		bodyResp:   object.SearchResponseBody{},
		tail:       nil,
	}

	return obj, func(id []oid.ID) *object.SearchResponse {
		resp := new(object.SearchResponse)
		resp.SetBody(new(object.SearchResponseBody))

		v2id := make([]refs.ObjectID, len(id))
		for i := range id {
			v2id[i] = *id[i].ToV2()
		}
		resp.GetBody().SetIDList(v2id)
		err := signatureV2.SignServiceMessage(&p.PrivateKey, resp)
		if err != nil {
			t.Fatalf("error: %v", err)
		}
		obj.ctxCall.resp = resp
		obj.bodyResp = *resp.GetBody()
		return resp
	}
}