From d5be5d3e8c06761d19677c122c2bcdfd85bbaed8 Mon Sep 17 00:00:00 2001
From: Philipp Heckel <pheckel@datto.com>
Date: Wed, 8 Dec 2021 23:24:09 -0500
Subject: [PATCH] Test schema migration

---
 server/cache_sqlite_test.go | 64 +++++++++++++++++++++++++++++++++++--
 1 file changed, 62 insertions(+), 2 deletions(-)

diff --git a/server/cache_sqlite_test.go b/server/cache_sqlite_test.go
index 4e7d2f1e..0f6c4302 100644
--- a/server/cache_sqlite_test.go
+++ b/server/cache_sqlite_test.go
@@ -1,8 +1,12 @@
 package server
 
 import (
+	"database/sql"
+	"fmt"
+	"github.com/stretchr/testify/assert"
 	"path/filepath"
 	"testing"
+	"time"
 )
 
 func TestSqliteCache_AddMessage(t *testing.T) {
@@ -21,8 +25,64 @@ func TestSqliteCache_Prune(t *testing.T) {
 	testCachePrune(t, newSqliteTestCache(t))
 }
 
-func newSqliteTestCache(t *testing.T) cache {
-	filename := filepath.Join(t.TempDir(), "cache.db")
+func TestSqliteCache_Migration_0to1(t *testing.T) {
+	filename := newSqliteTestCacheFile(t)
+	db, err := sql.Open("sqlite3", filename)
+	assert.Nil(t, err)
+
+	// Create "version 0" schema
+	_, err = db.Exec(`
+		BEGIN;
+		CREATE TABLE IF NOT EXISTS messages (
+			id VARCHAR(20) PRIMARY KEY,
+			time INT NOT NULL,
+			topic VARCHAR(64) NOT NULL,
+			message VARCHAR(1024) NOT NULL
+		);
+		CREATE INDEX IF NOT EXISTS idx_topic ON messages (topic);
+		COMMIT;
+	`)
+	assert.Nil(t, err)
+
+	// Insert a bunch of messages
+	for i := 0; i < 10; i++ {
+		_, err = db.Exec(`INSERT INTO messages (id, time, topic, message) VALUES (?, ?, ?, ?)`,
+			fmt.Sprintf("abcd%d", i), time.Now().Unix(), "mytopic", fmt.Sprintf("some message %d", i))
+		assert.Nil(t, err)
+	}
+
+	// Create cache to trigger migration
+	c := newSqliteTestCacheFromFile(t, filename)
+	messages, err := c.Messages("mytopic", sinceAllMessages)
+	assert.Nil(t, err)
+	assert.Equal(t, 10, len(messages))
+	assert.Equal(t, "some message 5", messages[5].Message)
+	assert.Equal(t, "", messages[5].Title)
+	assert.Nil(t, messages[5].Tags)
+	assert.Equal(t, 0, messages[5].Priority)
+
+	rows, err := c.db.Query(`SELECT version  FROM schemaVersion`)
+	assert.Nil(t, err)
+	assert.True(t, rows.Next())
+
+	var schemaVersion int
+	assert.Nil(t, rows.Scan(&schemaVersion))
+	assert.Equal(t, 1, schemaVersion)
+}
+
+func newSqliteTestCache(t *testing.T) *sqliteCache {
+	c, err := newSqliteCache(newSqliteTestCacheFile(t))
+	if err != nil {
+		t.Fatal(err)
+	}
+	return c
+}
+
+func newSqliteTestCacheFile(t *testing.T) string {
+	return filepath.Join(t.TempDir(), "cache.db")
+}
+
+func newSqliteTestCacheFromFile(t *testing.T, filename string) *sqliteCache {
 	c, err := newSqliteCache(filename)
 	if err != nil {
 		t.Fatal(err)