73b5e6256e
* added builtin splice * fix splice * remove splice negative index
353 lines
9.9 KiB
Go
353 lines
9.9 KiB
Go
package tengo_test
|
|
|
|
import (
|
|
"errors"
|
|
"reflect"
|
|
"testing"
|
|
|
|
"github.com/d5/tengo/v2"
|
|
)
|
|
|
|
func Test_builtinDelete(t *testing.T) {
|
|
var builtinDelete func(args ...tengo.Object) (tengo.Object, error)
|
|
for _, f := range tengo.GetAllBuiltinFunctions() {
|
|
if f.Name == "delete" {
|
|
builtinDelete = f.Value
|
|
break
|
|
}
|
|
}
|
|
if builtinDelete == nil {
|
|
t.Fatal("builtin delete not found")
|
|
}
|
|
type args struct {
|
|
args []tengo.Object
|
|
}
|
|
tests := []struct {
|
|
name string
|
|
args args
|
|
want tengo.Object
|
|
wantErr bool
|
|
wantedErr error
|
|
target interface{}
|
|
}{
|
|
{name: "invalid-arg", args: args{[]tengo.Object{&tengo.String{},
|
|
&tengo.String{}}}, wantErr: true,
|
|
wantedErr: tengo.ErrInvalidArgumentType{
|
|
Name: "first",
|
|
Expected: "map",
|
|
Found: "string"},
|
|
},
|
|
{name: "no-args",
|
|
wantErr: true, wantedErr: tengo.ErrWrongNumArguments},
|
|
{name: "empty-args", args: args{[]tengo.Object{}}, wantErr: true,
|
|
wantedErr: tengo.ErrWrongNumArguments,
|
|
},
|
|
{name: "3-args", args: args{[]tengo.Object{
|
|
(*tengo.Map)(nil), (*tengo.String)(nil), (*tengo.String)(nil)}},
|
|
wantErr: true, wantedErr: tengo.ErrWrongNumArguments,
|
|
},
|
|
{name: "nil-map-empty-key",
|
|
args: args{[]tengo.Object{&tengo.Map{}, &tengo.String{}}},
|
|
want: tengo.UndefinedValue,
|
|
},
|
|
{name: "nil-map-nonstr-key",
|
|
args: args{[]tengo.Object{
|
|
&tengo.Map{}, &tengo.Int{}}}, wantErr: true,
|
|
wantedErr: tengo.ErrInvalidArgumentType{
|
|
Name: "second", Expected: "string", Found: "int"},
|
|
},
|
|
{name: "nil-map-no-key",
|
|
args: args{[]tengo.Object{&tengo.Map{}}}, wantErr: true,
|
|
wantedErr: tengo.ErrWrongNumArguments,
|
|
},
|
|
{name: "map-missing-key",
|
|
args: args{
|
|
[]tengo.Object{
|
|
&tengo.Map{Value: map[string]tengo.Object{
|
|
"key": &tengo.String{Value: "value"},
|
|
}},
|
|
&tengo.String{Value: "key1"}}},
|
|
want: tengo.UndefinedValue,
|
|
target: &tengo.Map{
|
|
Value: map[string]tengo.Object{
|
|
"key": &tengo.String{
|
|
Value: "value"}}},
|
|
},
|
|
{name: "map-emptied",
|
|
args: args{
|
|
[]tengo.Object{
|
|
&tengo.Map{Value: map[string]tengo.Object{
|
|
"key": &tengo.String{Value: "value"},
|
|
}},
|
|
&tengo.String{Value: "key"}}},
|
|
want: tengo.UndefinedValue,
|
|
target: &tengo.Map{Value: map[string]tengo.Object{}},
|
|
},
|
|
{name: "map-multi-keys",
|
|
args: args{
|
|
[]tengo.Object{
|
|
&tengo.Map{Value: map[string]tengo.Object{
|
|
"key1": &tengo.String{Value: "value1"},
|
|
"key2": &tengo.Int{Value: 10},
|
|
}},
|
|
&tengo.String{Value: "key1"}}},
|
|
want: tengo.UndefinedValue,
|
|
target: &tengo.Map{Value: map[string]tengo.Object{
|
|
"key2": &tengo.Int{Value: 10}}},
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
got, err := builtinDelete(tt.args.args...)
|
|
if (err != nil) != tt.wantErr {
|
|
t.Errorf("builtinDelete() error = %v, wantErr %v",
|
|
err, tt.wantErr)
|
|
return
|
|
}
|
|
if tt.wantErr && !errors.Is(err, tt.wantedErr) {
|
|
if err.Error() != tt.wantedErr.Error() {
|
|
t.Errorf("builtinDelete() error = %v, wantedErr %v",
|
|
err, tt.wantedErr)
|
|
return
|
|
}
|
|
}
|
|
if got != tt.want {
|
|
t.Errorf("builtinDelete() = %v, want %v", got, tt.want)
|
|
return
|
|
}
|
|
if !tt.wantErr && tt.target != nil {
|
|
switch v := tt.args.args[0].(type) {
|
|
case *tengo.Map, *tengo.Array:
|
|
if !reflect.DeepEqual(tt.target, tt.args.args[0]) {
|
|
t.Errorf("builtinDelete() objects are not equal "+
|
|
"got: %+v, want: %+v", tt.args.args[0], tt.target)
|
|
}
|
|
default:
|
|
t.Errorf("builtinDelete() unsuporrted arg[0] type %s",
|
|
v.TypeName())
|
|
return
|
|
}
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func Test_builtinSplice(t *testing.T) {
|
|
var builtinSplice func(args ...tengo.Object) (tengo.Object, error)
|
|
for _, f := range tengo.GetAllBuiltinFunctions() {
|
|
if f.Name == "splice" {
|
|
builtinSplice = f.Value
|
|
break
|
|
}
|
|
}
|
|
if builtinSplice == nil {
|
|
t.Fatal("builtin splice not found")
|
|
}
|
|
tests := []struct {
|
|
name string
|
|
args []tengo.Object
|
|
deleted tengo.Object
|
|
Array *tengo.Array
|
|
wantErr bool
|
|
wantedErr error
|
|
}{
|
|
{name: "no args", args: []tengo.Object{}, wantErr: true,
|
|
wantedErr: tengo.ErrWrongNumArguments,
|
|
},
|
|
{name: "invalid args", args: []tengo.Object{&tengo.Map{}},
|
|
wantErr: true,
|
|
wantedErr: tengo.ErrInvalidArgumentType{
|
|
Name: "first", Expected: "array", Found: "map"},
|
|
},
|
|
{name: "invalid args",
|
|
args: []tengo.Object{&tengo.Array{}, &tengo.String{}},
|
|
wantErr: true,
|
|
wantedErr: tengo.ErrInvalidArgumentType{
|
|
Name: "second", Expected: "int", Found: "string"},
|
|
},
|
|
{name: "negative index",
|
|
args: []tengo.Object{&tengo.Array{}, &tengo.Int{Value: -1}},
|
|
wantErr: true,
|
|
wantedErr: tengo.ErrIndexOutOfBounds},
|
|
{name: "non int count",
|
|
args: []tengo.Object{
|
|
&tengo.Array{}, &tengo.Int{Value: 0},
|
|
&tengo.String{Value: ""}},
|
|
wantErr: true,
|
|
wantedErr: tengo.ErrInvalidArgumentType{
|
|
Name: "third", Expected: "int", Found: "string"},
|
|
},
|
|
{name: "negative count",
|
|
args: []tengo.Object{
|
|
&tengo.Array{Value: []tengo.Object{
|
|
&tengo.Int{Value: 0},
|
|
&tengo.Int{Value: 1},
|
|
&tengo.Int{Value: 2}}},
|
|
&tengo.Int{Value: 0},
|
|
&tengo.Int{Value: -1}},
|
|
wantErr: true,
|
|
wantedErr: tengo.ErrIndexOutOfBounds,
|
|
},
|
|
{name: "insert with zero count",
|
|
args: []tengo.Object{
|
|
&tengo.Array{Value: []tengo.Object{
|
|
&tengo.Int{Value: 0},
|
|
&tengo.Int{Value: 1},
|
|
&tengo.Int{Value: 2}}},
|
|
&tengo.Int{Value: 0},
|
|
&tengo.Int{Value: 0},
|
|
&tengo.String{Value: "b"}},
|
|
deleted: &tengo.Array{Value: []tengo.Object{}},
|
|
Array: &tengo.Array{Value: []tengo.Object{
|
|
&tengo.String{Value: "b"},
|
|
&tengo.Int{Value: 0},
|
|
&tengo.Int{Value: 1},
|
|
&tengo.Int{Value: 2}}},
|
|
},
|
|
{name: "insert",
|
|
args: []tengo.Object{
|
|
&tengo.Array{Value: []tengo.Object{
|
|
&tengo.Int{Value: 0},
|
|
&tengo.Int{Value: 1},
|
|
&tengo.Int{Value: 2}}},
|
|
&tengo.Int{Value: 1},
|
|
&tengo.Int{Value: 0},
|
|
&tengo.String{Value: "c"},
|
|
&tengo.String{Value: "d"}},
|
|
deleted: &tengo.Array{Value: []tengo.Object{}},
|
|
Array: &tengo.Array{Value: []tengo.Object{
|
|
&tengo.Int{Value: 0},
|
|
&tengo.String{Value: "c"},
|
|
&tengo.String{Value: "d"},
|
|
&tengo.Int{Value: 1},
|
|
&tengo.Int{Value: 2}}},
|
|
},
|
|
{name: "insert with zero count",
|
|
args: []tengo.Object{
|
|
&tengo.Array{Value: []tengo.Object{
|
|
&tengo.Int{Value: 0},
|
|
&tengo.Int{Value: 1},
|
|
&tengo.Int{Value: 2}}},
|
|
&tengo.Int{Value: 1},
|
|
&tengo.Int{Value: 0},
|
|
&tengo.String{Value: "c"},
|
|
&tengo.String{Value: "d"}},
|
|
deleted: &tengo.Array{Value: []tengo.Object{}},
|
|
Array: &tengo.Array{Value: []tengo.Object{
|
|
&tengo.Int{Value: 0},
|
|
&tengo.String{Value: "c"},
|
|
&tengo.String{Value: "d"},
|
|
&tengo.Int{Value: 1},
|
|
&tengo.Int{Value: 2}}},
|
|
},
|
|
{name: "insert with delete",
|
|
args: []tengo.Object{
|
|
&tengo.Array{Value: []tengo.Object{
|
|
&tengo.Int{Value: 0},
|
|
&tengo.Int{Value: 1},
|
|
&tengo.Int{Value: 2}}},
|
|
&tengo.Int{Value: 1},
|
|
&tengo.Int{Value: 1},
|
|
&tengo.String{Value: "c"},
|
|
&tengo.String{Value: "d"}},
|
|
deleted: &tengo.Array{
|
|
Value: []tengo.Object{&tengo.Int{Value: 1}}},
|
|
Array: &tengo.Array{Value: []tengo.Object{
|
|
&tengo.Int{Value: 0},
|
|
&tengo.String{Value: "c"},
|
|
&tengo.String{Value: "d"},
|
|
&tengo.Int{Value: 2}}},
|
|
},
|
|
{name: "insert with delete multi",
|
|
args: []tengo.Object{
|
|
&tengo.Array{Value: []tengo.Object{
|
|
&tengo.Int{Value: 0},
|
|
&tengo.Int{Value: 1},
|
|
&tengo.Int{Value: 2}}},
|
|
&tengo.Int{Value: 1},
|
|
&tengo.Int{Value: 2},
|
|
&tengo.String{Value: "c"},
|
|
&tengo.String{Value: "d"}},
|
|
deleted: &tengo.Array{Value: []tengo.Object{
|
|
&tengo.Int{Value: 1},
|
|
&tengo.Int{Value: 2}}},
|
|
Array: &tengo.Array{
|
|
Value: []tengo.Object{
|
|
&tengo.Int{Value: 0},
|
|
&tengo.String{Value: "c"},
|
|
&tengo.String{Value: "d"}}},
|
|
},
|
|
{name: "delete all with positive count",
|
|
args: []tengo.Object{
|
|
&tengo.Array{Value: []tengo.Object{
|
|
&tengo.Int{Value: 0},
|
|
&tengo.Int{Value: 1},
|
|
&tengo.Int{Value: 2}}},
|
|
&tengo.Int{Value: 0},
|
|
&tengo.Int{Value: 3}},
|
|
deleted: &tengo.Array{Value: []tengo.Object{
|
|
&tengo.Int{Value: 0},
|
|
&tengo.Int{Value: 1},
|
|
&tengo.Int{Value: 2}}},
|
|
Array: &tengo.Array{Value: []tengo.Object{}},
|
|
},
|
|
{name: "delete all with big count",
|
|
args: []tengo.Object{
|
|
&tengo.Array{Value: []tengo.Object{
|
|
&tengo.Int{Value: 0},
|
|
&tengo.Int{Value: 1},
|
|
&tengo.Int{Value: 2}}},
|
|
&tengo.Int{Value: 0},
|
|
&tengo.Int{Value: 5}},
|
|
deleted: &tengo.Array{Value: []tengo.Object{
|
|
&tengo.Int{Value: 0},
|
|
&tengo.Int{Value: 1},
|
|
&tengo.Int{Value: 2}}},
|
|
Array: &tengo.Array{Value: []tengo.Object{}},
|
|
},
|
|
{name: "nothing2",
|
|
args: []tengo.Object{
|
|
&tengo.Array{Value: []tengo.Object{
|
|
&tengo.Int{Value: 0},
|
|
&tengo.Int{Value: 1},
|
|
&tengo.Int{Value: 2}}}},
|
|
Array: &tengo.Array{Value: []tengo.Object{}},
|
|
deleted: &tengo.Array{Value: []tengo.Object{
|
|
&tengo.Int{Value: 0},
|
|
&tengo.Int{Value: 1},
|
|
&tengo.Int{Value: 2}}},
|
|
},
|
|
{name: "pop without count",
|
|
args: []tengo.Object{
|
|
&tengo.Array{Value: []tengo.Object{
|
|
&tengo.Int{Value: 0},
|
|
&tengo.Int{Value: 1},
|
|
&tengo.Int{Value: 2}}},
|
|
&tengo.Int{Value: 2}},
|
|
deleted: &tengo.Array{Value: []tengo.Object{&tengo.Int{Value: 2}}},
|
|
Array: &tengo.Array{Value: []tengo.Object{
|
|
&tengo.Int{Value: 0}, &tengo.Int{Value: 1}}},
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
got, err := builtinSplice(tt.args...)
|
|
if (err != nil) != tt.wantErr {
|
|
t.Errorf("builtinSplice() error = %v, wantErr %v",
|
|
err, tt.wantErr)
|
|
return
|
|
}
|
|
if !reflect.DeepEqual(got, tt.deleted) {
|
|
t.Errorf("builtinSplice() = %v, want %v", got, tt.deleted)
|
|
}
|
|
if tt.wantErr && tt.wantedErr.Error() != err.Error() {
|
|
t.Errorf("builtinSplice() error = %v, wantedErr %v",
|
|
err, tt.wantedErr)
|
|
}
|
|
if tt.Array != nil && !reflect.DeepEqual(tt.Array, tt.args[0]) {
|
|
t.Errorf("builtinSplice() arrays are not equal expected"+
|
|
" %s, got %s", tt.Array, tt.args[0].(*tengo.Array))
|
|
}
|
|
})
|
|
}
|
|
}
|