From a4ed8d07a512e31f35d423941ab17f2e96398d2f Mon Sep 17 00:00:00 2001
From: PThorpe92 <preston@unlockedlabs.org>
Date: Sat, 14 Dec 2024 18:55:10 -0500
Subject: [PATCH] Temporary tests for POC on in-memory IO impl

---
 Makefile                |   1 +
 testing/memory-repl.tcl | 114 ++++++++++++++++++++++++++++++++++++++++
 2 files changed, 115 insertions(+)
 create mode 100755 testing/memory-repl.tcl

diff --git a/Makefile b/Makefile
index 33f9bd5e9..56de221f1 100644
--- a/Makefile
+++ b/Makefile
@@ -57,6 +57,7 @@ test: limbo test-compat test-sqlite3
 
 test-compat:
 	SQLITE_EXEC=$(SQLITE_EXEC) ./testing/all.test
+	SQLITE_EXEC=$(SQLITE_EXEC) ./testing/memory-repl.tcl
 .PHONY: test-compat
 
 test-sqlite3:
diff --git a/testing/memory-repl.tcl b/testing/memory-repl.tcl
new file mode 100755
index 000000000..3b705ed44
--- /dev/null
+++ b/testing/memory-repl.tcl
@@ -0,0 +1,114 @@
+#!/usr/bin/env tclsh
+
+set sqlite_exec [expr {[info exists env(SQLITE_EXEC)] ? $env(SQLITE_EXEC) : "sqlite3"}]
+
+proc start_sqlite_repl {sqlite_exec init_commands} {
+    set command [list $sqlite_exec ":memory:"]
+    set pipe [open "|[join $command]" RDWR]
+    puts $pipe $init_commands
+    flush $pipe
+    return $pipe
+}
+
+proc execute_sql {pipe sql} {
+    puts $pipe $sql
+    flush $pipe
+    puts $pipe "SELECT 'END_OF_RESULT';"
+    flush $pipe
+    set output ""
+    while {[gets $pipe line] >= 0} {
+        if {$line eq "END_OF_RESULT"} {
+            break
+        }
+        append output "$line\n"
+    }
+    return [string trim $output]
+}
+
+proc run_test {pipe sql expected_output} {
+    set actual_output [execute_sql $pipe $sql]
+    if {$actual_output ne $expected_output} {
+        puts "Test FAILED: '$sql'"
+        puts "returned '$actual_output'"
+        puts "expected '$expected_output'"
+        exit 1
+    }
+}
+
+proc do_execsql_test {pipe test_name sql expected_output} {
+    puts "Running test: $test_name"
+    run_test $pipe $sql $expected_output
+}
+
+set init_commands {
+CREATE TABLE users (
+        id INTEGER PRIMARY KEY,
+		first_name TEXT,
+		last_name TEXT,
+		age INTEGER
+	);
+    CREATE TABLE products (
+	id INTEGER PRIMARY KEY,
+	name TEXT,
+	price INTEGER
+	);
+    INSERT INTO users (id, first_name, last_name, age) VALUES
+	(1, 'Alice', 'Smith', 30), (2, 'Bob', 'Johnson', 25), (3, 'Charlie', 'Brown', 66), (4, 'David', 'Nichols', 70); INSERT INTO products (id, name, price) VALUES (1, 'Hat', 19.99), (2, 'Shirt', 29.99), (3, 'Shorts', 39.99), (4, 'Dress', 49.99);
+    CREATE TABLE t(x1, x2, x3, x4);
+    INSERT INTO t VALUES (zeroblob(1024 * 1024 - 1), zeroblob(1024 * 1024 - 2), zeroblob(1024 * 1024 - 3), zeroblob(1024 * 1024 - 4));
+}
+
+set pipe [start_sqlite_repl $sqlite_exec $init_commands]
+
+do_execsql_test $pipe schema {
+  .schema 
+} {CREATE TABLE users (id INTEGER PRIMARY KEY, first_name TEXT, last_name TEXT, age INTEGER);
+CREATE TABLE products (id INTEGER PRIMARY KEY, name TEXT, price INTEGER);
+CREATE TABLE t (x1, x2, x3, x4);}
+
+do_execsql_test $pipe "select-avg" {
+  SELECT avg(age) FROM users;
+} {47.75}
+
+do_execsql_test $pipe select-avg-text {
+  SELECT avg(first_name) FROM users;
+} {0.0}
+
+do_execsql_test $pipe select-sum {
+  SELECT sum(age) FROM users;
+} {191}
+
+do_execsql_test $pipe select-sum-text {
+  SELECT sum(first_name) FROM users;
+} {0.0}
+
+do_execsql_test $pipe select-total {
+  SELECT total(age) FROM users;
+} {191.0}
+
+do_execsql_test $pipe select-total-text {
+  SELECT total(first_name) FROM users WHERE id < 3;
+} {0.0}
+
+do_execsql_test $pipe select-limit {
+  SELECT typeof(id) FROM users LIMIT 1;
+} {integer}
+
+do_execsql_test $pipe select-count {
+  SELECT count(id) FROM users;
+} {4}
+
+do_execsql_test $pipe select-count {
+  SELECT count(*) FROM users;
+} {4}
+
+do_execsql_test $pipe select-count-constant-true {
+  SELECT count(*) FROM users WHERE true;
+} {4}
+
+do_execsql_test $pipe select-count-constant-false {
+  SELECT count(*) FROM users WHERE false;
+} {0}
+
+puts "All tests passed successfully."
+close $pipe