Backup fix, tests.

This commit is contained in:
Nuno Cruces
2023-05-24 02:13:52 +01:00
parent cc2d16ac83
commit 597178f80d
6 changed files with 158 additions and 4 deletions

View File

@@ -129,6 +129,6 @@ func (b *Backup) Remaining() int {
// //
// https://www.sqlite.org/c3ref/backup_finish.html#sqlite3backuppagecount // https://www.sqlite.org/c3ref/backup_finish.html#sqlite3backuppagecount
func (b *Backup) PageCount() int { func (b *Backup) PageCount() int {
r := b.c.call(b.c.api.backupFinish, uint64(b.handle)) r := b.c.call(b.c.api.backupPageCount, uint64(b.handle))
return int(r[0]) return int(r[0])
} }

View File

@@ -4,7 +4,14 @@ import (
"database/sql" "database/sql"
"fmt" "fmt"
"log" "log"
"os"
"path/filepath"
"strings"
"testing"
_ "embed"
"github.com/ncruces/go-sqlite3"
_ "github.com/ncruces/go-sqlite3/driver" _ "github.com/ncruces/go-sqlite3/driver"
_ "github.com/ncruces/go-sqlite3/embed" _ "github.com/ncruces/go-sqlite3/embed"
"github.com/ncruces/go-sqlite3/sqlite3vfs" "github.com/ncruces/go-sqlite3/sqlite3vfs"
@@ -58,3 +65,85 @@ func ExampleReaderVFS() {
// 2012.03: 18477 million Dollars // 2012.03: 18477 million Dollars
// 2012.06: 18270 million Dollars // 2012.06: 18270 million Dollars
} }
//go:embed testdata/test.db
var testDB string
func TestReaderVFS_Open(t *testing.T) {
sqlite3vfs.Register("reader", sqlite3vfs.ReaderVFS{
"test.db": sqlite3vfs.NewSizeReaderAt(strings.NewReader(testDB)),
})
_, err := sqlite3.Open("file:demo.db?vfs=reader&mode=ro")
if err == nil {
t.Error("want error")
}
db, err := sqlite3.Open("file:test.db?vfs=reader&mode=ro")
if err != nil {
t.Error(err)
}
defer db.Close()
stmt, _, err := db.Prepare(`SELECT id, name FROM users`)
if err != nil {
t.Fatal(err)
}
defer stmt.Close()
row := 0
ids := []int{0, 1, 2}
names := []string{"go", "zig", "whatever"}
for ; stmt.Step(); row++ {
id := stmt.ColumnInt(0)
name := stmt.ColumnText(1)
if id != ids[row] {
t.Errorf("got %d, want %d", id, ids[row])
}
if name != names[row] {
t.Errorf("got %q, want %q", name, names[row])
}
}
if row != 3 {
t.Errorf("got %d, want %d", row, len(ids))
}
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)
}
}
func TestNewSizeReaderAt(t *testing.T) {
n, err := sqlite3vfs.NewSizeReaderAt(strings.NewReader("abc")).Size()
if err != nil {
t.Fatal(err)
}
if n != 3 {
t.Errorf("got %d", n)
}
f, err := os.Create(filepath.Join(t.TempDir(), "abc.txt"))
if err != nil {
t.Fatal(err)
}
defer f.Close()
n, err = sqlite3vfs.NewSizeReaderAt(f).Size()
if err != nil {
t.Fatal(err)
}
if n != 0 {
t.Errorf("got %d", n)
}
}

BIN
sqlite3vfs/testdata/test.db vendored Normal file

Binary file not shown.

View File

@@ -198,6 +198,15 @@ func Test_vfsAccess(t *testing.T) {
t.Error("can't access directory") t.Error("can't access directory")
} }
util.WriteString(mod, 8, file)
rc = vfsAccess(ctx, mod, 0, 8, ACCESS_READ, 4)
if rc != _OK {
t.Fatal("returned", rc)
}
if got := util.ReadUint32(mod, 4); got != 1 {
t.Error("can't access file")
}
util.WriteString(mod, 8, file) util.WriteString(mod, 8, file)
rc = vfsAccess(ctx, mod, 0, 8, ACCESS_READWRITE, 4) rc = vfsAccess(ctx, mod, 0, 8, ACCESS_READWRITE, 4)
if rc != _OK { if rc != _OK {

View File

@@ -124,4 +124,46 @@ func TestBackup(t *testing.T) {
t.Fatal(err) t.Fatal(err)
} }
}() }()
func() { // Incremental.
db, err := sqlite3.Open(backupName)
if err != nil {
t.Fatal(err)
}
defer db.Close()
b, err := db.BackupInit("main", ":memory:")
if err != nil {
t.Fatal(err)
}
defer b.Close()
done, err := b.Step(1)
if done {
t.Error("want false")
}
if err != nil {
t.Error(err)
}
n := b.Remaining()
if n != 1 {
t.Errorf("got %d", n)
}
n = b.PageCount()
if n != 2 {
t.Errorf("got %d", n)
}
err = b.Close()
if err != nil {
t.Fatal(err)
}
err = db.Close()
if err != nil {
t.Fatal(err)
}
}()
} }

View File

@@ -27,18 +27,32 @@ func TestDriver(t *testing.T) {
} }
defer conn.Close() defer conn.Close()
_, err = conn.ExecContext(ctx, res, err := conn.ExecContext(ctx,
`CREATE TABLE IF NOT EXISTS users (id INT, name VARCHAR(10))`) `CREATE TABLE IF NOT EXISTS users (id INT, name VARCHAR(10))`)
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
changes, err := res.RowsAffected()
if err != nil {
t.Fatal(err)
}
if changes != 0 {
t.Errorf("got %d want 0", changes)
}
id, err := res.LastInsertId()
if err != nil {
t.Fatal(err)
}
if id != 0 {
t.Errorf("got %d want 0", changes)
}
res, err := conn.ExecContext(ctx, res, err = conn.ExecContext(ctx,
`INSERT INTO users (id, name) VALUES (0, 'go'), (1, 'zig'), (2, 'whatever')`) `INSERT INTO users (id, name) VALUES (0, 'go'), (1, 'zig'), (2, 'whatever')`)
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
changes, err := res.RowsAffected() changes, err = res.RowsAffected()
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }