frostfs-node/pkg/network/transport/object/grpc/search_test.go
Stanislav Bogatyrev b7b5079934 Add Inner Ring code
2020-07-24 17:07:37 +03:00

265 lines
6.3 KiB
Go

package object
import (
"context"
"testing"
"github.com/nspcc-dev/neofs-api-go/object"
v1 "github.com/nspcc-dev/neofs-api-go/query"
"github.com/nspcc-dev/neofs-node/pkg/services/object_manager/transport"
"github.com/pkg/errors"
"github.com/stretchr/testify/require"
)
type (
// Entity for mocking interfaces.
// Implementation of any interface intercepts arguments via f (if not nil).
// If err is not nil, it returns as it is. Otherwise, casted to needed type res returns w/o error.
testSearchEntity struct {
// Set of interfaces which entity must implement, but some methods from those does not call.
object.Service_SearchServer
// Argument interceptor. Used for ascertain of correct parameter passage between components.
f func(...interface{})
// Mocked result of any interface.
res interface{}
// Mocked error of any interface.
err error
}
)
var (
_ requestHandler = (*testSearchEntity)(nil)
_ operationExecutor = (*testSearchEntity)(nil)
_ responsePreparer = (*testSearchEntity)(nil)
_ object.Service_SearchServer = (*testSearchEntity)(nil)
)
func (s *testSearchEntity) prepareResponse(_ context.Context, req serviceRequest, resp serviceResponse) error {
if s.f != nil {
s.f(req, resp)
}
return s.err
}
func (s *testSearchEntity) Send(r *object.SearchResponse) error {
if s.f != nil {
s.f(r)
}
return s.err
}
func (s *testSearchEntity) Context() context.Context { return context.TODO() }
func (s *testSearchEntity) executeOperation(_ context.Context, p transport.MetaInfo, h responseItemHandler) error {
if s.f != nil {
s.f(p, h)
}
return s.err
}
func (s *testSearchEntity) handleRequest(_ context.Context, p handleRequestParams) (interface{}, error) {
if s.f != nil {
s.f(p)
}
return s.res, s.err
}
func TestSearchVerify(t *testing.T) {
t.Run("KeyNoChildren", func(t *testing.T) {
var (
q = v1.Query{
Filters: []QueryFilter{
{
Type: v1.Filter_Exact,
Name: transport.KeyNoChildren,
},
},
}
obj = new(Object)
)
require.True(t, imposeQuery(q, obj))
obj.Headers = append(obj.Headers, Header{Value: &object.Header_Link{
Link: &object.Link{
Type: object.Link_Child,
},
}})
require.False(t, imposeQuery(q, obj))
})
}
func Test_coreObjAddrSet(t *testing.T) {
// create address accumulator
acc := newUniqueAddressAccumulator()
require.NotNil(t, acc)
// check type correctness
v, ok := acc.(*coreObjAddrSet)
require.True(t, ok)
// check fields initialization
require.NotNil(t, v.items)
require.NotNil(t, v.RWMutex)
t.Run("add/list", func(t *testing.T) {
// ascertain that initial list is empty
require.Empty(t, acc.list())
// add first set of addresses
addrList1 := testAddrList(t, 5)
acc.handleItem(addrList1)
// ascertain that list is equal to added list
require.Equal(t, addrList1, acc.list())
// add more addresses
addrList2 := testAddrList(t, 5)
acc.handleItem(addrList2)
twoLists := append(addrList1, addrList2...)
// ascertain that list is a concatenation of added lists
require.Equal(t, twoLists, acc.list())
// add second list again
acc.handleItem(addrList2)
// ascertain that list have not changed after adding existing elements
require.Equal(t, twoLists, acc.list())
})
}
func TestObjectService_Search(t *testing.T) {
req := &object.SearchRequest{
ContainerID: testObjectAddress(t).CID,
Query: testData(t, 10),
}
addrList := testAddrList(t, int(addrPerMsg)+5)
t.Run("request handler failure", func(t *testing.T) {
rhErr := errors.New("test error for request handler")
s := &objectService{
statusCalculator: newStatusCalculator(),
}
s.requestHandler = &testSearchEntity{
f: func(items ...interface{}) {
p := items[0].(handleRequestParams)
require.Equal(t, req, p.request)
require.Equal(t, s, p.executor)
},
err: rhErr,
}
require.EqualError(t, s.Search(req, new(testSearchEntity)), rhErr.Error())
})
t.Run("server error", func(t *testing.T) {
srvErr := errors.New("test error for search server")
resp := &object.SearchResponse{Addresses: addrList[:addrPerMsg]}
s := &objectService{
requestHandler: &testSearchEntity{
res: addrList,
},
respPreparer: &testSearchEntity{
f: func(items ...interface{}) {
require.Equal(t, req, items[0])
require.Equal(t, makeSearchResponse(addrList[:addrPerMsg]), items[1])
},
res: resp,
},
statusCalculator: newStatusCalculator(),
}
srv := &testSearchEntity{
f: func(items ...interface{}) {
require.Equal(t, resp, items[0])
},
err: srvErr, // force server to return srvErr
}
require.EqualError(t, s.Search(req, srv), srvErr.Error())
})
t.Run("correct result", func(t *testing.T) {
handler := &testSearchEntity{res: make([]Address, 0)}
off := 0
var resp *object.SearchResponse
s := &objectService{
requestHandler: handler,
respPreparer: &testSearchEntity{
f: func(items ...interface{}) {
require.Equal(t, req, items[0])
resp = items[1].(*object.SearchResponse)
sz := len(resp.GetAddresses())
require.Equal(t, makeSearchResponse(addrList[off:off+sz]), items[1])
off += sz
},
},
statusCalculator: newStatusCalculator(),
}
srv := &testSearchEntity{
f: func(items ...interface{}) {
require.Equal(t, resp, items[0])
},
}
require.NoError(t, s.Search(req, srv))
handler.res = addrList
require.NoError(t, s.Search(req, srv))
})
}
func Test_coreObjectSearcher(t *testing.T) {
ctx := context.TODO()
req := newRawSearchInfo()
req.setQuery(testData(t, 10))
t.Run("operation executor failure", func(t *testing.T) {
execErr := errors.New("test error for operation executor")
s := &coreObjectSearcher{
executor: &testSearchEntity{
f: func(items ...interface{}) {
require.Equal(t, req, items[0])
require.Equal(t, newUniqueAddressAccumulator(), items[1])
},
err: execErr,
},
}
res, err := s.searchObjects(ctx, req)
require.EqualError(t, err, execErr.Error())
require.Empty(t, res)
})
t.Run("correct result", func(t *testing.T) {
addrList := testAddrList(t, 5)
s := &coreObjectSearcher{
executor: &testSearchEntity{
f: func(items ...interface{}) {
items[1].(responseItemHandler).handleItem(addrList)
},
},
}
res, err := s.searchObjects(ctx, req)
require.NoError(t, err)
require.Equal(t, addrList, res)
})
}