dtoro
5e10437fe3
Phase 4 — Performance: - F1: SSH DialPool with key-by-host pooling and 5min idle TTL - F2: In-memory entity lookup cache (TTL 60s, HTTP resolveEntityID) - F3: Trigram GIN indexes on entities.slug and entities.name (migration 031) - F4: Partial index on executions(classification_id) for auto-act (migration 032) - Added missing RunOutput and RunStreaming in actuator/ (E3 gap fill) Phase 6 — Infrastructure: - H1: Infisical image pinned to v0.99.1 - H2: execworker daemon — polls pending executions with per-execution advisory locks, recovers orphaned running executions, wired as docker-compose service - H3: splitSQL hardened with block comment and string-literal support, 6 new edge-case tests (11 total) - H4: Scheduler acquires pg_try_advisory_lock(0x01c05e6) at startup
108 lines
2.7 KiB
Go
108 lines
2.7 KiB
Go
package db
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func nonEmpty(stmts []string) []string {
|
|
var out []string
|
|
for _, s := range stmts {
|
|
if strings.TrimSpace(s) != "" {
|
|
out = append(out, s)
|
|
}
|
|
}
|
|
return out
|
|
}
|
|
|
|
func TestSplitSQLBasic(t *testing.T) {
|
|
stmts := nonEmpty(splitSQL("CREATE TABLE a (id int); CREATE TABLE b (id int);"))
|
|
if len(stmts) != 2 {
|
|
t.Fatalf("got %d statements, want 2: %#v", len(stmts), stmts)
|
|
}
|
|
}
|
|
|
|
func TestSplitSQLDollarQuotedFunction(t *testing.T) {
|
|
sql := `CREATE FUNCTION f() RETURNS int AS $$
|
|
SELECT 1; SELECT 2;
|
|
$$ LANGUAGE sql;
|
|
CREATE TABLE t (id int);`
|
|
stmts := nonEmpty(splitSQL(sql))
|
|
if len(stmts) != 2 {
|
|
t.Fatalf("got %d statements, want 2: %#v", len(stmts), stmts)
|
|
}
|
|
if !strings.Contains(stmts[0], "SELECT 1; SELECT 2;") {
|
|
t.Errorf("dollar-quoted body was split: %q", stmts[0])
|
|
}
|
|
}
|
|
|
|
func TestSplitSQLTaggedDollarQuote(t *testing.T) {
|
|
sql := `DO $body$ BEGIN PERFORM 1; END $body$;SELECT 1;`
|
|
stmts := nonEmpty(splitSQL(sql))
|
|
if len(stmts) != 2 {
|
|
t.Fatalf("got %d statements, want 2: %#v", len(stmts), stmts)
|
|
}
|
|
}
|
|
|
|
func TestSplitSQLSemicolonInComment(t *testing.T) {
|
|
sql := "-- comment with ; semicolon\nCREATE TABLE t (id int); -- trailing; note\nSELECT 1;"
|
|
stmts := nonEmpty(splitSQL(sql))
|
|
if len(stmts) != 2 {
|
|
t.Fatalf("got %d statements, want 2: %#v", len(stmts), stmts)
|
|
}
|
|
}
|
|
|
|
func TestSplitSQLSemicolonInStringLiteral(t *testing.T) {
|
|
sql := `SELECT 'hello; world'; INSERT INTO t VALUES (1);`
|
|
stmts := nonEmpty(splitSQL(sql))
|
|
if len(stmts) != 2 {
|
|
t.Fatalf("got %d statements, want 2: %#v", len(stmts), stmts)
|
|
}
|
|
}
|
|
|
|
func TestSplitSQLDollarSignInStringLiteral(t *testing.T) {
|
|
sql := `SELECT '$100'; SELECT 2;`
|
|
stmts := nonEmpty(splitSQL(sql))
|
|
if len(stmts) != 2 {
|
|
t.Fatalf("got %d statements, want 2: %#v", len(stmts), stmts)
|
|
}
|
|
}
|
|
|
|
func TestSplitSQLBlockComment(t *testing.T) {
|
|
sql := `SELECT 1; /* block; with; semicolons */ SELECT 2;`
|
|
stmts := nonEmpty(splitSQL(sql))
|
|
if len(stmts) != 2 {
|
|
t.Fatalf("got %d statements, want 2: %#v", len(stmts), stmts)
|
|
}
|
|
}
|
|
|
|
func TestSplitSQLBlockCommentWithDollarQuote(t *testing.T) {
|
|
sql := `/* $$ not a dollar quote */ SELECT 1;`
|
|
stmts := nonEmpty(splitSQL(sql))
|
|
if len(stmts) != 1 {
|
|
t.Fatalf("got %d statements, want 1: %#v", len(stmts), stmts)
|
|
}
|
|
}
|
|
|
|
func TestSplitSQLDoubledQuoteInString(t *testing.T) {
|
|
sql := `SELECT 'O''Brien'; SELECT 2;`
|
|
stmts := nonEmpty(splitSQL(sql))
|
|
if len(stmts) != 2 {
|
|
t.Fatalf("got %d statements, want 2: %#v", len(stmts), stmts)
|
|
}
|
|
}
|
|
|
|
func TestSplitSQLEmptyInput(t *testing.T) {
|
|
stmts := nonEmpty(splitSQL(""))
|
|
if len(stmts) != 0 {
|
|
t.Fatalf("got %d statements, want 0", len(stmts))
|
|
}
|
|
}
|
|
|
|
func TestSplitSQLNoSemicolon(t *testing.T) {
|
|
stmts := nonEmpty(splitSQL("SELECT 1"))
|
|
if len(stmts) != 1 {
|
|
t.Fatalf("got %d statements, want 1", len(stmts))
|
|
}
|
|
}
|