From 35ff7ee3372d78c63e58be29451dec9dac498603 Mon Sep 17 00:00:00 2001
From: Oliver Rice <github@oliverrice.com>
Date: Wed, 21 Aug 2024 11:44:26 -0500
Subject: [PATCH] add pg-safeupdate tests

---
 nix/tests/expected/pg-safeupdate.out | 12 ++++++++++++
 nix/tests/sql/pg-safeupdate.sql      | 15 +++++++++++++++
 2 files changed, 27 insertions(+)
 create mode 100644 nix/tests/expected/pg-safeupdate.out
 create mode 100644 nix/tests/sql/pg-safeupdate.sql

diff --git a/nix/tests/expected/pg-safeupdate.out b/nix/tests/expected/pg-safeupdate.out
new file mode 100644
index 000000000..f9100116a
--- /dev/null
+++ b/nix/tests/expected/pg-safeupdate.out
@@ -0,0 +1,12 @@
+load 'safeupdate';
+set safeupdate.enabled=1;
+create schema v;
+create table v.foo(
+  id int,
+  val text
+);
+update v.foo
+  set val = 'bar';
+ERROR:  UPDATE requires a WHERE clause
+drop schema v cascade;
+NOTICE:  drop cascades to table v.foo
diff --git a/nix/tests/sql/pg-safeupdate.sql b/nix/tests/sql/pg-safeupdate.sql
new file mode 100644
index 000000000..790ec79fa
--- /dev/null
+++ b/nix/tests/sql/pg-safeupdate.sql
@@ -0,0 +1,15 @@
+load 'safeupdate';
+
+set safeupdate.enabled=1;
+
+create schema v;
+
+create table v.foo(
+  id int,
+  val text
+);
+
+update v.foo
+  set val = 'bar';
+
+drop schema v cascade;