From 30f692f3537ef7d027cc008a665307f52f13e25f Mon Sep 17 00:00:00 2001
From: Philip O'Toole <philip.otoole@yahoo.com>
Date: Fri, 12 Jan 2024 09:51:55 -0500
Subject: [PATCH] Improve Boot unit testing

---
 store/store_test.go | 25 ++++++++++++++++++++++++-
 1 file changed, 24 insertions(+), 1 deletion(-)

diff --git a/store/store_test.go b/store/store_test.go
index b88498bd..4aeaba8a 100644
--- a/store/store_test.go
+++ b/store/store_test.go
@@ -1351,7 +1351,7 @@ COMMIT;
 		t.Fatalf("failed to load SQLite file: %s", err.Error())
 	}
 
-	// Load a database should mark that the snapshot store needs a Full Snapshot
+	// Loading a database should mark that the snapshot store needs a Full Snapshot
 	fn, err := s.snapshotStore.FullNeeded()
 	if err != nil {
 		t.Fatalf("failed to check if snapshot store needs a full snapshot: %s", err.Error())
@@ -1563,6 +1563,29 @@ COMMIT;
 	if exp, got := `{"error":"no such table: bar"}`, asJSON(r[0]); exp != got {
 		t.Fatalf("unexpected results for query\nexp: %s\ngot: %s", exp, got)
 	}
+
+	// Confirm that after restarting the Store, the data still looks good.
+	if err := s.Close(true); err != nil {
+		t.Fatalf("failed to close store: %s", err.Error())
+	}
+	if err := s.Open(); err != nil {
+		t.Fatalf("failed to open store: %s", err.Error())
+	}
+	if _, err := s.WaitForLeader(10 * time.Second); err != nil {
+		t.Fatalf("Error waiting for leader: %s", err)
+	}
+	qr = queryRequestFromString("SELECT count(*) FROM foo", false, true)
+	r, err = s.Query(qr)
+	if err != nil {
+		t.Fatalf("failed to query single node: %s", err.Error())
+	}
+	if exp, got := `["count(*)"]`, asJSON(r[0].Columns); exp != got {
+		t.Fatalf("unexpected results for query\nexp: %s\ngot: %s", exp, got)
+	}
+	if exp, got := `[[3]]`, asJSON(r[0].Values); exp != got {
+		t.Fatalf("unexpected results for query\nexp: %s\ngot: %s", exp, got)
+	}
+
 }
 
 func Test_SingleNodeBoot_InvalidFail_WALOK(t *testing.T) {