forked from sqlc-dev/sqlc
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathparam_set_test.go
More file actions
58 lines (49 loc) · 1.49 KB
/
param_set_test.go
File metadata and controls
58 lines (49 loc) · 1.49 KB
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
package named
import "testing"
func TestParamSet_Add(t *testing.T) {
t.Parallel()
type test struct {
pset *ParamSet
param Param
expected int
}
named := NewParamSet(nil, true)
populatedNamed := NewParamSet(map[int]bool{1: true, 2: true, 4: true, 5: true, 6: true}, true)
populatedUnnamed := NewParamSet(map[int]bool{1: true, 2: true, 4: true, 5: true, 6: true}, false)
unnamed := NewParamSet(nil, false)
p1 := NewParam("hello")
p2 := NewParam("world")
tests := []test{
// First parameter should be 1
{named, p1, 1},
// Duplicate first parameters should be 1
{named, p1, 1},
// A new parameter receives a new parameter number
{named, p2, 2},
// An additional new parameter does _not_ receive a new
{named, p2, 2},
// First parameter should be 1
{unnamed, p1, 1},
// Duplicate first parameters should increment argn
{unnamed, p1, 2},
// A new parameter receives a new parameter number
{unnamed, p2, 3},
// An additional new parameter still does receive a new argn
{unnamed, p2, 4},
// First parameter of a pre-populated should be 3
{populatedNamed, p1, 3},
{populatedNamed, p1, 3},
{populatedNamed, p2, 7},
{populatedNamed, p2, 7},
{populatedUnnamed, p1, 3},
{populatedUnnamed, p1, 7},
{populatedUnnamed, p2, 8},
{populatedUnnamed, p2, 9},
}
for _, spec := range tests {
actual := spec.pset.Add(spec.param)
if actual != spec.expected {
t.Errorf("ParamSet.Add(%s) expected %v; got %v", spec.param.name, spec.expected, actual)
}
}
}