more fixtures

Author: pyramation

__fixtures__/generated/generated.json

@@ -5,6 +5,12 @@
   "pretty/select_statements-4.sql": "SELECT name FROM customers\nUNION ALL\nSELECT name FROM suppliers\nORDER BY name",
   "pretty/select_statements-5.sql": "SELECT name, email FROM users WHERE status = 'active'",
   "pretty/select_statements-6.sql": "SELECT u.name, o.total FROM users u, orders o WHERE u.id = o.user_id",
+  "pretty/misc-1.sql": "WITH recent_orders AS (\n  SELECT o.id, o.user_id, o.created_at\n  FROM orders o\n  WHERE o.created_at > NOW() - INTERVAL '30 days'\n), high_value_orders AS (\n  SELECT r.user_id, COUNT(*) AS order_count, SUM(oi.price * oi.quantity) AS total_spent\n  FROM recent_orders r\n  JOIN order_items oi ON r.id = oi.order_id\n  GROUP BY r.user_id\n)\nSELECT u.id, u.name, h.total_spent\nFROM users u\nJOIN high_value_orders h ON u.id = h.user_id\nWHERE h.total_spent > 1000\nORDER BY h.total_spent DESC",
+  "pretty/misc-2.sql": "SELECT\n  department,\n  employee_id,\n  COUNT(*) FILTER (WHERE status = 'active') OVER (PARTITION BY department) AS active_count,\n  RANK() OVER (PARTITION BY department ORDER BY salary DESC) AS salary_rank\nFROM employee_status\nGROUP BY GROUPING SETS ((department), (department, employee_id))",
+  "pretty/misc-3.sql": "SELECT u.id, u.name, j.key, j.value\nFROM users u,\nLATERAL jsonb_each_text(u.preferences) AS j(key, value)\nWHERE j.key LIKE 'notif_%' AND j.value::boolean = true",
+  "pretty/misc-4.sql": "SELECT p.id, p.title,\n  CASE\n    WHEN EXISTS (\n      SELECT 1 FROM reviews r\n      WHERE r.product_id = p.id AND r.rating >= 4\n    ) THEN 'Popular'\n    ELSE 'Unrated'\n  END AS status\nFROM products p\nWHERE p.archived = false",
+  "pretty/misc-5.sql": "WITH logs AS (\n  SELECT id, payload::json->>'event' AS event, (payload::json->>'ts')::timestamp AS ts\n  FROM event_log\n  WHERE ts > NOW() - INTERVAL '7 days'\n)\nSELECT event, COUNT(*) AS freq\nFROM (\n  SELECT DISTINCT event, ts::date AS event_day\n  FROM logs\n) d\nGROUP BY event\nORDER BY freq DESC",
+  "pretty/misc-6.sql": "SELECT \n  o.id AS order_id,\n  u.name AS user_name,\n  p.name AS product_name,\n  s.status,\n  sh.shipped_at,\n  r.refund_amount\nFROM orders o\nJOIN users u \n  ON o.user_id = u.id\nJOIN order_items oi \n  ON oi.order_id = o.id\nJOIN products p \n  ON (\n    (p.id = oi.product_id AND p.available = true)\n    OR \n    (p.sku = oi.product_sku AND (p.discontinued = false OR p.replacement_id IS NOT NULL))\n  )\nLEFT JOIN shipping sh \n  ON (\n    sh.order_id = o.id \n    AND (\n      (sh.carrier = 'UPS' AND sh.tracking_number IS NOT NULL)\n      OR \n      (sh.carrier = 'FedEx' AND sh.shipped_at > o.created_at + INTERVAL '1 day')\n    )\n  )\nLEFT JOIN statuses s \n  ON s.id = o.status_id \n  AND (\n    s.name != 'cancelled'\n    OR (s.name = 'cancelled' AND s.updated_at > NOW() - INTERVAL '7 days')\n  )\nLEFT JOIN refunds r \n  ON r.order_id = o.id \n  AND (\n    (r.status = 'approved' AND r.processed_at IS NOT NULL)\n    OR \n    (r.status = 'pending' AND r.requested_at < NOW() - INTERVAL '14 days')\n  )\nWHERE o.created_at > NOW() - INTERVAL '90 days'\n  AND u.active = true\n  AND (\n    s.status = 'shipped' \n    OR (\n      s.status = 'processing' \n      AND EXISTS (\n        SELECT 1 FROM order_notes n WHERE n.order_id = o.id AND n.note ILIKE '%expedite%'\n      )\n    )\n  )\nORDER BY o.created_at DESC",
   "pretty/create_table-1.sql": "CREATE TABLE users (\n  id SERIAL PRIMARY KEY,\n  name TEXT NOT NULL,\n  email TEXT UNIQUE\n)",
   "pretty/create_table-2.sql": "CREATE TABLE products (\n  id SERIAL PRIMARY KEY,\n  name VARCHAR(255) NOT NULL,\n  price DECIMAL(10,2) CHECK (price > 0),\n  category_id INTEGER,\n  description TEXT,\n  created_at TIMESTAMP DEFAULT now(),\n  updated_at TIMESTAMP,\n  UNIQUE (name, category_id),\n  FOREIGN KEY (category_id) REFERENCES categories(id)\n)",
   "pretty/create_table-3.sql": "CREATE TABLE orders (\n  id SERIAL PRIMARY KEY,\n  subtotal DECIMAL(10,2) NOT NULL,\n  tax_rate DECIMAL(5,4) DEFAULT 0.0825,\n  tax_amount DECIMAL(10,2) GENERATED ALWAYS AS (subtotal * tax_rate) STORED,\n  total DECIMAL(10,2) GENERATED ALWAYS AS (subtotal + tax_amount) STORED\n)",

__fixtures__/kitchen-sink/pretty/misc.sql

@@ -0,0 +1,112 @@
+-- 1. Deeply nested CTEs with joins and aggregation
+WITH recent_orders AS (
+  SELECT o.id, o.user_id, o.created_at
+  FROM orders o
+  WHERE o.created_at > NOW() - INTERVAL '30 days'
+), high_value_orders AS (
+  SELECT r.user_id, COUNT(*) AS order_count, SUM(oi.price * oi.quantity) AS total_spent
+  FROM recent_orders r
+  JOIN order_items oi ON r.id = oi.order_id
+  GROUP BY r.user_id
+)
+SELECT u.id, u.name, h.total_spent
+FROM users u
+JOIN high_value_orders h ON u.id = h.user_id
+WHERE h.total_spent > 1000
+ORDER BY h.total_spent DESC;
+
+-- 2. Complex SELECT with window functions, FILTER, and GROUPING SETS
+SELECT
+  department,
+  employee_id,
+  COUNT(*) FILTER (WHERE status = 'active') OVER (PARTITION BY department) AS active_count,
+  RANK() OVER (PARTITION BY department ORDER BY salary DESC) AS salary_rank
+FROM employee_status
+GROUP BY GROUPING SETS ((department), (department, employee_id));
+
+-- 3. Query with JSON accessors, lateral joins, and functions
+SELECT u.id, u.name, j.key, j.value
+FROM users u,
+LATERAL jsonb_each_text(u.preferences) AS j(key, value)
+WHERE j.key LIKE 'notif_%' AND j.value::boolean = true;
+
+-- 4. Use of EXISTS with nested correlated subqueries and CASE logic
+SELECT p.id, p.title,
+  CASE
+    WHEN EXISTS (
+      SELECT 1 FROM reviews r
+      WHERE r.product_id = p.id AND r.rating >= 4
+    ) THEN 'Popular'
+    ELSE 'Unrated'
+  END AS status
+FROM products p
+WHERE p.archived = false;
+
+-- 5. Inlined functions, CTEs, multi-level nesting with type casts
+WITH logs AS (
+  SELECT id, payload::json->>'event' AS event, (payload::json->>'ts')::timestamp AS ts
+  FROM event_log
+  WHERE ts > NOW() - INTERVAL '7 days'
+)
+SELECT event, COUNT(*) AS freq
+FROM (
+  SELECT DISTINCT event, ts::date AS event_day
+  FROM logs
+) d
+GROUP BY event
+ORDER BY freq DESC;
+
+-- 6. A massive one for ya
+
+SELECT 
+  o.id AS order_id,
+  u.name AS user_name,
+  p.name AS product_name,
+  s.status,
+  sh.shipped_at,
+  r.refund_amount
+FROM orders o
+JOIN users u 
+  ON o.user_id = u.id
+JOIN order_items oi 
+  ON oi.order_id = o.id
+JOIN products p 
+  ON (
+    (p.id = oi.product_id AND p.available = true)
+    OR 
+    (p.sku = oi.product_sku AND (p.discontinued = false OR p.replacement_id IS NOT NULL))
+  )
+LEFT JOIN shipping sh 
+  ON (
+    sh.order_id = o.id 
+    AND (
+      (sh.carrier = 'UPS' AND sh.tracking_number IS NOT NULL)
+      OR 
+      (sh.carrier = 'FedEx' AND sh.shipped_at > o.created_at + INTERVAL '1 day')
+    )
+  )
+LEFT JOIN statuses s 
+  ON s.id = o.status_id 
+  AND (
+    s.name != 'cancelled'
+    OR (s.name = 'cancelled' AND s.updated_at > NOW() - INTERVAL '7 days')
+  )
+LEFT JOIN refunds r 
+  ON r.order_id = o.id 
+  AND (
+    (r.status = 'approved' AND r.processed_at IS NOT NULL)
+    OR 
+    (r.status = 'pending' AND r.requested_at < NOW() - INTERVAL '14 days')
+  )
+WHERE o.created_at > NOW() - INTERVAL '90 days'
+  AND u.active = true
+  AND (
+    s.status = 'shipped' 
+    OR (
+      s.status = 'processing' 
+      AND EXISTS (
+        SELECT 1 FROM order_notes n WHERE n.order_id = o.id AND n.note ILIKE '%expedite%'
+      )
+    )
+  )
+ORDER BY o.created_at DESC;

packages/deparser/__tests__/kitchen-sink/pretty-misc.test.ts

@@ -0,0 +1,14 @@
+
+import { FixtureTestUtils } from '../../test-utils';
+const fixtures = new FixtureTestUtils();
+
+it('pretty-misc', async () => {
+  await fixtures.runFixtureTests([
+  "pretty/misc-1.sql",
+  "pretty/misc-2.sql",
+  "pretty/misc-3.sql",
+  "pretty/misc-4.sql",
+  "pretty/misc-5.sql",
+  "pretty/misc-6.sql"
+]);
+});