-
Notifications
You must be signed in to change notification settings - Fork 2
/
Copy pathdriver_test.go
131 lines (125 loc) · 3.62 KB
/
driver_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
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
package dingo
import (
"github.com/go-sql-driver/mysql"
"github.com/stretchr/testify/require"
"testing"
"io"
)
func Test_select(t *testing.T) {
should := require.New(t)
drv := mysql.MySQLDriver{}
conn, err := Open(drv, "root:123456@tcp(127.0.0.1:3306)/v2pro")
should.Nil(err)
defer conn.Close()
conn.Exec(Translate("TRUNCATE account"))
stmt := conn.TranslateStatement(
`SELECT :SELECT_COLUMNS FROM account
WHERE entity_id=:entity_id`, "entity_id", "event_id", "state")
should.Nil(err)
defer stmt.Close()
rows, err := stmt.Query(
"PREPARED", true,
"entity_id", "account1")
should.Nil(err)
defer rows.Close()
should.Equal(io.EOF, rows.Next())
}
func Test_select_in(t *testing.T) {
should := require.New(t)
drv := mysql.MySQLDriver{}
conn, err := Open(drv, "root:123456@tcp(127.0.0.1:3306)/v2pro")
should.Nil(err)
defer conn.Close()
conn.Exec(Translate("TRUNCATE account"))
conn.Exec(Translate("INSERT account :INSERT_COLUMNS",
"entity_id", "event_id", "event_name", "command", "state"),
"entity_id", "account1",
"event_id", int64(1),
"event_name", "created",
"command", "{}",
"state", "{}")
stmt := conn.TranslateStatement(
"SELECT * FROM account WHERE entity_id IN :STR_ENTITY_IDS")
defer stmt.Close()
rows, err := stmt.Query(
"STR_ENTITY_IDS", Tuple("account1"))
should.Nil(err)
defer rows.Close()
should.Nil(rows.Next())
should.Equal("{}", rows.Get(rows.C("state")))
}
func Test_update(t *testing.T) {
should := require.New(t)
drv := mysql.MySQLDriver{}
conn, err := Open(drv, "root:123456@tcp(127.0.0.1:3306)/v2pro")
should.Nil(err)
defer conn.Close()
conn.Exec(Translate("TRUNCATE account"))
conn.Exec(Translate("INSERT account :INSERT_COLUMNS",
"entity_id", "event_id", "event_name", "command", "state"),
"entity_id", "account1",
"event_id", int64(1),
"event_name", "created",
"command", "{}",
"state", "{}")
result, err := conn.Exec(Translate("UPDATE account SET :UPDATE_COLUMNS WHERE entity_id=:entity_id",
"event_id", "state"),
"entity_id", "account1",
"event_id", int64(2),
"state", "{}")
should.Nil(err)
rowsAffected, err := result.RowsAffected()
should.Nil(err)
should.Equal(int64(1), rowsAffected)
}
func Test_insert(t *testing.T) {
should := require.New(t)
drv := mysql.MySQLDriver{}
conn, err := Open(drv, "root:123456@tcp(127.0.0.1:3306)/v2pro")
should.Nil(err)
defer conn.Close()
conn.Exec(Translate("TRUNCATE account"))
stmt := conn.TranslateStatement(
"INSERT account :INSERT_COLUMNS",
"entity_id", "event_id", "event_name", "command", "state")
defer stmt.Close()
result, err := stmt.Exec(
"entity_id", "account1",
"event_id", int64(1),
"event_name", "created",
"command", "{}",
"state", "{}")
should.Nil(err)
rowsAffected, err := result.RowsAffected()
should.Nil(err)
should.Equal(int64(1), rowsAffected)
}
func Test_batch_insert(t *testing.T) {
should := require.New(t)
drv := mysql.MySQLDriver{}
conn, err := Open(drv, "root:123456@tcp(127.0.0.1:3306)/v2pro")
should.Nil(err)
defer conn.Close()
conn.Exec(Translate("TRUNCATE account"))
stmt := conn.TranslateStatement("INSERT account :BATCH_INSERT_COLUMNS",
BatchInsertColumns(2, "entity_id", "event_id", "event_name", "command", "state"))
should.Nil(err)
defer stmt.Close()
result, err := stmt.Exec(
BatchInsertRow(
"entity_id", "account1",
"event_id", int64(1),
"event_name", "created",
"command", "{}",
"state", "{}"),
BatchInsertRow(
"entity_id", "account1",
"event_id", int64(2),
"event_name", "bill1_transfer",
"command", "{}",
"state", "{}"))
should.Nil(err)
rowsAffected, err := result.RowsAffected()
should.Nil(err)
should.Equal(int64(2), rowsAffected)
}