feat: <go> add basic return type support

phodal · Feb 12, 2020 · 457648d · 457648d
1 parent 88557dc
commit 457648d
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 4 deletions.
diff --git a/chapi-ast-go/src/main/kotlin/chapi/ast/goast/GoFullIdentListener.kt b/chapi-ast-go/src/main/kotlin/chapi/ast/goast/GoFullIdentListener.kt
@@ -52,8 +52,13 @@ class GoFullIdentListener(var fileName: String) : GoAstListener() {
 
         val result = ctx.signature().result()
         if (result != null) {
-            println("enterMethodDecl -> " + result.text)
+            val returnType = CodeProperty(
+                TypeType = result.text,
+                TypeValue = ""
+            )
+            codeFunction.MultipleReturns += returnType
         }
+
         this.buildParameters(ctx.signature().parameters())
 
         val receiverName = this.getTypeNameFromReceiver(ctx.receiver().parameters())!!
@@ -80,8 +85,6 @@ class GoFullIdentListener(var fileName: String) : GoAstListener() {
             when (typeChild::class.java.simpleName) {
                 "StructTypeContext" -> {
                     buildStruct(identifyName, typeChild)
-
-
                 }
                 else -> {
 

diff --git a/chapi-ast-go/src/test/kotlin/chapi/ast/goast/GoFullIdentListenerTest.kt b/chapi-ast-go/src/test/kotlin/chapi/ast/goast/GoFullIdentListenerTest.kt
@@ -102,9 +102,30 @@ func (a *Animal) Move() {
         val codeFile = GoAnalyser().analysis(code, "")
         assertEquals(codeFile.DataStructures.size, 1)
         assertEquals(codeFile.DataStructures[0].NodeName, "Animal")
-        println(codeFile.DataStructures[0])
         assertEquals(codeFile.DataStructures[0].Fields.size, 1)
         assertEquals(codeFile.DataStructures[0].Functions.size, 1)
         assertEquals(codeFile.DataStructures[0].Functions[0].Name, "Move")
     }
+
+    @Test
+    internal fun shouldIdentifyStructFunctionReturnType() {
+        var code = """
+package main
+
+import "fmt"
+
+type Animal struct {
+	Age int
+}
+
+func (a *Animal) Move() string {
+	return ""
+}
+"""
+
+        val codeFile = GoAnalyser().analysis(code, "")
+        assertEquals(codeFile.DataStructures.size, 1)
+        assertEquals(codeFile.DataStructures[0].Functions[0].MultipleReturns.size, 1)
+        assertEquals(codeFile.DataStructures[0].Functions[0].MultipleReturns[0].TypeType, "string")
+    }
 }