github.com/viant/toolbox@v0.34.5/collections_test.go

package toolbox_test

import (
	"strings"
	"testing"

	"github.com/stretchr/testify/assert"
	"github.com/viant/toolbox"
	"reflect"
)

func TestIndexSlice(t *testing.T) {

	{
		type Foo struct {
			id   int
			name string
		}

		var fooCollection = []Foo{{1, "A"}, {2, "B"}}
		var indexedMap = make(map[int]Foo)
		toolbox.IndexSlice(fooCollection, indexedMap, func(foo Foo) int {
			return foo.id
		})
		assert.Equal(t, "A", indexedMap[1].name)
	}

	{
		aSlice := []string{"a", "c"}
		aMap := make(map[string]int)
		index := 0
		toolbox.SliceToMap(aSlice, aMap, toolbox.CopyStringValueProvider, func(s string) int {
			index++
			return index
		})
		assert.Equal(t, 2, len(aMap))

	}
}

type sliceItem struct {
	Id int
}

func TestReverseSlice(t *testing.T) {

	{
		aSlice := []interface{}{
			"abc", "def", "cyz", "adc", "z",
		}

		toolbox.ReverseSlice(aSlice)
		assert.Equal(t, []interface{}{"z", "adc", "cyz", "def", "abc"}, aSlice)
	}

	toolbox.ReverseSlice(nil)
	{
		aSlice := []*sliceItem{
			{1}, {10},
		}

		toolbox.ReverseSlice(aSlice)
		assert.Equal(t, []*sliceItem{
			{10}, {1},
		}, aSlice)
	}

}

func TestProcessSlice(t *testing.T) {
	{
		aSlice := []interface{}{
			"abc", "def", "cyz", "adc",
		}
		count := 0
		toolbox.ProcessSlice(aSlice, func(item interface{}) bool {
			count++
			return true
		})

		assert.Equal(t, 4, count)
	}

	{
		aSlice := []string{
			"abc", "def", "cyz", "adc",
		}
		count := 0
		toolbox.ProcessSlice(aSlice, func(item interface{}) bool {
			count++
			return false
		})

		assert.Equal(t, 1, count)
	}

	{
		aSlice := []int{
			1, 2, 3,
		}
		count := 0
		toolbox.ProcessSlice(aSlice, func(item interface{}) bool {
			count++
			return false
		})

		assert.Equal(t, 1, count)
	}
	{
		aSlice := []string{
			"abc", "def", "cyz", "adc",
		}
		count := 0
		toolbox.ProcessSlice(aSlice, func(item interface{}) bool {
			count++
			return true
		})

		assert.Equal(t, 4, count)
	}
	{
		aSlice := []interface{}{
			"abc", "def", "cyz", "adc",
		}

		count := 0
		toolbox.ProcessSlice(aSlice, func(item interface{}) bool {
			count++
			return false
		})
		assert.Equal(t, 1, count)
	}
}

func TestProcessSliceWithIndex(t *testing.T) {
	{
		aSlice := []interface{}{
			"abc", "def", "cyz", "adc",
		}
		count := 0
		toolbox.ProcessSliceWithIndex(aSlice, func(index int, item interface{}) bool {
			count = index
			return true
		})

		assert.Equal(t, 3, count)
	}
	{
		aSlice := []string{
			"abc", "def", "cyz", "adc",
		}
		count := 0
		toolbox.ProcessSliceWithIndex(aSlice, func(index int, item interface{}) bool {
			count = index
			return true
		})

		assert.Equal(t, 3, count)
	}
}

func TestMakeMapFromSlice(t *testing.T) {
	type Foo struct {
		id   int
		name string
	}

	var fooCollection = []Foo{{1, "A"}, {2, "B"}}
	var testMap = make(map[int]string)
	toolbox.SliceToMap(fooCollection, testMap, func(foo Foo) int {
		return foo.id
	},
		func(foo Foo) string {
			return foo.name
		})

	assert.Equal(t, "A", testMap[1])
	assert.Equal(t, "B", testMap[2])

}

func TestSliceToMap(t *testing.T) {
	aSlice := []string{"a", "c"}
	aMap := make(map[string]bool)

	toolbox.SliceToMap(aSlice, aMap, func(s string) string {
		return s
	}, func(s string) bool {
		return true
	})
	assert.Equal(t, 2, len(aMap))

}

func TestProcess2DSliceInBatches(t *testing.T) {
	slice := [][]interface{}{
		{1, 2, 3},
		{4, 5, 7},
		{7, 8, 9},
		{10, 11, 12},
		{13, 14, 15},
		{16, 17, 18},
		{19, 20, 21},
	}

	actualItemCount := 0
	toolbox.Process2DSliceInBatches(slice, 2, func(item [][]interface{}) {
		actualItemCount = actualItemCount + len(item)
	})
	assert.Equal(t, 7, actualItemCount)
}

func TestCopySliceElements(t *testing.T) {
	{
		source := []interface{}{
			"abc", "def", "cyz",
		}
		var target = make([]string, 0)
		toolbox.CopySliceElements(source, &target)
		assert.Equal(t, 3, len(target))
		for i := 0; i < len(source); i++ {
			assert.Equal(t, source[i], target[i])
		}
	}
	{
		source := []interface{}{
			1, 2, 3,
		}
		var target = make([]int, 0)
		toolbox.CopySliceElements(source, &target)
		assert.Equal(t, 3, len(target))
		for i := 0; i < len(source); i++ {
			assert.Equal(t, source[i], target[i])
		}
	}
	{
		source := []interface{}{
			1, 2, 3,
		}
		var target = make([]interface{}, 0)
		toolbox.CopySliceElements(source, &target)
		assert.Equal(t, 3, len(target))
		for i := 0; i < len(source); i++ {
			assert.Equal(t, source[i], target[i])
		}
	}

}

func TestFilterSliceElements(t *testing.T) {
	{
		source := []interface{}{
			1, 2, 3,
		}
		var target = make([]int, 0)
		//filter all elements starting with a
		toolbox.FilterSliceElements(source, func(item int) bool {
			return item > 1
		}, &target)
		assert.Equal(t, 2, len(target))
		assert.Equal(t, 2, target[0])
		assert.Equal(t, 3, target[1])
	}

	{
		source := []interface{}{
			"abc", "def", "cyz", "adc",
		}
		var target = make([]string, 0)
		//filter all elements starting with a
		toolbox.FilterSliceElements(source, func(item string) bool {
			return strings.HasPrefix(item, "a")
		}, &target)
		assert.Equal(t, 2, len(target))
		assert.Equal(t, "abc", target[0])
		assert.Equal(t, "adc", target[1])
	}

}

func TestHasSliceAnyElements(t *testing.T) {
	source := []interface{}{
		"abc", "def", "cyz", "adc",
	}
	assert.True(t, toolbox.HasSliceAnyElements(source, "cyz"))
	assert.False(t, toolbox.HasSliceAnyElements(source, "cyze"))
	assert.True(t, toolbox.HasSliceAnyElements(source, "cyze", "cyz"))
}

func TestMapKeysToSlice(t *testing.T) {
	m := map[string]int{
		"abc": 1,
		"efg": 2,
	}
	var keys = make([]string, 0)
	toolbox.MapKeysToSlice(m, &keys)
	assert.Equal(t, 2, len(keys))
}

func TestMapKeysToStringSlice(t *testing.T) {
	m := map[string]int{
		"abc": 1,
		"efg": 2,
	}
	slice := toolbox.MapKeysToStringSlice(m)
	assert.Equal(t, 2, len(slice))
}

func TestCopyMapEntries(t *testing.T) {
	type Foo struct {
		id   int
		name string
	}
	source := map[interface{}]interface{}{
		1: Foo{1, "A"},
		2: Foo{2, "B"},
	}
	var target = make(map[int]Foo)

	toolbox.CopyMapEntries(source, target)
	assert.Equal(t, 2, len(target))
	assert.Equal(t, "B", target[2].name)
}

func TestIndexMultimap(t *testing.T) {
	type Product struct{ vendor, name string }
	products := []Product{
		{"Vendor1", "Product1"},
		{"Vendor2", "Product2"},
		{"Vendor1", "Product3"},
		{"Vendor1", "Product4"},
	}

	productsByVendor := make(map[string][]Product)
	toolbox.GroupSliceElements(products, productsByVendor, func(product Product) string {
		return product.vendor
	})
	assert.Equal(t, 2, len(productsByVendor))
	assert.Equal(t, 3, len(productsByVendor["Vendor1"]))
	assert.Equal(t, "Product4", productsByVendor["Vendor1"][2].name)

}

func TestSliceToMultiMap(t *testing.T) {
	type Product struct {
		vendor, name string
		productId    int
	}

	products := []Product{
		{"Vendor1", "Product1", 1},
		{"Vendor2", "Product2", 2},
		{"Vendor1", "Product3", 3},
		{"Vendor1", "Product4", 4},
	}

	productsByVendor := make(map[string][]int)
	toolbox.SliceToMultimap(products, productsByVendor, func(product Product) string {
		return product.vendor
	},
		func(product Product) int {
			return product.productId
		})

	assert.Equal(t, 2, len(productsByVendor))
	assert.Equal(t, 3, len(productsByVendor["Vendor1"]))
	assert.Equal(t, 4, productsByVendor["Vendor1"][2])

}

func TestTransformSlice(t *testing.T) {
	type Product struct{ vendor, name string }
	products := []Product{
		{"Vendor1", "Product1"},
		{"Vendor2", "Product2"},
		{"Vendor1", "Product3"},
		{"Vendor1", "Product4"},
	}
	var vendors = make([]string, 0)
	toolbox.TransformSlice(products, &vendors, func(product Product) string {
		return product.vendor
	})
	assert.Equal(t, 4, len(vendors))
	assert.Equal(t, "Vendor1", vendors[3])
}

func TestMakeStringMap(t *testing.T) {
	aMap := toolbox.MakeStringMap("a:1, b:2", ":", ",")
	assert.Equal(t, 2, len(aMap))
	assert.Equal(t, "1", aMap["a"])
	assert.Equal(t, "2", aMap["b"])
}

func TestMakeReverseStringMap(t *testing.T) {
	aMap := toolbox.MakeReverseStringMap("a:1, b:2", ":", ",")
	assert.Equal(t, 2, len(aMap))
	assert.Equal(t, "a", aMap["1"])
	assert.Equal(t, "b", aMap["2"])
}

func TestSortStrings(t *testing.T) {
	sorted := toolbox.SortStrings([]string{"z", "b", "c", "a"})
	assert.Equal(t, "a", sorted[0])
	assert.Equal(t, "z", sorted[3])

}

func TestJoinAsString(t *testing.T) {
	assert.Equal(t, "a,b", toolbox.JoinAsString([]string{"a", "b"}, ","))
}

func TestSetSliceValue(t *testing.T) {

	{
		var aSlice = make([]string, 2)
		toolbox.SetSliceValue(aSlice, 0, "abc")
		assert.Equal(t, "abc", aSlice[0])
		assert.Equal(t, "abc", toolbox.GetSliceValue(aSlice, 0))
	}

	{
		var aSlice = make([]int, 2)
		toolbox.SetSliceValue(aSlice, 0, 100)
		assert.Equal(t, 100, aSlice[0])
		assert.Equal(t, 100, toolbox.GetSliceValue(aSlice, 0))
	}
	{
		var aSlice = make([]interface{}, 2)
		toolbox.SetSliceValue(aSlice, 0, "a")
		assert.Equal(t, "a", aSlice[0])
		assert.Equal(t, "a", toolbox.GetSliceValue(aSlice, 0))
	}
}

func TestTrueValueProvider(t *testing.T) {
	assert.True(t, toolbox.TrueValueProvider(1))
}

func Test_DeleteEmptyKeys(t *testing.T) {
	aMap := map[string]interface{}{
		"k1": []int{},
		"k2": []int{1},
		"k3": "",
		"k40": map[interface{}]interface{}{
			"k1":  nil,
			1:     2,
			"k31": []map[string]interface{}{},
			"k41": []map[string]interface{}{
				{
					"z": 1,
				},
			},
		},
		"k5": map[string]interface{}{
			"k1": "",
			"10": 20,
		},
	}
	cloned := toolbox.DeleteEmptyKeys(aMap)
	assert.Equal(t, map[string]interface{}{
		"k2": []interface{}{1},
		"k40": map[interface{}]interface{}{
			1: 2,
			"k41": []interface{}{
				map[string]interface{}{
					"z": 1,
				},
			},
		},
		"k5": map[string]interface{}{
			"10": 20,
		},
	}, cloned)
}

func TestReplaceMapKeys(t *testing.T) {

	aMap := map[string]interface{}{
		"k1": []int{},
		"k2": []int{1},
		"k3": "",
		"k40": map[interface{}]interface{}{
			"k1":  nil,
			1:     2,
			"k31": []map[string]interface{}{},
			"k41": []map[string]interface{}{
				{
					"z": 1,
				},
			},
		},
		"k5": map[string]interface{}{
			"k1":  "",
			"k10": 20,
		},
	}
	replaced := toolbox.ReplaceMapKeys(aMap, map[string]interface{}{
		"k1":  123,
		"k10": 30,
	}, true)

	assert.EqualValues(t, replaced, map[string]interface{}{
		"k1": 123,
		"k2": []interface{}{1},
		"k40": map[interface{}]interface{}{
			"k1": 123,
			1:    2,
			"k41": []interface{}{
				map[string]interface{}{
					"z": 1,
				},
			},
		},
		"k5": map[string]interface{}{
			"k10": 30,
			"k1":  123,
		},
	})
}





func TestIntersection(t *testing.T) {

	useCase1Actual := []string{}
	useCase2Actual := []int{}
	useCase3Actual := []float32{}

	var useCases = []struct {
		description string
		sliceA      interface{}
		sliceB      interface{}
		actual      interface{}
		expect      interface{}
		hasError    bool
	}{
		{
			description: "string slice intersection",
			sliceA:      []string{"a", "bc", "z", "eee"},
			sliceB:      []string{"a2", "bc", "5z", "eee"},
			actual:      &useCase1Actual,
			expect:      []string{"bc", "eee"},
		},
		{
			description: "int slice intersection",
			sliceA:      []int{1, 2, 3, 4},
			sliceB:      []int{3, 4, 5, 6},
			actual:      &useCase2Actual,
			expect:      []int{3, 4},
		},
		{
			description: "float slice intersection",
			sliceA:      []float32{1.1, 2.1, 3.1, 4.1},
			sliceB:      []float32{3.1, 4.1, 5.1, 6.1},
			actual:      &useCase3Actual,
			expect:      []float32{3.1, 4.1},
		},
	}

	for _, useCase := range useCases {
		err := toolbox.Intersect(useCase.sliceA, useCase.sliceB, useCase.actual)
		if useCase.hasError {
			assert.NotNil(t, err, useCase.description)
			continue
		}
		if !assert.Nil(t, err, useCase.description) {
			continue
		}
		actual := reflect.ValueOf(useCase.actual).Elem().Interface()
		assert.EqualValues(t, useCase.expect, actual, useCase.description)
	}

}