Added test to reproduce bug perseas#103

dvarrazzo · Sep 6, 2014 · 9b03d3e · 9b03d3e
1 parent 2ef4829
commit 9b03d3e
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/tests/dbobject/test_function.py b/tests/dbobject/test_function.py
@@ -2,6 +2,7 @@
 """Test functions"""
 
 import pytest
+import psycopg2
 
 from pyrseas.testutils import DatabaseToMapTestCase
 from pyrseas.testutils import InputMapToSqlTestCase, fix_indent
@@ -124,6 +125,16 @@ def test_map_function_leakproof(self):
         assert dbmap['schema public']['function f1(integer, integer)'] == \
             expmap
 
+    def test_bug_103(self):
+        try:
+            self.to_map(["CREATE OR REPLACE LANGUAGE plpythonu"])
+        except psycopg2.OperationalError as e:
+            self.skipTest("plpython installation failed: %s" % e)
+
+        self.to_map(["""
+            CREATE FUNCTION test103() RETURNS int AS
+            'return 1' LANGUAGE plpythonu"""])
+
 
 class FunctionToSqlTestCase(InputMapToSqlTestCase):
     """Test SQL generation from input functions"""