package ns import ( "errors" "fmt" "math/big" "math/rand" "strings" "testing" nns "github.com/nspcc-dev/neo-go/examples/nft-nd-nns" "github.com/nspcc-dev/neo-go/pkg/rpc/response/result" "github.com/nspcc-dev/neo-go/pkg/smartcontract" "github.com/nspcc-dev/neo-go/pkg/util" "github.com/nspcc-dev/neo-go/pkg/vm" "github.com/nspcc-dev/neo-go/pkg/vm/stackitem" cidtest "github.com/nspcc-dev/neofs-sdk-go/container/id/test" "github.com/stretchr/testify/require" ) // testNeoClient represents test Neo client which checks invocation arguments // and returns predefined result. type testNeoClient struct { t *testing.T expectedContract util.Uint160 res result.Invoke err error } func (x *testNeoClient) invoke(contract util.Uint160, method string, prm []smartcontract.Parameter) (*result.Invoke, error) { require.Equal(x.t, x.expectedContract, contract) require.Equal(x.t, "resolve", method) require.Len(x.t, prm, 2) require.Equal(x.t, smartcontract.StringType, prm[0].Type) require.Equal(x.t, smartcontract.IntegerType, prm[1].Type) require.EqualValues(x.t, nns.TXT, prm[1].Value) val, ok := prm[0].Value.(string) require.True(x.t, ok) require.True(x.t, strings.HasSuffix(val, ".container")) require.NotEmpty(x.t, strings.TrimSuffix(val, ".container")) return &x.res, x.err } // implements test stackitem.Item which is obviously incorrect: // it returns itself on Convert(stackitem.ArrayT), but returns integer from Value. type brokenArrayStackItem struct { stackitem.Item } func (x brokenArrayStackItem) Value() interface{} { return 1 } func (x brokenArrayStackItem) Convert(t stackitem.Type) (stackitem.Item, error) { if t != stackitem.ArrayT { panic(fmt.Sprintf("unexpected stack item type %s", t)) } return x, nil } func TestNNS_ResolveContainerName(t *testing.T) { const testContainerName = "some_container" var nnsContract util.Uint160 rand.Read(nnsContract[:]) testC := &testNeoClient{ t: t, expectedContract: nnsContract, } n := NNS{ nnsContract: nnsContract, neoClient: testC, } t.Run("invocation failure", func(t *testing.T) { testC.err = errors.New("invoke err") _, err := n.ResolveContainerName(testContainerName) require.Error(t, err) }) testC.err = nil t.Run("fault exception", func(t *testing.T) { _, err := n.ResolveContainerName(testContainerName) require.Error(t, err) }) testC.res.State = vm.HaltState.String() t.Run("empty stack", func(t *testing.T) { _, err := n.ResolveContainerName(testContainerName) require.Error(t, err) }) testC.res.Stack = make([]stackitem.Item, 1) t.Run("non-array last stack item", func(t *testing.T) { testC.res.Stack[0] = stackitem.NewBigInteger(big.NewInt(11)) _, err := n.ResolveContainerName(testContainerName) require.Error(t, err) }) t.Run("null array", func(t *testing.T) { testC.res.Stack[0] = stackitem.Null{} _, err := n.ResolveContainerName(testContainerName) require.Error(t, err) }) t.Run("array stack item with non-slice value", func(t *testing.T) { testC.res.Stack[0] = brokenArrayStackItem{} _, err := n.ResolveContainerName(testContainerName) require.Error(t, err) }) arr := make([]stackitem.Item, 2) testC.res.Stack[0] = stackitem.NewArray(arr) t.Run("non-bytes array element", func(t *testing.T) { arr[0] = stackitem.NewArray(nil) _, err := n.ResolveContainerName(testContainerName) require.Error(t, err) }) arr[0] = stackitem.NewByteArray([]byte("some byte array 1")) t.Run("non-container array elements", func(t *testing.T) { arr[1] = stackitem.NewByteArray([]byte("some byte array 2")) _, err := n.ResolveContainerName(testContainerName) require.Error(t, err) }) t.Run("with container array element", func(t *testing.T) { id := cidtest.ID() arr[1] = stackitem.NewByteArray([]byte(id.String())) res, err := n.ResolveContainerName(testContainerName) require.NoError(t, err) require.Equal(t, id, res) }) }