sqlite3/tests/wal_test.go

package tests

import (
	"os"
	"path/filepath"
	"testing"

	"github.com/ncruces/go-sqlite3"
	_ "github.com/ncruces/go-sqlite3/embed"
	_ "github.com/ncruces/go-sqlite3/internal/testcfg"
	"github.com/ncruces/go-sqlite3/vfs"
)

func TestWAL_enter_exit(t *testing.T) {
	if !vfs.SupportsFileLocking {
		t.Skip("skipping without locks")
	}
	t.Parallel()

	file := filepath.Join(t.TempDir(), "test.db")

	db, err := sqlite3.Open(file)
	if err != nil {
		t.Fatal(err)
	}
	defer db.Close()

	if !vfs.SupportsSharedMemory {
		err = db.Exec(`PRAGMA locking_mode=exclusive`)
		if err != nil {
			t.Fatal(err)
		}
	}

	err = db.Exec(`
		CREATE TABLE test (col);
		PRAGMA journal_mode=wal;
		SELECT * FROM test;
		PRAGMA journal_mode=delete;
		SELECT * FROM test;
		PRAGMA journal_mode=wal;
		SELECT * FROM test;
	`)
	if err != nil {
		t.Fatal(err)
	}
}

func TestWAL_readonly(t *testing.T) {
	if !vfs.SupportsSharedMemory {
		t.Skip("skipping without shared memory")
	}
	t.Parallel()

	tmp := filepath.Join(t.TempDir(), "test.db")
	err := os.WriteFile(tmp, walDB, 0666)
	if err != nil {
		t.Fatal(err)
	}

	db, err := sqlite3.OpenFlags(tmp, sqlite3.OPEN_READONLY)
	if err != nil {
		t.Fatal(err)
	}
	defer db.Close()

	stmt, _, err := db.Prepare(`SELECT * FROM sqlite_master`)
	if err != nil {
		t.Fatal(err)
	}
	defer stmt.Close()

	if stmt.Step() {
		t.Error("want no rows")
	}
}

func TestConn_WalCheckpoint(t *testing.T) {
	if !vfs.SupportsFileLocking {
		t.Skip("skipping without locks")
	}
	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 locking_mode=exlusive;
		PRAGMA journal_mode=wal;
		CREATE TABLE test (col);
	`)
	if err != nil {
		t.Fatal(err)
	}
}