Changeset 13 for trunk/epic

Timestamp:

08/20/08 14:05:58 (4 years ago)

Author:

fumanchu

Message:

New test.assert_empty functions.

Location:

trunk/epic

Files:

• epic.sql (modified) (1 diff)
• test/test_asserts.sql (modified) (2 diffs)

trunk/epic/epic.sql

@@ -632 +632 @@
 END;
 $$ LANGUAGE plpgsql;
+
+
+CREATE OR REPLACE FUNCTION test.assert_empty(tablenames text[]) RETURNS VOID AS $$
+-- Raises an exception if the given tables have any rows.
+DECLARE
+  result      bool;
+  failed      text[];
+  failed_len  int;
+  i           int;
+BEGIN
+  FOR i in array_lower(tablenames, 1)..array_upper(tablenames, 1)
+  LOOP
+    EXECUTE 'SELECT (EXISTS (SELECT 1 FROM ' || tablenames[i] || '));' INTO result;
+    IF result THEN
+      failed := failed || ('"' || btrim(tablenames[i]) || '"');
+    END IF;
+  END LOOP;
+  
+  failed_len := (array_upper(failed, 1) - array_lower(failed, 1)) + 1;
+  IF failed_len = 1 THEN
+    PERFORM test.fail('The table ' || array_to_string(failed, ', ') || ' is not empty.');
+  ELSEIF failed_len > 1 THEN
+    PERFORM test.fail('The tables ' || array_to_string(failed, ', ') || ' are not empty.');
+  END IF;
+END;
+$$ LANGUAGE plpgsql;
+
+CREATE OR REPLACE FUNCTION test.assert_empty(tablename text) RETURNS VOID AS $$
+-- Raises an exception if the given table has any rows.
+DECLARE
+  result    bool;
+BEGIN
+  IF tablename LIKE '%,%' THEN
+    PERFORM test.assert_empty(string_to_array(tablename, ','));
+    RETURN;
+  END IF;
+  
+  EXECUTE 'SELECT (EXISTS (SELECT 1 FROM ' || tablename || '));' INTO result;
+  IF result THEN
+    PERFORM test.fail('The table "' || tablename || '" is not empty.');
+  END IF;
+END;
+$$ LANGUAGE plpgsql;
+

trunk/epic/test/test_asserts.sql

@@ -103 +103 @@
   
   -- assert_void() MUST raise an exception if the given call does not return void.
-  PERFORM test.assert_raises('test.assert_void(''pg_namespace'')', 
-    'Call: ''pg_namespace'' did not return void. Got ''pg_toast'' instead.',
+  PERFORM test.assert_raises('test.assert_void(''pg_namespace WHERE nspname = ''''pg_catalog'''''')', 
+    'Call: ''pg_namespace WHERE nspname = ''pg_catalog'''' did not return void. Got ''pg_catalog'' instead.',
     'P0001');
   
@@ -374 +374 @@
 END;
 $$ LANGUAGE plpgsql;
+
+
+CREATE OR REPLACE FUNCTION test.test_assert_empty() RETURNS VOID AS $$
+-- Assert the correct operation of test.assert_empty
+-- module: test_asserts
+DECLARE
+  failed     bool;
+BEGIN
+  -- Test an assertion that should pass
+  CREATE TEMP TABLE testtemp (a int);
+  PERFORM test.assert_empty('testtemp');
+  -- array version
+  CREATE TEMP TABLE testtemp2 (a int);
+  PERFORM test.assert_empty('testtemp, testtemp2');
+  PERFORM test.assert_empty(ARRAY['testtemp', 'testtemp2']);
+  
+  -- ...and an assertion that should fail
+  failed := false;
+  BEGIN
+    PERFORM test.assert_empty('pg_type,pg_proc');
+  EXCEPTION WHEN OTHERS THEN
+    failed := true;
+    IF SQLERRM = '[FAIL] The tables "pg_type", "pg_proc" are not empty.' THEN
+      NULL;
+    ELSE
+      RAISE EXCEPTION 'test.assert_empty() did not raise the correct error. Raised: %', SQLERRM;
+    END IF;
+  END;
+  IF NOT failed THEN
+    PERFORM test.fail('test.assert_empty() did not fail.');
+  END IF;
+  
+  RAISE EXCEPTION '[OK]';
+END;
+$$ LANGUAGE plpgsql;