test: getRequestFromFunction

src/main/kotlin/fr/postgresjson/connexion/Connection.kt

@@ -151,8 +151,12 @@ class Connection(
     }
 
     private fun fetchFunctions(functionsDirectory: File) {
-        functionsDirectory.walk().filter { it.isDirectory }.forEach { directory ->
+        functionsDirectory.walk().filter {
-            directory.walk().filter { it.isFile }.forEach { file ->
+            it.isDirectory
+        }.forEach { directory ->
+            directory.walk().filter {
+                it.isFile
+            }.forEach { file ->
                 val fileContent = file.readText()
                 getDefinitions(fileContent).forEach {
                     functions[it.name] = it

src/test/kotlin/fr/postgresjson/RequestTest.kt

@@ -12,7 +12,15 @@ class RequestTest {
     @Test
     fun getRequestFromFile() {
         val resources = File(this::class.java.getResource("/sql/query").toURI())
-        val objTest: ObjTest? = Connection(queriesDirectory = resources).getQuery("Test/test").selectOne()
+        val objTest: ObjTest? = Connection(queriesDirectory = resources).getQuery("Test/selectOne").selectOne()
+        assertTrue(objTest!!.id == 2)
+        assertTrue(objTest.name == "test")
+    }
+
+    @Test
+    fun getRequestFromFunction() {
+        val resources = File(this::class.java.getResource("/sql/function").toURI())
+        val objTest: ObjTest? = Connection(functionsDirectory = resources).getFunction("test_function").selectOne()
         assertTrue(objTest!!.id == 2)
         assertTrue(objTest.name == "test")
     }

src/test/resources/sql/function/Test/function_test.sql

@@ -0,0 +1,8 @@
+CREATE OR REPLACE FUNCTION test_function (name text, IN hi text default 'hello', out result json)
+LANGUAGE plpgsql
+AS
+$$
+BEGIN
+    result = json_build_object('id', 2, 'name', 'test');
+END;
+$$