Generate basic YAML file

Author: StefanGajic

test-crawler/collector/collector.go

@@ -29,6 +29,7 @@ type Function struct {
 }
 
 type FunctionAnnotation struct {
+	ID           int    `yaml:"id"`
 	Name         string `yaml:"name"`
 	InputParams  string `yaml:"inputParams"`
 	ReturnValues string `yaml:"returnValues"`

test-crawler/main.go

@@ -5,6 +5,7 @@ import (
 	"context"
 	"encoding/json"
 	"fmt"
+	"io/ioutil"
 	"os"
 	"os/exec"
 	"strings"
@@ -13,6 +14,8 @@ import (
 	a "testsuites/annotations"
 	c "testsuites/collector"
 	ex "testsuites/extractor"
+
+	y "gopkg.in/yaml.v2"
 )
 
 type FnLink struct {
@@ -87,7 +90,8 @@ func crawlSingleFileForMethods(path string) {
 	}
 	for _, fn := range fns {
 		fmt.Printf(
-			"name %s : public %v : params: %s : return values : %s\n",
+			"ID %d : name %s : public %v : params: %s : return values : %s\n",
+			fn.ID,
 			fn.Name,
 			fn.Public,
 			fn.InputParams,
@@ -100,6 +104,36 @@ func extractPublicMethodsFromFile(ctx context.Context, filePath string) ([]c.Fun
 	return ex.GetExportedFunctions(ctx, filePath)
 }
 
+// makeYAML will make yaml file from public methods.
+func makeYAML(ctx context.Context, filePath string) ([]byte, error) {
+
+	publicMethods, err := extractPublicMethodsFromFile(ctx, filePath)
+	if err != nil {
+		fmt.Print(err)
+		os.Exit(1)
+	}
+
+	var count int
+	for _, fn := range publicMethods {
+		count++
+		fn.ID = count
+		publicMethods = append(publicMethods[1:], fn)
+
+	}
+
+	yamlData, err := y.Marshal(&publicMethods)
+	if err != nil {
+		fmt.Printf("Error while Marshaling. %v", err)
+	}
+
+	fileName := "test.yaml"
+	err = ioutil.WriteFile(fileName, yamlData, 0644)
+	if err != nil {
+		panic("Unable to write data into the file")
+	}
+	return yamlData, nil
+}
+
 func linkFiles(flist []c.Function) (links [][]FnLink) {
 
 	functions := make(map[string]c.Function)