This repository has been archived by the owner on Sep 14, 2023. It is now read-only.
/
insert_test.go
95 lines (73 loc) · 2.79 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
package dat
import (
"testing"
"gopkg.in/stretchr/testify.v1/assert"
)
type missingDbTag struct {
ID int64
}
type someRecord struct {
SomethingID int `db:"something_id"`
UserID int64 `db:"user_id"`
Other bool `db:"other"`
}
func BenchmarkInsertValuesSql(b *testing.B) {
b.ResetTimer()
for i := 0; i < b.N; i++ {
InsertInto("alpha").Columns("something_id", "user_id", "other").Values(1, 2, true).ToSQL()
}
}
func BenchmarkInsertRecordsSql(b *testing.B) {
obj := someRecord{1, 99, false}
b.ResetTimer()
for i := 0; i < b.N; i++ {
InsertInto("alpha").Columns("something_id", "user_id", "other").Record(obj).ToSQL()
}
}
func TestInsertSingleToSql(t *testing.T) {
sql, args := InsertInto("a").Columns("b", "c").Values(1, 2).ToSQL()
assert.Equal(t, sql, quoteSQL("INSERT INTO a (%s,%s) VALUES ($1,$2)", "b", "c"))
assert.Equal(t, args, []interface{}{1, 2})
}
func TestDefaultValue(t *testing.T) {
sql, args := InsertInto("a").Columns("b", "c").Values(1, DEFAULT).ToSQL()
assert.Equal(t, sql, quoteSQL("INSERT INTO a (%s,%s) VALUES ($1,$2)", "b", "c"))
assert.Equal(t, args, []interface{}{1, DEFAULT})
}
func TestInsertMultipleToSql(t *testing.T) {
sql, args := InsertInto("a").Columns("b", "c").Values(1, 2).Values(3, 4).ToSQL()
assert.Equal(t, sql, quoteSQL("INSERT INTO a (%s,%s) VALUES ($1,$2),($3,$4)", "b", "c"))
assert.Equal(t, args, []interface{}{1, 2, 3, 4})
}
func TestInsertRecordsToSql(t *testing.T) {
objs := []someRecord{{1, 88, false}, {2, 99, true}}
sql, args := InsertInto("a").Columns("something_id", "user_id", "other").Record(objs[0]).Record(objs[1]).ToSQL()
assert.Equal(t, sql, quoteSQL("INSERT INTO a (%s,%s,%s) VALUES ($1,$2,$3),($4,$5,$6)", "something_id", "user_id", "other"))
checkSliceEqual(t, args, []interface{}{1, 88, false, 2, 99, true})
}
func TestInsertWhitelist(t *testing.T) {
objs := []someRecord{{1, 88, false}, {2, 99, true}}
sql, args := InsertInto("a").
Whitelist("*").
Record(objs[0]).
Record(objs[1]).
ToSQL()
assert.Equal(t, sql, quoteSQL("INSERT INTO a (%s,%s,%s) VALUES ($1,$2,$3),($4,$5,$6)", "something_id", "user_id", "other"))
checkSliceEqual(t, []interface{}{1, 88, false, 2, 99, true}, args)
assert.Panics(t, func() {
InsertInto("a").Whitelist("*").Values("foo").ToSQL()
}, `must use "*" in conjunction with Record`)
}
func TestInsertBlacklist(t *testing.T) {
objs := []someRecord{{1, 88, false}, {2, 99, true}}
sql, args := InsertInto("a").
Blacklist("something_id").
Record(objs[0]).
Record(objs[1]).
ToSQL()
assert.Equal(t, sql, quoteSQL("INSERT INTO a (%s,%s) VALUES ($1,$2),($3,$4)", "user_id", "other"))
checkSliceEqual(t, args, []interface{}{88, false, 99, true})
assert.Panics(t, func() {
InsertInto("a").Blacklist("something_id").Values("foo").ToSQL()
}, `must use "*" in conjunction with Record`)
}