test: add llm_chat_vtab test and improve test utilities

Introduces a new test, test_llm_chat_vtab, to verify the llm_chat virtual table functionality and memory usage. Adds the exec_select_rows helper for row-counting and verbose output. Updates test_llm_chat_respond_repeated to use varied prompts and more detailed context usage queries. Removes redundant SQLITE_ENABLE_LOAD_EXTENSION macro definition.

Author: andinux

tests/c/unittest.c

@@ -1,6 +1,3 @@
-#ifndef SQLITE_ENABLE_LOAD_EXTENSION
-#define SQLITE_ENABLE_LOAD_EXTENSION 1
-#endif
 #include "sqlite3.h"
 
 #include <stdbool.h>
@@ -69,6 +66,10 @@ static int open_db_and_load(const test_env *env, sqlite3 **out_db) {
     return SQLITE_OK;
 }
 
+// ---------------------------------------------------------------------
+// Helper utilities
+// ---------------------------------------------------------------------
+
 typedef struct {
     const test_env *env;
 } exec_userdata;
@@ -120,6 +121,45 @@ static int exec_expect_ok(const test_env *env, sqlite3 *db, const char *sql) {
     return 0;
 }
 
+static int exec_select_rows(const test_env *env, sqlite3 *db, const char *sql, int *rows_out) {
+    if (env->verbose) {
+        printf("[SQL] %s\n", sql);
+    }
+    sqlite3_stmt *stmt = NULL;
+    int rc = sqlite3_prepare_v2(db, sql, -1, &stmt, NULL);
+    if (rc != SQLITE_OK) {
+        fprintf(stderr, "sqlite3_prepare_v2 failed (%d): %s\n", rc, sqlite3_errmsg(db));
+        if (stmt) sqlite3_finalize(stmt);
+        return 1;
+    }
+    int row_count = 0;
+    while ((rc = sqlite3_step(stmt)) == SQLITE_ROW) {
+        row_count++;
+        if (env->verbose) {
+            printf("[SQL][ROW] ");
+            int cols = sqlite3_column_count(stmt);
+            for (int i = 0; i < cols; ++i) {
+                const char *name = sqlite3_column_name(stmt, i);
+                const unsigned char *val = sqlite3_column_text(stmt, i);
+                printf("%s=%s%s", name ? name : "(null)", val ? (const char *)val : "NULL", (i + 1 < cols) ? ", " : "");
+            }
+            printf("\n");
+        }
+    }
+    if (rc != SQLITE_DONE) {
+        fprintf(stderr, "sqlite3_step failed (%d): %s\n", rc, sqlite3_errmsg(db));
+        sqlite3_finalize(stmt);
+        return 1;
+    }
+    sqlite3_finalize(stmt);
+    if (rows_out) *rows_out = row_count;
+    return 0;
+}
+
+// ---------------------------------------------------------------------
+// Tests
+// ---------------------------------------------------------------------
+
 static int test_issue15_chat_without_context(const test_env *env) {
     sqlite3 *db = NULL;
     if (open_db_and_load(env, &db) != SQLITE_OK) {
@@ -144,7 +184,7 @@ static int test_llm_chat_respond_repeated(const test_env *env) {
         sqlite3_close(db);
         return 1;
     }
-    if (exec_expect_ok(env, db, "SELECT llm_context_create('context_size=32000');") != 0) {
+    if (exec_expect_ok(env, db, "SELECT llm_context_create('context_size=1000');") != 0) {
         sqlite3_close(db);
         return 1;
     }
@@ -154,13 +194,18 @@ static int test_llm_chat_respond_repeated(const test_env *env) {
     }
 
     const int iterations = 3;
+    char *prompts[] = {
+        "SELECT llm_chat_respond('Hi');", 
+        "SELECT llm_chat_respond('How are you?');",
+        "SELECT llm_chat_respond('Again');"
+    };
     for (int i = 0; i < iterations; ++i) {
-        if (exec_expect_ok(env, db, "SELECT llm_chat_respond('Hello world');") != 0) {
+        if (exec_expect_ok(env, db, prompts[i]) != 0) {
             sqlite3_close(db);
             return 1;
         }
 
-        if (exec_expect_ok(env, db, "SELECT llm_context_usage();") != 0) {
+        if (exec_expect_ok(env, db, "SELECT llm_context_used() AS context_used, llm_context_size() AS context_size, CAST(llm_context_used() AS FLOAT)/CAST(llm_context_size() AS FLOAT) || '%' AS 'context_usage_percentage';") != 0) {
             sqlite3_close(db);
             return 1;
         }
@@ -199,9 +244,57 @@ static int test_llm_chat_respond_repeated(const test_env *env) {
     return 0;
 }
 
+static int test_llm_chat_vtab(const test_env *env) {
+    sqlite3 *db = NULL;
+    if (open_db_and_load(env, &db) != SQLITE_OK) {
+        return 1;
+    }
+
+    const char *model = env->model_path ? env->model_path : DEFAULT_MODEL_PATH;
+    char sqlbuf[512];
+    snprintf(sqlbuf, sizeof(sqlbuf), "SELECT llm_model_load('%s');", model);
+    if (exec_expect_ok(env, db, sqlbuf) != 0) goto fail;
+    if (exec_expect_ok(env, db, "SELECT llm_context_create('context_size=1000');") != 0) goto fail;
+    if (exec_expect_ok(env, db, "SELECT llm_chat_create();") != 0) goto fail;
+    int rows = 0;
+    if (exec_select_rows(env, db, "SELECT * FROM llm_chat('Hi');", &rows) != 0) goto fail;
+    if (rows <= 0) {
+        fprintf(stderr, "[chat_vtab] expected rows but got %d\n", rows);
+        goto fail;
+    }
+    rows = 0;
+    if (exec_select_rows(env, db, "SELECT * FROM llm_chat('How are you');", &rows) != 0) goto fail;
+    if (rows <= 0) {
+        fprintf(stderr, "[chat_vtab] expected rows but got %d\n", rows);
+        goto fail;
+    }
+    if (exec_expect_ok(env, db, "SELECT llm_chat_free();") != 0) goto fail;
+    if (exec_expect_ok(env, db, "SELECT llm_context_free();") != 0) goto fail;
+    if (exec_expect_ok(env, db, "SELECT llm_model_free();") != 0) goto fail;
+    sqlite3_close(db);
+
+    sqlite3_int64 current = 0;
+    sqlite3_int64 highwater = 0;
+    if (sqlite3_status64(SQLITE_STATUS_MEMORY_USED, &current, &highwater, 0) != SQLITE_OK) {
+        fprintf(stderr, "[chat_vtab] sqlite3_status64 failed\n");
+        return 1;
+    }
+    if (current > 0 || highwater <= 0) {
+        fprintf(stderr, "[chat_vtab] invalid memory stats: current=%lld highwater=%lld\n",
+                (long long)current, (long long)highwater);
+        return 1;
+    }
+    return 0;
+
+fail:
+    if (db) sqlite3_close(db);
+    return 1;
+}
+
 static const test_case TESTS[] = {
     {"issue15_llm_chat_without_context", test_issue15_chat_without_context},
     {"llm_chat_respond_repeated", test_llm_chat_respond_repeated},
+    {"llm_chat_vtab", test_llm_chat_vtab},
 };
 
 int main(int argc, char **argv) {