weak-linq/weakLinq_test.go at main · StackAdapt/weak-linq · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
package weaklinq

import (
	"iter"
	"testing"
)

////////////////////////////////////////////////////////////////////////////////

type testStruct struct {
	Id       int
	Name     string
	IsActive bool
}

//----------------------------------------------------------------------------//
// Constructors                                                               //
//----------------------------------------------------------------------------//

////////////////////////////////////////////////////////////////////////////////

func TestFrom(t *testing.T) {

	//----------------------------------------------------------------------------//

	t.Run("generic", func(t *testing.T) {

		testItems := []testStruct{
			{Id: 1, Name: "Test 1"},
			{Id: 2, Name: "Test 2"},
		}

		result := From(testItems)
		resultIterator, _ := iter.Pull(result.Seq)
		if resultIterator == nil {
			t.Errorf("Expected iterator but got nil")
		}

		if item, ok := resultIterator(); ok && item != testItems[0] {
			t.Errorf("Expected first item but got %v", item)
		}

		if item, ok := resultIterator(); ok && item != testItems[1] {
			t.Errorf("Expected second item but got %v", item)
		}

		if item, ok := resultIterator(); ok {
			t.Errorf("Expected no item but got %v", item)
		}
	})

	//----------------------------------------------------------------------------//

	t.Run("empty", func(t *testing.T) {

		var testItems []testStruct
		result := From(testItems)
		resultIterator, _ := iter.Pull(result.Seq)

		if resultIterator == nil {
			t.Errorf("Expected iterator but got nil")
		}

		if item, ok := resultIterator(); ok {
			t.Errorf("Expected no item but got %v", item)
		}
	})

	//----------------------------------------------------------------------------//
}