WAL and vacuum hooks.

tests/conn_test.go

@@ -11,6 +11,7 @@ import (
 
 	"github.com/ncruces/go-sqlite3"
 	_ "github.com/ncruces/go-sqlite3/embed"
+	_ "github.com/ncruces/go-sqlite3/vfs/memdb"
 )
 
 func TestConn_Open_dir(t *testing.T) {
@@ -449,3 +450,35 @@ func TestConn_DBName(t *testing.T) {
 		t.Errorf("got %s", name)
 	}
 }
+
+func TestConn_AutoVacuumPages(t *testing.T) {
+	t.Parallel()
+
+	db, err := sqlite3.Open("file:test.db?vfs=memdb&_pragma=auto_vacuum(FULL)")
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer db.Close()
+
+	err = db.AutoVacuumPages(func(schema string, dbPages, freePages, bytesPerPage uint) uint {
+		return freePages
+	})
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	err = db.Exec(`CREATE TABLE test (col)`)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	err = db.Exec(`INSERT INTO test VALUES (zeroblob(1024*1024))`)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	err = db.Exec(`DROP TABLE test`)
+	if err != nil {
+		t.Fatal(err)
+	}
+}

tests/wal_test.go

@@ -31,3 +31,34 @@ func TestWAL_enter_exit(t *testing.T) {
 		t.Fatal(err)
 	}
 }
+
+func TestConn_WalCheckpoint(t *testing.T) {
+	t.Parallel()
+
+	file := filepath.Join(t.TempDir(), "test.db")
+
+	db, err := sqlite3.Open(file)
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer db.Close()
+
+	err = db.WalAutoCheckpoint(1000)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	db.WalHook(func(db *sqlite3.Conn, schema string, pages int) error {
+		log, ckpt, err := db.WalCheckpoint(schema, sqlite3.CHECKPOINT_FULL)
+		t.Log(log, ckpt, err)
+		return err
+	})
+
+	err = db.Exec(`
+		PRAGMA journal_mode=WAL;
+		CREATE TABLE test (col);
+	`)
+	if err != nil {
+		t.Fatal(err)
+	}
+}