int64_collection_test.go

package collection

import (
	"errors"
	"reflect"
	"testing"
)

func TestInt64Collection(t *testing.T) {
	arr := NewInt64Collection([]int64{1,2,3,4,5})

	arr.DD()

	max, err := arr.Max().ToInt64()
	if err != nil {
		t.Error(err)
	}

	if max != 5 {
		t.Error(errors.New("max error"))
	}


	arr2 := arr.Filter(func(obj interface{}, index int) bool {
		val := obj.(int64)
		if val > 2 {
			return true
		}
		return false
	})
	if arr2.Count() != 3 {
		t.Error(errors.New("filter error"))
	}

	out, err := arr2.ToInt64s()
	if err != nil || len(out) != 3{
		t.Error("ToInt64s error")
	}
}

func TestInt64Collection_Insert(t *testing.T) {
	{
		a := NewInt64Collection([]int64{1,2,3})
		b, err := a.Insert(1, int64(10)).ToInt64s()
		if err != nil {
			t.Error(err)
		}
		if !reflect.DeepEqual(b, []int64{1, 10, 2, 3}) {
			t.Error("insert error")
		}
	}
	{
		a := NewInt64Collection([]int64{1,2,3})
		b, err := a.Insert(0, int64(10)).ToInt64s()
		if err != nil {
			t.Error(err)
		}
		if !reflect.DeepEqual(b, []int64{10, 1, 2, 3}) {
			t.Error("insert 0 error")
		}
	}

	{
		a := NewInt64Collection([]int64{1,2,3})
		b, err := a.Insert(3, int64(10)).ToInt64s()
		if err != nil {
			t.Error(err)
		}
		if !reflect.DeepEqual(b, []int64{1, 2, 3, 10}) {
			t.Error("insert length error")
		}
	}
}