Add more tests

language-server/modules/langserver-core/src/main/java/org/ballerinalang/langserver/command/visitors/FunctionCallExpressionTypeFinder.java

@@ -29,6 +29,7 @@
 import io.ballerina.compiler.api.symbols.ParameterSymbol;
 import io.ballerina.compiler.api.symbols.RecordFieldSymbol;
 import io.ballerina.compiler.api.symbols.RecordTypeSymbol;
+import io.ballerina.compiler.api.symbols.StreamTypeSymbol;
 import io.ballerina.compiler.api.symbols.Symbol;
 import io.ballerina.compiler.api.symbols.SymbolKind;
 import io.ballerina.compiler.api.symbols.TypeDescKind;
@@ -490,8 +491,11 @@ public void visit(GroupingKeyVarDeclarationNode groupingKeyVarDeclarationNode) {
     public void visit(SelectClauseNode selectClauseNode) {
         selectClauseNode.parent().parent().accept(this);
         if (resultFound) {
-            if (this.returnTypeSymbol.typeKind() == TypeDescKind.ARRAY) {
+            TypeDescKind kind = this.returnTypeSymbol.typeKind();
+            if (kind == TypeDescKind.ARRAY) {
                 checkAndSetTypeResult(((ArrayTypeSymbol) returnTypeSymbol).memberTypeDescriptor());
+            } else if (kind == TypeDescKind.STREAM) {
+                checkAndSetTypeResult(((StreamTypeSymbol) returnTypeSymbol).typeParameter());
             }
         }
     }

language-server/modules/langserver-core/src/test/java/org/ballerinalang/langserver/codeaction/CreateFunctionTest.java

@@ -182,6 +182,7 @@ public Object[][] dataProvider() {
                 {"undefinedFunctionCodeAction28.json"},
                 {"undefinedFunctionCodeAction29.json"},
                 {"undefinedFunctionCodeAction30.json"},
+                {"undefinedFunctionCodeAction31.json"},
         };
     }
 

language-server/modules/langserver-core/src/test/resources/codeaction/create-function/config/undefinedFunctionCodeAction31.json

@@ -0,0 +1,57 @@
+{
+  "position": {
+    "line": 6,
+    "character": 22
+  },
+  "source": "createUndefinedFunction20.bal",
+  "expected": [
+    {
+      "title": "Create function 'foo(...)'",
+      "kind": "quickfix",
+      "edits": [
+        {
+          "range": {
+            "start": {
+              "line": 7,
+              "character": 1
+            },
+            "end": {
+              "line": 7,
+              "character": 1
+            }
+          },
+          "newText": "\n\nfunction foo(string letter) returns MyType {\n    return {value: \"\"};\n}"
+        }
+      ],
+      "resolvable": true,
+      "data": {
+        "extName": "org.ballerinalang.langserver.codeaction.BallerinaCodeActionExtension",
+        "codeActionName": "Create Function",
+        "fileUri": "createUndefinedFunction20.bal",
+        "range": {
+          "start": {
+            "line": 6,
+            "character": 19
+          },
+          "end": {
+            "line": 6,
+            "character": 30
+          }
+        },
+        "actionData": {
+          "key": "node.range",
+          "value": {
+            "start": {
+              "line": 6.0,
+              "character": 19.0
+            },
+            "end": {
+              "line": 6.0,
+              "character": 30.0
+            }
+          }
+        }
+      }
+    }
+  ]
+}

language-server/modules/langserver-core/src/test/resources/codeaction/create-function/source/createUndefinedFunction20.bal

@@ -0,0 +1,8 @@
+type MyType record {|
+    string value;
+|};
+
+function func() {
+    stream<MyType> streamName = stream from string letter in "JohnSnow"
+            select foo(letter);
+}