1// go:build unit
 2
 3package u
 4
 5import (
 6	"testing"
 7
 8	"github.com/google/go-cmp/cmp"
 9)
10
11func TestFirst(t *testing.T) {
12	testCases := []struct {
13		name  string
14		slice []int
15		first int
16		exist bool
17	}{
18		{
19			name:  "multiple items slice",
20			slice: []int{1, 2, 3},
21			first: 1,
22			exist: true,
23		},
24		{
25			name:  "single item slice",
26			slice: []int{1},
27			first: 1,
28			exist: true,
29		},
30		{
31			name:  "empty slice",
32			slice: []int{},
33			first: 0,
34			exist: false,
35		},
36	}
37	for _, tc := range testCases {
38		t.Run(tc.name, func(t *testing.T) {
39
40			first, empty := First(tc.slice)
41
42			if first != tc.first {
43				t.Errorf("Error first, want %d got %d", tc.first, first)
44			}
45
46			if empty != tc.exist {
47				t.Errorf("Error empty, want %t got %t", tc.exist, empty)
48			}
49
50		})
51	}
52}
53
54func TestSubList(t *testing.T) {
55	testCases := []struct {
56		name  string
57		slice []int
58		size  int
59		want  [][]int
60	}{
61		{
62			name:  "sigle size sub list",
63			slice: []int{1, 2, 3},
64			size:  1,
65			want:  [][]int{{1}, {2}, {3}},
66		},
67		{
68			name:  "multiple size sub list",
69			slice: []int{1, 2, 3, 4},
70			size:  2,
71			want:  [][]int{{1, 2}, {3, 4}},
72		},
73		{
74			name:  "uneven multiple size sub list",
75			slice: []int{1, 2, 3, 4, 5},
76			size:  2,
77			want:  [][]int{{1, 2}, {3, 4}, {5}},
78		},
79		{
80			name:  "empty sub list",
81			slice: []int{},
82			size:  2,
83			want:  [][]int{{}},
84		},
85	}
86	for _, tc := range testCases {
87		t.Run(tc.name, func(t *testing.T) {
88
89			subList := ChunkBy(tc.slice, tc.size)
90
91			if diff := cmp.Diff(tc.want, subList); diff != "" {
92				t.Errorf("Wrong result given - wanted + got\n %s", diff)
93			}
94		})
95	}
96}