fix(db): PostgreSQL P0 fixes - SQL_KEYWORDS expansion, timeout config, and tests
- Expand SQL_KEYWORDS from ~120 to 226+ words covering: - Window functions (ROW_NUMBER, RANK, LAG, LEAD, etc.) - CTEs (WITH, RECURSIVE, MATERIALIZED, etc.) - Advanced grouping (ROLLUP, CUBE, GROUPING SETS) - JSON operations, types, table sampling - Transaction control and other PostgreSQL-specific keywords - Add connection pool timeout configuration: - connectionTimeoutMillis: 10s - statement_timeout: 30s (PostgreSQL level) - idleTimeoutMillis: 30s (connection cleanup) - query_timeout: 60s (driver-level fallback) - Add 12 comprehensive edge case tests covering: - Window functions, CTEs, advanced grouping - CASE expressions, set operations, JSON operators - All 38 tests pass Production-ready: prevents hung queries and supports complex SQL.
This commit is contained in:
@@ -15,9 +15,20 @@ export type { PostgreSqlConfig } from '../../types/database.types'
|
|||||||
* SQL keywords that should NOT be double-quoted during identifier preprocessing.
|
* SQL keywords that should NOT be double-quoted during identifier preprocessing.
|
||||||
* PostgreSQL lowercases unquoted identifiers, but SSMA-migrated databases
|
* PostgreSQL lowercases unquoted identifiers, but SSMA-migrated databases
|
||||||
* have uppercase column names that require double-quoting to preserve case.
|
* have uppercase column names that require double-quoting to preserve case.
|
||||||
|
*
|
||||||
|
* This list covers PostgreSQL reserved words across multiple categories:
|
||||||
|
* - DML (Data Manipulation Language)
|
||||||
|
* - DDL (Data Definition Language)
|
||||||
|
* - Window functions
|
||||||
|
* - CTEs (Common Table Expressions)
|
||||||
|
* - Advanced GROUP BY clauses
|
||||||
|
* - JSON operations
|
||||||
|
* - Type system
|
||||||
|
* - Table sampling
|
||||||
|
* - Transaction control
|
||||||
*/
|
*/
|
||||||
const SQL_KEYWORDS = new Set([
|
const SQL_KEYWORDS = new Set([
|
||||||
// DML
|
// ==================== DML (Data Manipulation Language) ====================
|
||||||
'SELECT',
|
'SELECT',
|
||||||
'FROM',
|
'FROM',
|
||||||
'WHERE',
|
'WHERE',
|
||||||
@@ -33,7 +44,8 @@ const SQL_KEYWORDS = new Set([
|
|||||||
'UPDATE',
|
'UPDATE',
|
||||||
'SET',
|
'SET',
|
||||||
'DELETE',
|
'DELETE',
|
||||||
// Ordering & limiting
|
|
||||||
|
// ==================== Ordering & Limiting ====================
|
||||||
'ORDER',
|
'ORDER',
|
||||||
'BY',
|
'BY',
|
||||||
'ASC',
|
'ASC',
|
||||||
@@ -44,7 +56,8 @@ const SQL_KEYWORDS = new Set([
|
|||||||
'NEXT',
|
'NEXT',
|
||||||
'ROWS',
|
'ROWS',
|
||||||
'ONLY',
|
'ONLY',
|
||||||
// Joins
|
|
||||||
|
// ==================== Joins ====================
|
||||||
'JOIN',
|
'JOIN',
|
||||||
'LEFT',
|
'LEFT',
|
||||||
'RIGHT',
|
'RIGHT',
|
||||||
@@ -53,16 +66,67 @@ const SQL_KEYWORDS = new Set([
|
|||||||
'CROSS',
|
'CROSS',
|
||||||
'FULL',
|
'FULL',
|
||||||
'ON',
|
'ON',
|
||||||
// Set operations
|
'NATURAL',
|
||||||
|
'LATERAL',
|
||||||
|
|
||||||
|
// ==================== Set Operations ====================
|
||||||
'UNION',
|
'UNION',
|
||||||
'ALL',
|
'ALL',
|
||||||
'INTERSECT',
|
'INTERSECT',
|
||||||
'EXCEPT',
|
'EXCEPT',
|
||||||
// Grouping
|
|
||||||
|
// ==================== Grouping & Aggregation ====================
|
||||||
'GROUP',
|
'GROUP',
|
||||||
'HAVING',
|
'HAVING',
|
||||||
'DISTINCT',
|
'DISTINCT',
|
||||||
// DDL
|
'GROUPING',
|
||||||
|
'SETS',
|
||||||
|
'ROLLUP',
|
||||||
|
'CUBE',
|
||||||
|
'FILTER',
|
||||||
|
'WITHIN',
|
||||||
|
|
||||||
|
// ==================== Window Functions ====================
|
||||||
|
'OVER',
|
||||||
|
'PARTITION',
|
||||||
|
'WINDOW',
|
||||||
|
'RANGE',
|
||||||
|
'UNBOUNDED',
|
||||||
|
'PRECEDING',
|
||||||
|
'FOLLOWING',
|
||||||
|
'CURRENT',
|
||||||
|
'ROW',
|
||||||
|
'GROUPS',
|
||||||
|
'EXCLUDE',
|
||||||
|
'TIES',
|
||||||
|
'RANK',
|
||||||
|
'DENSE_RANK',
|
||||||
|
'ROW_NUMBER',
|
||||||
|
'NTILE',
|
||||||
|
'LAG',
|
||||||
|
'LEAD',
|
||||||
|
'FIRST_VALUE',
|
||||||
|
'LAST_VALUE',
|
||||||
|
'NTH_VALUE',
|
||||||
|
|
||||||
|
// ==================== CTE (Common Table Expressions) ====================
|
||||||
|
'WITH',
|
||||||
|
'RECURSIVE',
|
||||||
|
'MATERIALIZED',
|
||||||
|
'SEARCH',
|
||||||
|
'CYCLE',
|
||||||
|
'PATH',
|
||||||
|
'ROOT',
|
||||||
|
'SIBLINGS',
|
||||||
|
|
||||||
|
// ==================== CASE Expressions ====================
|
||||||
|
'CASE',
|
||||||
|
'WHEN',
|
||||||
|
'THEN',
|
||||||
|
'ELSE',
|
||||||
|
'END',
|
||||||
|
|
||||||
|
// ==================== DDL (Data Definition Language) ====================
|
||||||
'CREATE',
|
'CREATE',
|
||||||
'ALTER',
|
'ALTER',
|
||||||
'DROP',
|
'DROP',
|
||||||
@@ -73,7 +137,15 @@ const SQL_KEYWORDS = new Set([
|
|||||||
'MODIFY',
|
'MODIFY',
|
||||||
'RENAME',
|
'RENAME',
|
||||||
'TO',
|
'TO',
|
||||||
// PostgreSQL specific
|
'GENERATED',
|
||||||
|
'ALWAYS',
|
||||||
|
'IDENTITY',
|
||||||
|
'INCLUDE',
|
||||||
|
'TEMP',
|
||||||
|
'TEMPORARY',
|
||||||
|
'UNLOGGED',
|
||||||
|
|
||||||
|
// ==================== PostgreSQL Specific - UPSERT/MERGE ====================
|
||||||
'CONFLICT',
|
'CONFLICT',
|
||||||
'DO',
|
'DO',
|
||||||
'NOTHING',
|
'NOTHING',
|
||||||
@@ -82,32 +154,76 @@ const SQL_KEYWORDS = new Set([
|
|||||||
'MERGE',
|
'MERGE',
|
||||||
'USING',
|
'USING',
|
||||||
'MATCHED',
|
'MATCHED',
|
||||||
'WHEN',
|
|
||||||
'THEN',
|
|
||||||
'ELSE',
|
|
||||||
'END',
|
|
||||||
'TARGET',
|
'TARGET',
|
||||||
'SOURCE',
|
'SOURCE',
|
||||||
// Functions
|
|
||||||
|
// ==================== Aggregate Functions ====================
|
||||||
'COUNT',
|
'COUNT',
|
||||||
'SUM',
|
'SUM',
|
||||||
'AVG',
|
'AVG',
|
||||||
'MIN',
|
'MIN',
|
||||||
'MAX',
|
'MAX',
|
||||||
'EXISTS',
|
'EXISTS',
|
||||||
'CURRENT_TIMESTAMP',
|
|
||||||
'NOW',
|
|
||||||
'GETDATE',
|
|
||||||
'COALESCE',
|
'COALESCE',
|
||||||
'NULLIF',
|
'NULLIF',
|
||||||
'CAST',
|
'CAST',
|
||||||
'AS',
|
'AS',
|
||||||
// Transaction
|
|
||||||
|
// ==================== JSON Operations ====================
|
||||||
|
'JSON',
|
||||||
|
'JSONB',
|
||||||
|
'JSON_ARRAY',
|
||||||
|
'JSON_OBJECT',
|
||||||
|
'JSON_AGG',
|
||||||
|
'JSONB_AGG',
|
||||||
|
'JSONB_OBJECT_AGG',
|
||||||
|
|
||||||
|
// ==================== Types & Casting ====================
|
||||||
|
'DECIMAL',
|
||||||
|
'NUMERIC',
|
||||||
|
'BOOLEAN',
|
||||||
|
'CHARACTER',
|
||||||
|
'VARYING',
|
||||||
|
'PRECISION',
|
||||||
|
'REAL',
|
||||||
|
'DOUBLE',
|
||||||
|
'FLOAT',
|
||||||
|
'TEXT',
|
||||||
|
'INTEGER',
|
||||||
|
'SERIAL',
|
||||||
|
'BIGINT',
|
||||||
|
'SMALLINT',
|
||||||
|
'DATE',
|
||||||
|
'TIME',
|
||||||
|
'TIMESTAMP',
|
||||||
|
'TIMESTAMPTZ',
|
||||||
|
'TIMEZONE',
|
||||||
|
'INTERVAL',
|
||||||
|
'BIGSERIAL',
|
||||||
|
'SMALLSERIAL',
|
||||||
|
|
||||||
|
// ==================== Table Sampling ====================
|
||||||
|
'TABLESAMPLE',
|
||||||
|
'BERNOULLI',
|
||||||
|
'SYSTEM',
|
||||||
|
'REPEATABLE',
|
||||||
|
'SEED',
|
||||||
|
|
||||||
|
// ==================== Transaction Control ====================
|
||||||
'BEGIN',
|
'BEGIN',
|
||||||
'COMMIT',
|
'COMMIT',
|
||||||
'ROLLBACK',
|
'ROLLBACK',
|
||||||
'SAVEPOINT',
|
'SAVEPOINT',
|
||||||
// Types & values
|
'WORK',
|
||||||
|
'ISOLATION',
|
||||||
|
'LEVEL',
|
||||||
|
'READ',
|
||||||
|
'WRITE',
|
||||||
|
'COMMITTED',
|
||||||
|
'REPEATABLE',
|
||||||
|
'SERIALIZABLE',
|
||||||
|
|
||||||
|
// ==================== Types & Values ====================
|
||||||
'TRUE',
|
'TRUE',
|
||||||
'FALSE',
|
'FALSE',
|
||||||
'DEFAULT',
|
'DEFAULT',
|
||||||
@@ -118,23 +234,65 @@ const SQL_KEYWORDS = new Set([
|
|||||||
'CONSTRAINT',
|
'CONSTRAINT',
|
||||||
'UNIQUE',
|
'UNIQUE',
|
||||||
'CHECK',
|
'CHECK',
|
||||||
'CASE',
|
'NULLS',
|
||||||
|
'FIRST',
|
||||||
|
'LAST',
|
||||||
|
|
||||||
|
// ==================== Pattern Matching ====================
|
||||||
'BETWEEN',
|
'BETWEEN',
|
||||||
'LIKE',
|
'LIKE',
|
||||||
'ILIKE',
|
'ILIKE',
|
||||||
|
'SIMILAR',
|
||||||
|
'ESCAPE',
|
||||||
'ANY',
|
'ANY',
|
||||||
'SOME',
|
'SOME',
|
||||||
// Common
|
|
||||||
'IF',
|
// ==================== Functions & Procedures ====================
|
||||||
'WITH',
|
|
||||||
'RECURSIVE',
|
|
||||||
'OVER',
|
|
||||||
'PARTITION',
|
|
||||||
'WINDOW',
|
|
||||||
'ROW',
|
|
||||||
'FIRST',
|
|
||||||
'AFTER',
|
'AFTER',
|
||||||
'BEFORE'
|
'BEFORE',
|
||||||
|
'EACH',
|
||||||
|
'STATEMENT',
|
||||||
|
'TRIGGER',
|
||||||
|
'FUNCTION',
|
||||||
|
'PROCEDURE',
|
||||||
|
'LANGUAGE',
|
||||||
|
'SQL',
|
||||||
|
'PLPGSQL',
|
||||||
|
'RETURNS',
|
||||||
|
'CALLED',
|
||||||
|
'STRICT',
|
||||||
|
'SECURITY',
|
||||||
|
'INVOKER',
|
||||||
|
'DEFINER',
|
||||||
|
'VOLATILE',
|
||||||
|
'STABLE',
|
||||||
|
'IMMUTABLE',
|
||||||
|
'PARALLEL',
|
||||||
|
'SAFE',
|
||||||
|
'RESTRICTED',
|
||||||
|
'UNSAFE',
|
||||||
|
|
||||||
|
// ==================== Utility Commands ====================
|
||||||
|
'CONCURRENTLY',
|
||||||
|
'REINDEX',
|
||||||
|
'VACUUM',
|
||||||
|
'ANALYZE',
|
||||||
|
'EXPLAIN',
|
||||||
|
'LOCAL',
|
||||||
|
'GLOBAL',
|
||||||
|
'ORDINALITY',
|
||||||
|
'FREEZE',
|
||||||
|
'VERBOSE',
|
||||||
|
'BUFFERS',
|
||||||
|
'FORMAT',
|
||||||
|
'XML',
|
||||||
|
'YAML',
|
||||||
|
|
||||||
|
// ==================== Additional Reserved Words ====================
|
||||||
|
'IF',
|
||||||
|
'CURRENT_TIMESTAMP',
|
||||||
|
'NOW',
|
||||||
|
'GETDATE'
|
||||||
])
|
])
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@@ -286,7 +444,31 @@ export class PostgreSqlService implements IDatabaseService {
|
|||||||
user: this.config.user,
|
user: this.config.user,
|
||||||
password: this.config.password,
|
password: this.config.password,
|
||||||
database: this.config.database,
|
database: this.config.database,
|
||||||
max: this.config.maxPoolSize ?? 10
|
max: this.config.maxPoolSize ?? 10,
|
||||||
|
/**
|
||||||
|
* Connection timeout in milliseconds.
|
||||||
|
* Time to wait when connecting to PostgreSQL before failing.
|
||||||
|
* Prevents hanging during network issues or server overload.
|
||||||
|
*/
|
||||||
|
connectionTimeoutMillis: 10000,
|
||||||
|
/**
|
||||||
|
* PostgreSQL statement timeout in milliseconds.
|
||||||
|
* Limits execution time for individual SQL statements.
|
||||||
|
* Prevents long-running queries from blocking the connection pool.
|
||||||
|
*/
|
||||||
|
statement_timeout: 30000,
|
||||||
|
/**
|
||||||
|
* Idle connection timeout in milliseconds.
|
||||||
|
* Closes connections that have been idle for this duration.
|
||||||
|
* Frees up pool resources and prevents stale connections.
|
||||||
|
*/
|
||||||
|
idleTimeoutMillis: 30000,
|
||||||
|
/**
|
||||||
|
* Query timeout in milliseconds (pg driver level).
|
||||||
|
* Fallback protection to abort queries that exceed this duration.
|
||||||
|
* Should be longer than statement_timeout to allow PG to handle first.
|
||||||
|
*/
|
||||||
|
query_timeout: 60000
|
||||||
})
|
})
|
||||||
|
|
||||||
// Test connection
|
// Test connection
|
||||||
|
|||||||
@@ -206,4 +206,225 @@ describe('prepareSql', () => {
|
|||||||
expect(result).toContain('as count')
|
expect(result).toContain('as count')
|
||||||
expect(result).toContain('"UserName"')
|
expect(result).toContain('"UserName"')
|
||||||
})
|
})
|
||||||
|
|
||||||
|
// ==================== Window Functions ====================
|
||||||
|
|
||||||
|
it('should handle ROW_NUMBER() OVER (PARTITION BY ... ORDER BY ...)', () => {
|
||||||
|
const sql = `
|
||||||
|
SELECT UserName, ROW_NUMBER() OVER (PARTITION BY UserType ORDER BY CreatedAt DESC) as rn
|
||||||
|
FROM "dbo"."BIPUsers"
|
||||||
|
`
|
||||||
|
const result = prepareSql(sql)
|
||||||
|
expect(result).toContain('"UserName"')
|
||||||
|
expect(result).toContain('"UserType"')
|
||||||
|
expect(result).toContain('"CreatedAt"')
|
||||||
|
expect(result).not.toContain('"ROW_NUMBER"')
|
||||||
|
expect(result).not.toContain('"OVER"')
|
||||||
|
expect(result).not.toContain('"PARTITION"')
|
||||||
|
expect(result).not.toContain('"ORDER"')
|
||||||
|
})
|
||||||
|
|
||||||
|
it('should handle RANK() and DENSE_RANK()', () => {
|
||||||
|
const sql = `
|
||||||
|
SELECT MaterialCode, RANK() OVER (ORDER BY Quantity DESC) as rnk, DENSE_RANK() OVER (ORDER BY Quantity DESC) as drnk
|
||||||
|
FROM "dbo"."Materials"
|
||||||
|
`
|
||||||
|
const result = prepareSql(sql)
|
||||||
|
expect(result).toContain('"MaterialCode"')
|
||||||
|
expect(result).toContain('"Quantity"')
|
||||||
|
expect(result).not.toContain('"RANK"')
|
||||||
|
expect(result).not.toContain('"DENSE_RANK"')
|
||||||
|
})
|
||||||
|
|
||||||
|
it('should handle LAG() and LEAD()', () => {
|
||||||
|
const sql = `
|
||||||
|
SELECT OrderId, LAG(TotalAmount, 1) OVER (ORDER BY OrderDate) as prevAmount, LEAD(TotalAmount, 1) OVER (ORDER BY OrderDate) as nextAmount
|
||||||
|
FROM "dbo"."Orders"
|
||||||
|
`
|
||||||
|
const result = prepareSql(sql)
|
||||||
|
expect(result).toContain('"OrderId"')
|
||||||
|
expect(result).toContain('"TotalAmount"')
|
||||||
|
expect(result).toContain('"OrderDate"')
|
||||||
|
expect(result).not.toContain('"LAG"')
|
||||||
|
expect(result).not.toContain('"LEAD"')
|
||||||
|
})
|
||||||
|
|
||||||
|
// ==================== CTEs (Common Table Expressions) ====================
|
||||||
|
|
||||||
|
it('should handle WITH clause', () => {
|
||||||
|
const sql = `
|
||||||
|
WITH UserSummary AS (
|
||||||
|
SELECT UserId, COUNT(OrderId) as OrderCount
|
||||||
|
FROM "dbo"."Orders"
|
||||||
|
GROUP BY UserId
|
||||||
|
)
|
||||||
|
SELECT UserName, OrderCount
|
||||||
|
FROM UserSummary
|
||||||
|
JOIN "dbo"."BIPUsers" ON UserSummary.UserId = "dbo"."BIPUsers".ID
|
||||||
|
`
|
||||||
|
const result = prepareSql(sql)
|
||||||
|
expect(result).toContain('"UserId"')
|
||||||
|
expect(result).toContain('"OrderId"')
|
||||||
|
expect(result).toContain('"UserName"')
|
||||||
|
expect(result).not.toContain('"WITH"')
|
||||||
|
expect(result).not.toContain('"AS"')
|
||||||
|
expect(result).not.toContain('"FROM"')
|
||||||
|
expect(result).not.toContain('"JOIN"')
|
||||||
|
expect(result).not.toContain('"ON"')
|
||||||
|
})
|
||||||
|
|
||||||
|
it('should handle recursive CTE', () => {
|
||||||
|
const sql = `
|
||||||
|
WITH RECURSIVE CategoryTree AS (
|
||||||
|
SELECT CategoryId, ParentCategoryId, CategoryName, 0 as Level
|
||||||
|
FROM "dbo"."Categories"
|
||||||
|
WHERE ParentCategoryId IS NULL
|
||||||
|
UNION ALL
|
||||||
|
SELECT c.CategoryId, c.ParentCategoryId, c.CategoryName, ct.Level + 1
|
||||||
|
FROM "dbo"."Categories" c
|
||||||
|
INNER JOIN CategoryTree ct ON c.ParentCategoryId = ct.CategoryId
|
||||||
|
)
|
||||||
|
SELECT * FROM CategoryTree
|
||||||
|
`
|
||||||
|
const result = prepareSql(sql)
|
||||||
|
expect(result).toContain('"CategoryId"')
|
||||||
|
expect(result).toContain('"ParentCategoryId"')
|
||||||
|
expect(result).toContain('"CategoryName"')
|
||||||
|
expect(result).not.toContain('"WITH"')
|
||||||
|
expect(result).not.toContain('"RECURSIVE"')
|
||||||
|
expect(result).not.toContain('"UNION"')
|
||||||
|
expect(result).not.toContain('"ALL"')
|
||||||
|
expect(result).not.toContain('"INNER"')
|
||||||
|
expect(result).not.toContain('"JOIN"')
|
||||||
|
})
|
||||||
|
|
||||||
|
// ==================== Advanced Grouping ====================
|
||||||
|
|
||||||
|
it('should handle ROLLUP', () => {
|
||||||
|
const sql = `
|
||||||
|
SELECT DepartmentId, JobTitle, COUNT(*) as EmployeeCount
|
||||||
|
FROM "dbo"."Employees"
|
||||||
|
GROUP BY ROLLUP (DepartmentId, JobTitle)
|
||||||
|
`
|
||||||
|
const result = prepareSql(sql)
|
||||||
|
expect(result).toContain('"DepartmentId"')
|
||||||
|
expect(result).toContain('"JobTitle"')
|
||||||
|
expect(result).not.toContain('"GROUP"')
|
||||||
|
expect(result).not.toContain('"BY"')
|
||||||
|
expect(result).not.toContain('"ROLLUP"')
|
||||||
|
})
|
||||||
|
|
||||||
|
it('should handle CUBE', () => {
|
||||||
|
const sql = `
|
||||||
|
SELECT Year, Quarter, Region, SUM(SalesAmount) as TotalSales
|
||||||
|
FROM "dbo"."Sales"
|
||||||
|
GROUP BY CUBE (Year, Quarter, Region)
|
||||||
|
`
|
||||||
|
const result = prepareSql(sql)
|
||||||
|
expect(result).toContain('"Year"')
|
||||||
|
expect(result).toContain('"Quarter"')
|
||||||
|
expect(result).toContain('"Region"')
|
||||||
|
expect(result).toContain('"SalesAmount"')
|
||||||
|
expect(result).not.toContain('"CUBE"')
|
||||||
|
expect(result).not.toContain('"GROUP"')
|
||||||
|
expect(result).not.toContain('"BY"')
|
||||||
|
})
|
||||||
|
|
||||||
|
it('should handle GROUPING SETS', () => {
|
||||||
|
const sql = `
|
||||||
|
SELECT DepartmentId, JobTitle, COUNT(*) as EmployeeCount
|
||||||
|
FROM "dbo"."Employees"
|
||||||
|
GROUP BY GROUPING SETS ((DepartmentId, JobTitle), (DepartmentId), ())
|
||||||
|
`
|
||||||
|
const result = prepareSql(sql)
|
||||||
|
expect(result).toContain('"DepartmentId"')
|
||||||
|
expect(result).toContain('"JobTitle"')
|
||||||
|
expect(result).not.toContain('"GROUPING"')
|
||||||
|
expect(result).not.toContain('"SETS"')
|
||||||
|
expect(result).not.toContain('"GROUP"')
|
||||||
|
expect(result).not.toContain('"BY"')
|
||||||
|
})
|
||||||
|
|
||||||
|
// ==================== CASE Expressions ====================
|
||||||
|
|
||||||
|
it('should handle simple CASE', () => {
|
||||||
|
const sql = `
|
||||||
|
SELECT UserName, CASE UserType
|
||||||
|
WHEN 'admin' THEN 'Administrator'
|
||||||
|
WHEN 'user' THEN 'Regular User'
|
||||||
|
ELSE 'Guest'
|
||||||
|
END as UserRole
|
||||||
|
FROM "dbo"."BIPUsers"
|
||||||
|
`
|
||||||
|
const result = prepareSql(sql)
|
||||||
|
expect(result).toContain('"UserName"')
|
||||||
|
expect(result).toContain('"UserType"')
|
||||||
|
expect(result).not.toContain('"CASE"')
|
||||||
|
expect(result).not.toContain('"WHEN"')
|
||||||
|
expect(result).not.toContain('"THEN"')
|
||||||
|
expect(result).not.toContain('"ELSE"')
|
||||||
|
expect(result).not.toContain('"END"')
|
||||||
|
})
|
||||||
|
|
||||||
|
it('should handle searched CASE', () => {
|
||||||
|
const sql = `
|
||||||
|
SELECT OrderId, TotalAmount,
|
||||||
|
CASE
|
||||||
|
WHEN TotalAmount > 10000 THEN 'Large'
|
||||||
|
WHEN TotalAmount > 1000 THEN 'Medium'
|
||||||
|
ELSE 'Small'
|
||||||
|
END as OrderSize
|
||||||
|
FROM "dbo"."Orders"
|
||||||
|
`
|
||||||
|
const result = prepareSql(sql)
|
||||||
|
expect(result).toContain('"OrderId"')
|
||||||
|
expect(result).toContain('"TotalAmount"')
|
||||||
|
expect(result).not.toContain('"CASE"')
|
||||||
|
expect(result).not.toContain('"WHEN"')
|
||||||
|
expect(result).not.toContain('"THEN"')
|
||||||
|
expect(result).not.toContain('"ELSE"')
|
||||||
|
expect(result).not.toContain('"END"')
|
||||||
|
})
|
||||||
|
|
||||||
|
// ==================== Set Operations ====================
|
||||||
|
|
||||||
|
it('should handle UNION, UNION ALL, INTERSECT, EXCEPT', () => {
|
||||||
|
const sql = `
|
||||||
|
SELECT UserId FROM "dbo"."ActiveUsers"
|
||||||
|
UNION
|
||||||
|
SELECT UserId FROM "dbo"."PremiumUsers"
|
||||||
|
UNION ALL
|
||||||
|
SELECT UserId FROM "dbo"."TrialUsers"
|
||||||
|
INTERSECT
|
||||||
|
SELECT UserId FROM "dbo"."VerifiedUsers"
|
||||||
|
EXCEPT
|
||||||
|
SELECT UserId FROM "dbo"."BannedUsers"
|
||||||
|
`
|
||||||
|
const result = prepareSql(sql)
|
||||||
|
expect(result).toContain('"UserId"')
|
||||||
|
expect(result).not.toContain('"UNION"')
|
||||||
|
expect(result).not.toContain('"ALL"')
|
||||||
|
expect(result).not.toContain('"INTERSECT"')
|
||||||
|
expect(result).not.toContain('"EXCEPT"')
|
||||||
|
expect(result).not.toContain('"SELECT"')
|
||||||
|
expect(result).not.toContain('"FROM"')
|
||||||
|
})
|
||||||
|
|
||||||
|
// ==================== JSON Operators ====================
|
||||||
|
|
||||||
|
it('should handle -> and ->> operators', () => {
|
||||||
|
const sql = `
|
||||||
|
SELECT UserId, ProfileData->'address'->>'city' as City, ProfileData->'contact'->>'phone' as Phone
|
||||||
|
FROM "dbo"."Users"
|
||||||
|
WHERE ProfileData->'preferences'->>'newsletter' = 'true'
|
||||||
|
`
|
||||||
|
const result = prepareSql(sql)
|
||||||
|
expect(result).toContain('"UserId"')
|
||||||
|
expect(result).toContain('"ProfileData"')
|
||||||
|
expect(result).toContain('->')
|
||||||
|
expect(result).toContain('->>')
|
||||||
|
expect(result).not.toContain('"SELECT"')
|
||||||
|
expect(result).not.toContain('"FROM"')
|
||||||
|
expect(result).not.toContain('"WHERE"')
|
||||||
|
})
|
||||||
})
|
})
|
||||||
|
|||||||
Reference in New Issue
Block a user