-
Notifications
You must be signed in to change notification settings - Fork 0
/
stream_test.go
68 lines (58 loc) · 1.24 KB
/
stream_test.go
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
package sqlstream_test
import (
"errors"
"sqlstream"
"testing"
"github.com/stretchr/testify/assert"
)
type testObject struct {
ID int
Field1 string
}
func TestStream_ShouldReadRows(t *testing.T) {
t.Parallel()
type testCase struct {
name string
in *mockTRows[testObject]
expResult []sqlstream.Entry[testObject]
}
tests := []testCase{
{
name: "read rows",
in: newMockTRows[testObject]([]testObject{
{1, "1"},
{2, "2"},
}),
expResult: []sqlstream.Entry[testObject]{
{Value: testObject{1, "1"}},
{Value: testObject{2, "2"}},
},
},
{
name: "read rows errors",
in: &mockTRows[testObject]{
scanErr: errors.New("scan error"), // nolint:all
idx: -1,
data: []testObject{
{1, "1"},
{2, "2"},
},
},
expResult: []sqlstream.Entry[testObject]{
{Err: errors.New("scan error")}, // nolint:all
},
},
}
for _, tt := range tests {
tt := tt
t.Run(tt.name, func(t *testing.T) {
t.Parallel()
var results []sqlstream.Entry[testObject]
for entry := range sqlstream.Read[testObject](tt.in) {
results = append(results, entry)
}
assert.Equal(t, tt.expResult, results)
assert.True(t, tt.in.Closed(), "expected rows to be closed")
})
}
}