Parse comment if exist

Author: StefanGajic

test-crawler/collector/collector.go

@@ -33,7 +33,7 @@ type FunctionAnnotation struct {
 	Name         string `yaml:"name"`
 	InputParams  string `yaml:"inputParams"`
 	ReturnValues string `yaml:"returnValues"`
-	Description  string
+	Description  string `yaml:"description"`
 	Public       bool
 }
 

test-crawler/extractor/extractor.go

@@ -212,7 +212,6 @@ func getFunctionNodes(content string, treeCursor *sitter.TreeCursor, parser *a.P
 		if child != nil {
 			if child.Type() == string(FUNCTION_DECLARATION) {
 				if prevNode.Type() == string(COMMENT) {
-
 					value, annotationType, _ := parser.Parse(content[prevNode.StartByte():prevNode.EndByte()])
 					if value != nil && annotationType == a.Ignore {
 						isIgnored = value.(bool)
@@ -261,6 +260,13 @@ func getFunctionNodes(content string, treeCursor *sitter.TreeCursor, parser *a.P
 				child.Child(4): parameter_list:  (<-chan *types2.RequestEvent, error)
 			*/
 			if child.Type() == string(METHOD_DECLARATION) {
+				if prevNode.Type() == string(COMMENT) {
+					value, annotationType, _ := parser.Parse(content[prevNode.StartByte():prevNode.EndByte()])
+					if value != nil && annotationType == a.Ignore {
+						isIgnored = value.(bool)
+					}
+				}
+
 				funcName = content[child.Child(2).StartByte():child.Child(2).EndByte()]
 				params := ""
 				returnValues := ""