Merge pull request #79 from zalando-stups/add-test-with-enum

Add test which returns enum from a function

Author: vadeg

database/test/21_shard1/21_schema1/05_stored_procedures/52_return_enum_from_function.sql

@@ -0,0 +1,9 @@
+CREATE OR REPLACE FUNCTION ztest_schema1.return_enum_from_function() RETURNS example_enum AS
+$BODY$
+DECLARE
+    r ztest_schema1.example_enum;
+BEGIN
+    select 'ENUM_CONST_1'::example_enum into r;
+    return r;
+END;
+$BODY$ LANGUAGE plpgsql;

src/test/java/org/zalando/sprocwrapper/example/ExampleSProcService.java

@@ -292,4 +292,6 @@ int createOrder(@SProcParam String orderNumber, @SProcParam OrderMonetaryAmount
     @SProcCall
     List<ExampleEnumDomainObject> listExampleEnumDomainObjects();
 
+    @SProcCall
+    ExampleEnum returnEnumFromFunction();
 }

src/test/java/org/zalando/sprocwrapper/example/ExampleSProcServiceImpl.java

@@ -464,4 +464,9 @@ public ExampleEnumDomainObject getExampleEnumDomainObject(Integer id) {
     public List<ExampleEnumDomainObject> listExampleEnumDomainObjects() {
         return sproc.listExampleEnumDomainObjects();
     }
+
+    @Override
+    public ExampleEnum returnEnumFromFunction() {
+        return sproc.returnEnumFromFunction();
+    }
 }

src/test/java/org/zalando/typemapper/namedresult/EnumTestIT.java

@@ -9,9 +9,15 @@
 
 import org.junit.runner.RunWith;
 
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.test.context.ContextConfiguration;
 import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
 
 
+import static org.junit.Assert.assertNotNull;
+
+
+import org.zalando.sprocwrapper.example.ExampleSProcService;
 import org.zalando.typemapper.AbstractTest;
 import org.zalando.typemapper.namedresult.results.ClassWithEmbedEnumClass;
 import org.zalando.typemapper.core.TypeMapper;
@@ -20,8 +26,12 @@
 import org.zalando.typemapper.namedresult.results.Enumeration;
 
 @RunWith(SpringJUnit4ClassRunner.class)
+@ContextConfiguration(locations = {"classpath:backendContextTest.xml"})
 public class EnumTestIT extends AbstractTest {
 
+    @Autowired
+    private ExampleSProcService exampleSProcService;
+
     @Test
     public void testEnumMappings() throws SQLException {
         final PreparedStatement ps = connection.prepareStatement("SELECT 0 as a, 'VALUE_2' as b");
@@ -65,4 +75,9 @@ public void testEnumMappingsWithEnumInComplexType() throws SQLException {
             Assert.assertEquals(Enumeration.VALUE_2, result.getEmbeddedEnum().getValue2());
         }
     }
+
+    @Test
+    public void testReturnEnumValue() {
+        assertNotNull(exampleSProcService.returnEnumFromFunction());
+    }
 }