VFS integration test (#722)

Author: benbjohnson

Makefile

@@ -34,7 +34,7 @@ endif
 .PHONY: vfs
 vfs:
 	mkdir -p dist
-	go build -tags SQLITE3VFS_LOADABLE_EXT,sqlite_os_trace -o dist/litestream-vfs.a -buildmode=c-archive cmd/litestream-vfs/litestream-vfs.go
+	go build -tags SQLITE3VFS_LOADABLE_EXT -o dist/litestream-vfs.a -buildmode=c-archive cmd/litestream-vfs/litestream-vfs.go
 	mv dist/litestream-vfs.h src/litestream-vfs.h
 	gcc -framework CoreFoundation -framework Security -lresolv -g -fPIC -shared -o dist/litestream-vfs.so src/litestream-vfs.c dist/litestream-vfs.a
 

cmd/litestream-vfs-test/main.go

@@ -0,0 +1,65 @@
+package main_test
+
+import (
+	"database/sql"
+	"log/slog"
+	"os"
+	"path/filepath"
+	"testing"
+	"time"
+
+	_ "github.com/mattn/go-sqlite3"
+	"github.com/psanford/sqlite3vfs"
+
+	"github.com/benbjohnson/litestream"
+	"github.com/benbjohnson/litestream/file"
+	"github.com/benbjohnson/litestream/internal/testingutil"
+)
+
+func TestVFS_Integration(t *testing.T) {
+	t.Run("Simple", func(t *testing.T) {
+		client := file.NewReplicaClient(t.TempDir())
+		vfs := newVFS(t, client)
+		if err := sqlite3vfs.RegisterVFS("litestream", vfs); err != nil {
+			t.Fatalf("failed to register litestream vfs: %v", err)
+		}
+
+		db := testingutil.NewDB(t, filepath.Join(t.TempDir(), "db"))
+		db.MonitorInterval = 100 * time.Millisecond
+		db.Replica = litestream.NewReplica(db)
+		db.Replica.Client = client
+		if err := db.Open(); err != nil {
+			t.Fatal(err)
+		}
+		sqldb0 := testingutil.MustOpenSQLDB(t, db.Path())
+		defer testingutil.MustCloseSQLDB(t, sqldb0)
+
+		if _, err := sqldb0.Exec("CREATE TABLE t (x)"); err != nil {
+			t.Fatal(err)
+		}
+		if _, err := sqldb0.Exec("INSERT INTO t (x) VALUES (100)"); err != nil {
+			t.Fatal(err)
+		}
+		time.Sleep(2 * db.MonitorInterval)
+
+		sqldb1, err := sql.Open("sqlite3", "file:/tmp/test.db?vfs=litestream")
+		if err != nil {
+			t.Fatalf("failed to open database: %v", err)
+		}
+		defer sqldb1.Close()
+
+		// Execute query
+		var x int
+		if err := sqldb1.QueryRow("SELECT * FROM t").Scan(&x); err != nil {
+			t.Fatalf("failed to query database: %v", err)
+		} else if got, want := x, 100; got != want {
+			t.Fatalf("got %d, want %d", got, want)
+		}
+	})
+}
+
+func newVFS(tb testing.TB, client litestream.ReplicaClient) *litestream.VFS {
+	tb.Helper()
+	logger := slog.New(slog.NewTextHandler(os.Stderr, nil))
+	return litestream.NewVFS(client, logger)
+}