forked from shogo82148/go-sql-proxy
/
tracer_test.go
78 lines (68 loc) · 2.13 KB
/
tracer_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
package proxy
import (
"bufio"
"bytes"
"database/sql"
"log"
"regexp"
"testing"
"github.com/mattn/go-sqlite3"
"github.com/shogo82148/txmanager"
)
func TestTraceProxy(t *testing.T) {
buf := &bytes.Buffer{}
logger := log.New(buf, "", log.Lshortfile)
sql.Register("sqlite3-trace-proxy", NewTraceProxy(&sqlite3.SQLiteDriver{}, logger))
db, err := sql.Open("sqlite3-trace-proxy", ":memory:")
if err != nil {
t.Fatalf("Open filed: %v", err)
}
_, err = db.Exec(
"CREATE TABLE t1 (id INTEGER PRIMARY KEY)",
)
if err != nil {
t.Fatalf("create table failed: %v", err)
}
dbm := txmanager.NewDB(db)
txmanager.Do(dbm, func(tx txmanager.Tx) error {
_, err := tx.Exec("INSERT INTO t1 (id) VALUES(?)", 1)
return err
})
if err != nil {
t.Fatalf("do failed: %v", err)
}
row := dbm.QueryRow("SELECT id FROM t1 WHERE id = ?", 1)
var id int
if err = row.Scan(&id); err != nil {
t.Fatalf("selecting row failed: %v", err)
}
if id != 1 {
t.Errorf("got %d\nwant 1", id)
}
timeComponent := `\(\d+(?:\.\d+)?[^\)]+\)`
expected := []*regexp.Regexp{
// Fake time compinent with (\d+\.\d+[^\)]+)
regexp.MustCompile(`tracer_test.go:27: Open ` + timeComponent),
regexp.MustCompile(`tracer_test.go:27: Exec: CREATE TABLE t1 \(id INTEGER PRIMARY KEY\); args = \[\] ` + timeComponent),
// The line numbers on these two lines may change depending on the version
// of txmanager that you have. For now, we cross our fingers that the
// filename doesn't change ever, but the line numbers may change.
regexp.MustCompile(`txmanager.go:\d+: Begin ` + timeComponent),
regexp.MustCompile(`txmanager.go:\d+: Exec: INSERT INTO t1 \(id\) VALUES\(\?\); args = \[1\] ` + timeComponent),
regexp.MustCompile(`tracer_test.go:36: Commit ` + timeComponent),
regexp.MustCompile(`tracer_test.go:41: Query: SELECT id FROM t1 WHERE id = \?; args = \[1\] ` + timeComponent),
}
scanner := bufio.NewScanner(buf)
i := 0
for scanner.Scan() {
line := scanner.Text()
if i >= len(expected) {
t.Errorf("Got more lines than expected (%s)", line)
break
}
if !expected[i].MatchString(line) {
t.Errorf("\ngot: %s\nwant: %s", line, expected[i])
}
i++
}
}