From dfc4cad712c2494b04f67c87e8e12e2efd4f8bda Mon Sep 17 00:00:00 2001
From: Pavel Djundik <github@xpaw.me>
Date: Fri, 2 Mar 2018 10:42:12 +0200
Subject: [PATCH] Add test coverage for sqlite plugin

---
 src/plugins/sqlite.js  |   2 +-
 test/plugins/sqlite.js | 102 +++++++++++++++++++++++++++++++++++++++++
 2 files changed, 103 insertions(+), 1 deletion(-)
 create mode 100644 test/plugins/sqlite.js

diff --git a/src/plugins/sqlite.js b/src/plugins/sqlite.js
index a00dd29e..7480ed1d 100644
--- a/src/plugins/sqlite.js
+++ b/src/plugins/sqlite.js
@@ -6,7 +6,7 @@ const sqlite3 = require("sqlite3");
 const Helper = require("../helper");
 const Msg = require("../models/msg");
 
-const currentSchemaVersion = 1;
+const currentSchemaVersion = 1520239200;
 
 const schema = [
 	// Schema version #1
diff --git a/test/plugins/sqlite.js b/test/plugins/sqlite.js
new file mode 100644
index 00000000..b5cbc52e
--- /dev/null
+++ b/test/plugins/sqlite.js
@@ -0,0 +1,102 @@
+"use strict";
+
+const fs = require("fs");
+const path = require("path");
+const expect = require("chai").expect;
+const Msg = require("../../src/models/msg");
+const Helper = require("../../src/helper");
+const MessageStorage = require("../../src/plugins/sqlite.js");
+
+describe("SQLite Message Storage", function() {
+	const expectedPath = path.join(Helper.getHomePath(), "logs", "testUser.sqlite3");
+	let store;
+
+	// Delete database file from previous test run
+	before(function(done) {
+		store = new MessageStorage();
+
+		if (fs.existsSync(expectedPath)) {
+			fs.unlink(expectedPath, done);
+		} else {
+			done();
+		}
+	});
+
+	it("should resolve an empty array when disabled", function(done) {
+		store.getMessages(null, null).then((messages) => {
+			expect(messages).to.be.empty;
+			done();
+		});
+	});
+
+	it("should create database file", function() {
+		expect(store.isEnabled).to.be.false;
+		expect(fs.existsSync(expectedPath)).to.be.false;
+
+		store.enable("testUser");
+
+		expect(store.isEnabled).to.be.true;
+		expect(fs.existsSync(expectedPath)).to.be.true;
+	});
+
+	it("should create tables", function(done) {
+		store.database.serialize(() =>
+			store.database.all("SELECT name, tbl_name, sql FROM sqlite_master WHERE type = 'table'", (err, row) => {
+				expect(err).to.be.null;
+				expect(row).to.deep.equal([{
+					name: "options",
+					tbl_name: "options",
+					sql: "CREATE TABLE options (name TEXT, value TEXT, CONSTRAINT name_unique UNIQUE (name))",
+				},
+				{
+					name: "messages",
+					tbl_name: "messages",
+					sql: "CREATE TABLE messages (network TEXT, channel TEXT, time INTEGER, type TEXT, msg TEXT)",
+				}]);
+
+				done();
+			})
+		);
+	});
+
+	it("should insert schema version to options table", function(done) {
+		store.database.serialize(() =>
+			store.database.get("SELECT value FROM options WHERE name = 'schema_version'", (err, row) => {
+				expect(err).to.be.null;
+
+				// Should be sqlite.currentSchemaVersion,
+				// compared as string because it's returned as such from the database
+				expect(row.value).to.equal("1520239200");
+
+				done();
+			})
+		);
+	});
+
+	it("should store a message", function(done) {
+		store.index("this-is-a-network-guid", "#ThisIsAChannel", new Msg({
+			time: 123456789,
+			text: "Hello from sqlite world!",
+		}));
+
+		store.database.serialize(done);
+	});
+
+	it("should retrieve previously stored message", function(done) {
+		store.getMessages({
+			uuid: "this-is-a-network-guid",
+		}, {
+			name: "#thisisaCHANNEL",
+		}).then((messages) => {
+			expect(messages).to.have.lengthOf(1);
+
+			const msg = messages[0];
+
+			expect(msg.text).to.equal("Hello from sqlite world!");
+			expect(msg.type).to.equal(Msg.Type.MESSAGE);
+			expect(msg.time.getTime()).to.equal(123456789);
+
+			done();
+		});
+	});
+});