mirror of
https://github.com/ncruces/go-sqlite3.git
synced 2026-01-12 05:59:14 +00:00
342 lines
6.4 KiB
Go
342 lines
6.4 KiB
Go
package tests
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"io"
|
|
"log"
|
|
"net/url"
|
|
"os"
|
|
"os/exec"
|
|
"path/filepath"
|
|
"testing"
|
|
"time"
|
|
|
|
"golang.org/x/sync/errgroup"
|
|
|
|
"github.com/ncruces/go-sqlite3"
|
|
_ "github.com/ncruces/go-sqlite3/embed"
|
|
_ "github.com/ncruces/go-sqlite3/internal/testcfg"
|
|
"github.com/ncruces/go-sqlite3/vfs"
|
|
_ "github.com/ncruces/go-sqlite3/vfs/adiantum"
|
|
"github.com/ncruces/go-sqlite3/vfs/memdb"
|
|
)
|
|
|
|
func TestMain(m *testing.M) {
|
|
sqlite3.AutoExtension(func(c *sqlite3.Conn) error {
|
|
return c.ConfigLog(func(code sqlite3.ExtendedErrorCode, msg string) {
|
|
// Having to do journal recovery is unexpected.
|
|
if errors.Is(code, sqlite3.NOTICE) {
|
|
log.Panicf("%v (%d): %s", code, code, msg)
|
|
} else {
|
|
log.Printf("%v (%d): %s", code, code, msg)
|
|
}
|
|
})
|
|
})
|
|
m.Run()
|
|
}
|
|
|
|
func Test_parallel(t *testing.T) {
|
|
if !vfs.SupportsFileLocking {
|
|
t.Skip("skipping without locks")
|
|
}
|
|
|
|
var iter int
|
|
if testing.Short() {
|
|
iter = 1000
|
|
} else {
|
|
iter = 5000
|
|
}
|
|
|
|
name := "file:" +
|
|
filepath.ToSlash(filepath.Join(t.TempDir(), "test.db")) +
|
|
"?_pragma=busy_timeout(10000)" +
|
|
"&_pragma=journal_mode(truncate)" +
|
|
"&_pragma=synchronous(off)"
|
|
testParallel(t, name, iter)
|
|
testIntegrity(t, name)
|
|
}
|
|
|
|
func Test_wal(t *testing.T) {
|
|
if !vfs.SupportsSharedMemory {
|
|
t.Skip("skipping without shared memory")
|
|
}
|
|
|
|
var iter int
|
|
if testing.Short() {
|
|
iter = 1000
|
|
} else {
|
|
iter = 2500
|
|
}
|
|
|
|
name := "file:" +
|
|
filepath.ToSlash(filepath.Join(t.TempDir(), "test.db")) +
|
|
"?_pragma=busy_timeout(10000)" +
|
|
"&_pragma=journal_mode(wal)" +
|
|
"&_pragma=synchronous(off)"
|
|
testParallel(t, name, iter)
|
|
testIntegrity(t, name)
|
|
}
|
|
|
|
func Test_memdb(t *testing.T) {
|
|
var iter int
|
|
if testing.Short() {
|
|
iter = 1000
|
|
} else {
|
|
iter = 5000
|
|
}
|
|
|
|
name := memdb.TestDB(t, url.Values{
|
|
"_pragma": {"busy_timeout(10000)"},
|
|
})
|
|
testParallel(t, name, iter)
|
|
testIntegrity(t, name)
|
|
}
|
|
|
|
func Test_adiantum(t *testing.T) {
|
|
if !vfs.SupportsFileLocking {
|
|
t.Skip("skipping without locks")
|
|
}
|
|
|
|
var iter int
|
|
if testing.Short() {
|
|
iter = 1000
|
|
} else {
|
|
iter = 5000
|
|
}
|
|
|
|
name := "file:" +
|
|
filepath.ToSlash(filepath.Join(t.TempDir(), "test.db")) +
|
|
"?vfs=adiantum" +
|
|
"&_pragma=hexkey(e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855)" +
|
|
"&_pragma=busy_timeout(10000)" +
|
|
"&_pragma=journal_mode(truncate)" +
|
|
"&_pragma=synchronous(off)"
|
|
testParallel(t, name, iter)
|
|
testIntegrity(t, name)
|
|
}
|
|
|
|
func TestMultiProcess(t *testing.T) {
|
|
if !vfs.SupportsFileLocking {
|
|
t.Skip("skipping without locks")
|
|
}
|
|
if testing.Short() {
|
|
t.Skip("skipping in short mode")
|
|
}
|
|
|
|
file := filepath.Join(t.TempDir(), "test.db")
|
|
t.Setenv("TestMultiProcess_dbfile", file)
|
|
|
|
name := "file:" + filepath.ToSlash(file) +
|
|
"?_pragma=busy_timeout(10000)" +
|
|
"&_pragma=journal_mode(truncate)" +
|
|
"&_pragma=synchronous(off)"
|
|
|
|
exe, err := os.Executable()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
cmd := exec.Command(exe, append(os.Args[1:], "-test.v", "-test.run=TestChildProcess")...)
|
|
out, err := cmd.StdoutPipe()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := cmd.Start(); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
var buf [3]byte
|
|
// Wait for child to start.
|
|
if _, err := io.ReadFull(out, buf[:]); err != nil {
|
|
t.Fatal(err)
|
|
} else if str := string(buf[:]); str != "===" {
|
|
t.Fatal(str)
|
|
}
|
|
|
|
testParallel(t, name, 1000)
|
|
if err := cmd.Wait(); err != nil {
|
|
t.Error(err)
|
|
}
|
|
testIntegrity(t, name)
|
|
}
|
|
|
|
func TestChildProcess(t *testing.T) {
|
|
file := os.Getenv("TestMultiProcess_dbfile")
|
|
if file == "" || testing.Short() {
|
|
t.SkipNow()
|
|
}
|
|
|
|
name := "file:" + filepath.ToSlash(file) +
|
|
"?_pragma=busy_timeout(10000)" +
|
|
"&_pragma=journal_mode(truncate)" +
|
|
"&_pragma=synchronous(off)"
|
|
|
|
testParallel(t, name, 1000)
|
|
}
|
|
|
|
func Benchmark_parallel(b *testing.B) {
|
|
if !vfs.SupportsSharedMemory {
|
|
b.Skip("skipping without shared memory")
|
|
}
|
|
|
|
sqlite3.Initialize()
|
|
b.ResetTimer()
|
|
|
|
name := "file:" +
|
|
filepath.Join(b.TempDir(), "test.db") +
|
|
"?_pragma=busy_timeout(10000)" +
|
|
"&_pragma=journal_mode(truncate)" +
|
|
"&_pragma=synchronous(off)"
|
|
testParallel(b, name, b.N)
|
|
}
|
|
|
|
func Benchmark_wal(b *testing.B) {
|
|
if !vfs.SupportsSharedMemory {
|
|
b.Skip("skipping without shared memory")
|
|
}
|
|
|
|
sqlite3.Initialize()
|
|
b.ResetTimer()
|
|
|
|
name := "file:" +
|
|
filepath.Join(b.TempDir(), "test.db") +
|
|
"?_pragma=busy_timeout(10000)" +
|
|
"&_pragma=journal_mode(wal)" +
|
|
"&_pragma=synchronous(off)"
|
|
testParallel(b, name, b.N)
|
|
}
|
|
|
|
func Benchmark_memdb(b *testing.B) {
|
|
sqlite3.Initialize()
|
|
b.ResetTimer()
|
|
|
|
name := memdb.TestDB(b, url.Values{
|
|
"_pragma": {"busy_timeout(10000)"},
|
|
})
|
|
testParallel(b, name, b.N)
|
|
}
|
|
|
|
func testParallel(t testing.TB, name string, n int) {
|
|
writer := func() error {
|
|
db, err := sqlite3.Open(name)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer db.Close()
|
|
|
|
err = db.BusyHandler(func(ctx context.Context, count int) (retry bool) {
|
|
select {
|
|
case <-time.After(time.Millisecond):
|
|
return true
|
|
case <-ctx.Done():
|
|
return false
|
|
}
|
|
})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
err = db.Exec(`CREATE TABLE IF NOT EXISTS users (id INT, name VARCHAR(10))`)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
err = db.Exec(`INSERT INTO users (id, name) VALUES (0, 'go'), (1, 'zig'), (2, 'whatever')`)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return db.Close()
|
|
}
|
|
|
|
reader := func() error {
|
|
db, err := sqlite3.Open(name)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer db.Close()
|
|
|
|
stmt, _, err := db.Prepare(`SELECT id, name FROM users`)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer stmt.Close()
|
|
|
|
row := 0
|
|
for stmt.Step() {
|
|
row++
|
|
}
|
|
if err := stmt.Err(); err != nil {
|
|
return err
|
|
}
|
|
if row%3 != 0 {
|
|
t.Errorf("got %d rows, want multiple of 3", row)
|
|
}
|
|
|
|
err = stmt.Close()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return db.Close()
|
|
}
|
|
|
|
err := writer()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
var group errgroup.Group
|
|
group.SetLimit(6)
|
|
for i := 0; i < n; i++ {
|
|
if i&7 != 7 {
|
|
group.Go(reader)
|
|
} else {
|
|
group.Go(writer)
|
|
}
|
|
}
|
|
err = group.Wait()
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func testIntegrity(t testing.TB, name string) {
|
|
db, err := sqlite3.Open(name)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer db.Close()
|
|
|
|
test := `PRAGMA integrity_check`
|
|
if testing.Short() {
|
|
test = `PRAGMA quick_check`
|
|
}
|
|
|
|
stmt, _, err := db.Prepare(test)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer stmt.Close()
|
|
|
|
for stmt.Step() {
|
|
if row := stmt.ColumnText(0); row != "ok" {
|
|
t.Error(row)
|
|
}
|
|
}
|
|
if err := stmt.Err(); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
err = stmt.Close()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
err = db.Close()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|