SET citus.next_shard_id TO 100500;
SELECT citus.mitmproxy('conn.allow()');
 mitmproxy
---------------------------------------------------------------------

(1 row)

CREATE TABLE ref_table (key int, value int);
SELECT create_reference_table('ref_table');
 create_reference_table
---------------------------------------------------------------------

(1 row)

\copy ref_table FROM stdin delimiter ',';
SELECT citus.clear_network_traffic();
 clear_network_traffic
---------------------------------------------------------------------

(1 row)

SELECT COUNT(*) FROM ref_table;
 count
---------------------------------------------------------------------
     4
(1 row)

-- verify behavior of single INSERT; should fail to execute
SELECT citus.mitmproxy('conn.onQuery(query="INSERT").kill()');
 mitmproxy
---------------------------------------------------------------------

(1 row)

INSERT INTO ref_table VALUES (5, 6);
ERROR:  connection to the remote node localhost:xxxxx failed with the following error: connection not open
SELECT COUNT(*) FROM ref_table WHERE key=5;
 count
---------------------------------------------------------------------
     0
(1 row)

-- verify behavior of UPDATE ... RETURNING; should not execute
SELECT citus.mitmproxy('conn.onQuery(query="UPDATE").kill()');
 mitmproxy
---------------------------------------------------------------------

(1 row)

UPDATE ref_table SET key=7 RETURNING value;
ERROR:  connection to the remote node localhost:xxxxx failed with the following error: connection not open
SELECT COUNT(*) FROM ref_table WHERE key=7;
 count
---------------------------------------------------------------------
     0
(1 row)

-- verify fix to #2214; should raise error and fail to execute
SELECT citus.mitmproxy('conn.onQuery(query="UPDATE").kill()');
 mitmproxy
---------------------------------------------------------------------

(1 row)

BEGIN;
DELETE FROM ref_table WHERE key=5;
UPDATE ref_table SET key=value;
ERROR:  connection to the remote node localhost:xxxxx failed with the following error: connection not open
COMMIT;
SELECT COUNT(*) FROM ref_table WHERE key=value;
 count
---------------------------------------------------------------------
     0
(1 row)

-- all shards should still be healthy
SELECT COUNT(*) FROM pg_dist_shard_placement WHERE shardstate = 3;
 count
---------------------------------------------------------------------
     0
(1 row)

-- ==== Clean up, we're done here ====
SELECT citus.mitmproxy('conn.allow()');
 mitmproxy
---------------------------------------------------------------------

(1 row)

DROP TABLE ref_table;