Switch plan(int) to plan(bigint)

Changes

@@ -10,6 +10,9 @@ Revision history for pgTAP
   to Jim Nasby for reporting the issue (#234) Rodolphe Quiédeville for for the
   pull request (#339).
 * Added `index_is_partial()`, thanks to Rodolphe Quiédeville (#342).
+* Change plan() to accept bigint. While only an insane person would need more
+  than 2^31 tests, the bigint version makes calling plan with the count(*) of
+  rows in a table of tests easier.
 
 1.3.3 2024-04-08T13:44:11Z
 --------------------------

sql/pgtap--1.3.3--1.3.4.sql

@@ -87,3 +87,43 @@ BEGIN
       );
 END;
 $$ LANGUAGE plpgsql;
+
+DROP FUNCTION plan( integer );
+CREATE OR REPLACE FUNCTION plan( bigint )
+RETURNS TEXT AS $$
+DECLARE
+    rcount INTEGER;
+BEGIN
+    BEGIN
+        EXECUTE '
+            CREATE TEMP SEQUENCE __tcache___id_seq;
+            CREATE TEMP TABLE __tcache__ (
+                id    INTEGER NOT NULL DEFAULT nextval(''__tcache___id_seq''),
+                label TEXT    NOT NULL,
+                value INTEGER NOT NULL,
+                note  TEXT    NOT NULL DEFAULT ''''
+            );
+            CREATE UNIQUE INDEX __tcache___key ON __tcache__(id);
+            GRANT ALL ON TABLE __tcache__ TO PUBLIC;
+            GRANT ALL ON TABLE __tcache___id_seq TO PUBLIC;
+
+            CREATE TEMP SEQUENCE __tresults___numb_seq;
+            GRANT ALL ON TABLE __tresults___numb_seq TO PUBLIC;
+        ';
+
+    EXCEPTION WHEN duplicate_table THEN
+        -- Raise an exception if there's already a plan.
+        EXECUTE 'SELECT TRUE FROM __tcache__ WHERE label = ''plan''';
+      GET DIAGNOSTICS rcount = ROW_COUNT;
+        IF rcount > 0 THEN
+           RAISE EXCEPTION 'You tried to plan twice!';
+        END IF;
+    END;
+
+    -- Save the plan and return.
+    PERFORM _set('plan', $1::int );
+    PERFORM _set('failed', 0 );
+    RETURN '1..' || $1;
+END;
+$$ LANGUAGE plpgsql strict;
+

sql/pgtap.sql.in

@@ -22,7 +22,7 @@ CREATE OR REPLACE FUNCTION pgtap_version()
 RETURNS NUMERIC AS 'SELECT __VERSION__;'
 LANGUAGE SQL IMMUTABLE;
 
-CREATE OR REPLACE FUNCTION plan( integer )
+CREATE OR REPLACE FUNCTION plan( bigint )
 RETURNS TEXT AS $$
 DECLARE
     rcount INTEGER;
@@ -54,7 +54,7 @@ BEGIN
     END;
 
     -- Save the plan and return.
-    PERFORM _set('plan', $1 );
+    PERFORM _set('plan', $1::int );
     PERFORM _set('failed', 0 );
     RETURN '1..' || $1;
 END;