Initial pragma support.

litestream/api_test.go

@@ -0,0 +1,104 @@
+package litestream
+
+import (
+	"path/filepath"
+	"testing"
+	"time"
+
+	"github.com/benbjohnson/litestream"
+	"github.com/benbjohnson/litestream/file"
+	"github.com/ncruces/go-sqlite3/driver"
+	_ "github.com/ncruces/go-sqlite3/embed"
+)
+
+func Test_integration(t *testing.T) {
+	dir := t.TempDir()
+	dbpath := filepath.Join(dir, "test.db")
+	backup := filepath.Join(dir, "backup", "test.db")
+
+	db, err := driver.Open(dbpath)
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer db.Close()
+
+	client := file.NewReplicaClient(backup)
+	NewReplica("test.db", client, ReplicaOptions{})
+
+	if err := setupPrimary(t, dbpath, client); err != nil {
+		t.Fatal(err)
+	}
+
+	replica, err := driver.Open("file:test.db?vfs=litestream")
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer replica.Close()
+
+	_, err = db.ExecContext(t.Context(), `CREATE TABLE users (id INT, name VARCHAR(10))`)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	_, err = db.ExecContext(t.Context(),
+		`INSERT INTO users (id, name) VALUES (0, 'go'), (1, 'zig'), (2, 'whatever')`)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	time.Sleep(DefaultPollInterval + litestream.DefaultMonitorInterval)
+
+	rows, err := replica.QueryContext(t.Context(), `SELECT id, name FROM users`)
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer rows.Close()
+
+	row := 0
+	ids := []int{0, 1, 2}
+	names := []string{"go", "zig", "whatever"}
+	for ; rows.Next(); row++ {
+		var id int
+		var name string
+		err := rows.Scan(&id, &name)
+		if err != nil {
+			t.Fatal(err)
+		}
+
+		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))
+	}
+
+	var lag int
+	err = replica.QueryRowContext(t.Context(), `PRAGMA litestream_lag`).Scan(&lag)
+	if err != nil {
+		t.Fatal(err)
+	}
+	if lag < 0 || lag > 2 {
+		t.Errorf("got %d", lag)
+	}
+
+	var txid string
+	err = replica.QueryRowContext(t.Context(), `PRAGMA litestream_txid`).Scan(&txid)
+	if err != nil {
+		t.Fatal(err)
+	}
+	if txid != "0000000000000001" {
+		t.Errorf("got %q", txid)
+	}
+}
+
+func setupPrimary(tb testing.TB, path string, client ReplicaClient) error {
+	db, err := NewPrimary(tb.Context(), path, client, nil)
+	if err == nil {
+		tb.Cleanup(func() { db.Close(tb.Context()) })
+	}
+	return err
+}