-
Notifications
You must be signed in to change notification settings - Fork 465
/
insert_test.go
112 lines (83 loc) · 2.57 KB
/
insert_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
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
package squirrel
import (
"testing"
"github.com/stretchr/testify/assert"
)
func TestInsertBuilderToSql(t *testing.T) {
b := Insert("").
Prefix("WITH prefix AS ?", 0).
Into("a").
Options("DELAYED", "IGNORE").
Columns("b", "c").
Values(1, 2).
Values(3, Expr("? + 1", 4)).
Suffix("RETURNING ?", 5)
sql, args, err := b.ToSql()
assert.NoError(t, err)
expectedSQL :=
"WITH prefix AS ? " +
"INSERT DELAYED IGNORE INTO a (b,c) VALUES (?,?),(?,? + 1) " +
"RETURNING ?"
assert.Equal(t, expectedSQL, sql)
expectedArgs := []interface{}{0, 1, 2, 3, 4, 5}
assert.Equal(t, expectedArgs, args)
}
func TestInsertBuilderToSqlErr(t *testing.T) {
_, _, err := Insert("").Values(1).ToSql()
assert.Error(t, err)
_, _, err = Insert("x").ToSql()
assert.Error(t, err)
}
func TestInsertBuilderMustSql(t *testing.T) {
defer func() {
if r := recover(); r == nil {
t.Errorf("TestInsertBuilderMustSql should have panicked!")
}
}()
Insert("").MustSql()
}
func TestInsertBuilderPlaceholders(t *testing.T) {
b := Insert("test").Values(1, 2)
sql, _, _ := b.PlaceholderFormat(Question).ToSql()
assert.Equal(t, "INSERT INTO test VALUES (?,?)", sql)
sql, _, _ = b.PlaceholderFormat(Dollar).ToSql()
assert.Equal(t, "INSERT INTO test VALUES ($1,$2)", sql)
}
func TestInsertBuilderRunners(t *testing.T) {
db := &DBStub{}
b := Insert("test").Values(1).RunWith(db)
expectedSQL := "INSERT INTO test VALUES (?)"
b.Exec()
assert.Equal(t, expectedSQL, db.LastExecSql)
}
func TestInsertBuilderNoRunner(t *testing.T) {
b := Insert("test").Values(1)
_, err := b.Exec()
assert.Equal(t, RunnerNotSet, err)
}
func TestInsertBuilderSetMap(t *testing.T) {
b := Insert("table").SetMap(Eq{"field1": 1, "field2": 2, "field3": 3})
sql, args, err := b.ToSql()
assert.NoError(t, err)
expectedSQL := "INSERT INTO table (field1,field2,field3) VALUES (?,?,?)"
assert.Equal(t, expectedSQL, sql)
expectedArgs := []interface{}{1, 2, 3}
assert.Equal(t, expectedArgs, args)
}
func TestInsertBuilderSelect(t *testing.T) {
sb := Select("field1").From("table1").Where(Eq{"field1": 1})
ib := Insert("table2").Columns("field1").Select(sb)
sql, args, err := ib.ToSql()
assert.NoError(t, err)
expectedSQL := "INSERT INTO table2 (field1) SELECT field1 FROM table1 WHERE field1 = ?"
assert.Equal(t, expectedSQL, sql)
expectedArgs := []interface{}{1}
assert.Equal(t, expectedArgs, args)
}
func TestInsertBuilderReplace(t *testing.T) {
b := Replace("table").Values(1)
expectedSQL := "REPLACE INTO table VALUES (?)"
sql, _, err := b.ToSql()
assert.NoError(t, err)
assert.Equal(t, expectedSQL, sql)
}