Add lua_sqlite3 unit tests

2 files changed, 52 insertions, 3 deletions

diff --git a/src/lua/lua_sqlite3.c b/src/lua/lua_sqlite3.c
index 34a5a9e19..4442cacab 100644
--- a/src/lua/lua_sqlite3.c
+++ b/src/lua/lua_sqlite3.c
@@ -136,7 +136,7 @@ lua_sqlite3_bind_statements (lua_State *L, gint start, gint end,
 	gsize slen;
 	gdouble n;
 
-	g_assert (start < end && start > 0 && end > 0);
+	g_assert (start <= end && start > 0 && end > 0);
 
 	for (i = start; i <= end; i ++) {
 		type = lua_type (L, i);
@@ -194,7 +194,7 @@ lua_sqlite3_sql (lua_State *L)
 
 			if (top > 2) {
 				/* Push additional arguments to sqlite3 */
-				lua_sqlite3_bind_statements (L, 2, top, stmt);
+				lua_sqlite3_bind_statements (L, 3, top, stmt);
 			}
 
 			rc = sqlite3_step (stmt);
@@ -305,7 +305,7 @@ lua_sqlite3_rows (lua_State *L)
 
 			if (top > 2) {
 				/* Push additional arguments to sqlite3 */
-				lua_sqlite3_bind_statements (L, 2, top, stmt);
+				lua_sqlite3_bind_statements (L, 3, top, stmt);
 			}
 
 			/* Create C closure */
diff --git a/test/lua/unit/sqlite3.lua b/test/lua/unit/sqlite3.lua
new file mode 100644
index 000000000..9fec2b694
--- /dev/null
+++ b/test/lua/unit/sqlite3.lua
@@ -0,0 +1,49 @@
+context("Sqlite3 API", function()
+  local sqlite3 = require "rspamd_sqlite3"
+  
+  test("Sqlite3 open", function()
+    os.remove('/tmp/rspamd_unit_test_sqlite3.sqlite')
+    local db = sqlite3.open('/tmp/rspamd_unit_test_sqlite3.sqlite')
+    assert_not_nil(db, "should be able to create sqlite3 db")
+    db = sqlite3.open('/non/existent/path/rspamd_unit_test_sqlite3.sqlite')
+    assert_nil(db, "should not be able to create sqlite3 db")
+    os.remove('/tmp/rspamd_unit_test_sqlite3.sqlite')
+  end)
+
+  test("Sqlite3 query", function()
+    os.remove('/tmp/rspamd_unit_test_sqlite3-1.sqlite')
+    local db = sqlite3.open('/tmp/rspamd_unit_test_sqlite3-1.sqlite')
+    assert_not_nil(db, "should be able to create sqlite3 db")
+    
+    local ret = db:sql([[
+      CREATE TABLE x (id INT, value TEXT);
+    ]])
+    assert_true(ret, "should be able to create table")
+    local ret = db:sql([[
+      INSERT INTO x VALUES (?1, ?2);
+    ]], 1, 'test')
+    assert_true(ret, "should be able to insert row")
+    os.remove('/tmp/rspamd_unit_test_sqlite3-1.sqlite')
+  end)
+
+  test("Sqlite3 rows", function()
+    os.remove('/tmp/rspamd_unit_test_sqlite3-2.sqlite')
+    local db = sqlite3.open('/tmp/rspamd_unit_test_sqlite3-2.sqlite')
+    assert_not_nil(db, "should be able to create sqlite3 db")
+    
+    local ret = db:sql([[
+      CREATE TABLE x (id INT, value TEXT);
+    ]])
+    assert_true(ret, "should be able to create table")
+    local ret = db:sql([[
+      INSERT INTO x VALUES (?1, ?2);
+    ]], 1, 'test')
+    assert_true(ret, "should be able to insert row")
+
+    for row in db:rows([[SELECT * FROM x;]]) do
+      assert_equal(row.id, 1)
+      assert_equal(row.value, 'test')
+    end
+    os.remove('/tmp/rspamd_unit_test_sqlite3-2.sqlite')
+  end)
+end)
\ No newline at end of file