From c42ca007cf2258339bd39392ed70cf57eaed58d7 Mon Sep 17 00:00:00 2001
From: Ben Johnson <benbjohnson@yahoo.com>
Date: Mon, 26 Dec 2022 14:12:07 -0700
Subject: [PATCH] Add test for rollback on database creation (#248)

---
 cmd/litefs/mount_test.go | 54 ++++++++++++++++++++++++++++++++++++++++
 1 file changed, 54 insertions(+)

diff --git a/cmd/litefs/mount_test.go b/cmd/litefs/mount_test.go
index 8c5b754..a73a053 100644
--- a/cmd/litefs/mount_test.go
+++ b/cmd/litefs/mount_test.go
@@ -161,6 +161,60 @@ func TestSingleNode_DatabaseChecksumMismatch(t *testing.T) {
 	}
 }
 
+// Ensure that node can recover if the initial database creation rolls back.
+func TestSingleNode_RecoverFromInitialRollback(t *testing.T) {
+	dir := t.TempDir()
+	cmd := runMountCommand(t, newMountCommand(t, dir, nil))
+
+	db, err := sql.Open("sqlite3", filepath.Join(cmd.Config.FUSE.Dir, "db"))
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer func() { _ = db.Close() }()
+
+	tx, err := db.Begin()
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer func() { _ = tx.Rollback() }()
+
+	// Start creating the database and then rollback.
+	if _, err := tx.Exec(`CREATE TABLE t (x)`); err != nil {
+		t.Fatal(err)
+	} else if err := tx.Rollback(); err != nil {
+		t.Fatal(err)
+	}
+
+	// Close database & mount.
+	if err := db.Close(); err != nil {
+		t.Fatal(err)
+	} else if err := cmd.Close(); err != nil {
+		t.Fatal(err)
+	}
+
+	// Reopen process. Should be able to start up and recreate the database.
+	cmd = newMountCommand(t, dir, nil)
+	if err := cmd.Run(context.Background()); err != nil {
+		t.Fatal(err)
+	}
+
+	db, err = sql.Open("sqlite3", filepath.Join(cmd.Config.FUSE.Dir, "db"))
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer func() { _ = db.Close() }()
+
+	if _, err := db.Exec(`CREATE TABLE t (x)`); err != nil {
+		t.Fatal(err)
+	} else if err := db.Close(); err != nil {
+		t.Fatal(err)
+	}
+
+	if err := cmd.Close(); err != nil {
+		t.Fatal(err)
+	}
+}
+
 func TestMultiNode_Simple(t *testing.T) {
 	cmd0 := runMountCommand(t, newMountCommand(t, t.TempDir(), nil))
 	waitForPrimary(t, cmd0)