Added task for unit test generation

Author: codemakerai-dev

cli/cli.go

@@ -11,6 +11,7 @@ import (
 	"go.uber.org/zap/zapcore"
 	"os"
 	"path/filepath"
+	"strings"
 	"time"
 )
 
@@ -97,6 +98,36 @@ func (c *Cli) parseGenerateArgs() {
 			c.logger.Errorf("Could not generate the documentation %v", err)
 		}
 		break
+	case "unit-tests":
+		generateTestsCmd := flag.NewFlagSet("generateUnitTests", flag.ExitOnError)
+		lang := generateTestsCmd.String("language", "", "Programming language: Java")
+		outputDir := generateTestsCmd.String("output-dir", "", "The output directory")
+
+		err := generateTestsCmd.Parse(os.Args[3:])
+		if err != nil {
+			c.logger.Error("Could not parse args %v", err)
+			os.Exit(1)
+		}
+
+		if len(generateTestsCmd.Args()) == 0 {
+			c.logger.Errorf("Expected file input")
+			fmt.Printf("Usage: codemaker generate tests <file>")
+			os.Exit(1)
+		}
+
+		config, err := createConfig()
+		if err != nil {
+			c.logger.Error("No valid api key found %v", err)
+			os.Exit(1)
+		}
+
+		cl := c.createClient(*config)
+		input := generateTestsCmd.Args()[0:]
+
+		if err := c.generateTests(cl, lang, input, outputDir); err != nil {
+			c.logger.Errorf("Could not generate the test %v", err)
+		}
+		break
 	default:
 		fmt.Printf("Unknown command %s\n", os.Args[2])
 		c.printGenerateHelp()
@@ -122,7 +153,7 @@ func (c *Cli) parseMigrateArgs() {
 
 		if len(migrateSyntaxCmd.Args()) == 0 {
 			c.logger.Errorf("Expected file input")
-			fmt.Printf("Usage: codemaker migrate syntax <file>\n")
+			fmt.Printf("Usage: codemaker migrate syntax <file>")
 			os.Exit(1)
 		}
 
@@ -148,7 +179,7 @@ func (c *Cli) parseMigrateArgs() {
 func (c *Cli) generateDocumentation(cl client.Client, lang *string, files []string) error {
 	return c.walkPath(files, func(file string) error {
 		if lang == nil || len(*lang) == 0 {
-			actLang, err := LanguageFromExtension(filepath.Ext(file))
+			actLang, err := languageFromExtension(filepath.Ext(file))
 			if err != nil {
 				return err
 			}
@@ -174,10 +205,59 @@ func (c *Cli) generateDocumentation(cl client.Client, lang *string, files []stri
 	})
 }
 
+func (c *Cli) generateTests(cl client.Client, lang *string, files []string, outputDir *string) error {
+	return c.walkPath(files, func(file string) error {
+		if lang == nil || len(*lang) == 0 {
+			actLang, err := languageFromExtension(filepath.Ext(file))
+			if err != nil {
+				c.logger.Errorf("skipping unsupported file %s", file)
+				return err
+			}
+			lang = &actLang
+		}
+
+		c.logger.Infof("Generating tests for file %s", file)
+		source, err := c.readFile(file)
+		if err != nil {
+			c.logger.Errorf("failed to read file %s %v", file, err)
+			return err
+		}
+
+		output, err := c.process(cl, client.ModeUnitTest, *lang, "", source)
+		if err != nil {
+			c.logger.Errorf("failed to generate documentation for file %s %v", file, err)
+			return err
+		}
+
+		suffix, err := testFileSuffix(*lang)
+		if err != nil {
+			c.logger.Errorf("could not get suffix for file %s %v", file, err)
+			return err
+		}
+
+		var outputFile string
+		if outputDir != nil && len(*outputDir) > 0 {
+			err := os.MkdirAll(*outputDir, 0755)
+			if err != nil {
+				c.logger.Errorf("could not create directory %s", *outputDir)
+				return err
+			}
+			outputFile = filepath.Join(*outputDir, strings.TrimSuffix(filepath.Base(file), filepath.Ext(file))+suffix)
+		} else {
+			outputFile = strings.TrimSuffix(file, filepath.Ext(file)) + suffix
+		}
+		if err := c.writeFile(outputFile, *output); err != nil {
+			c.logger.Errorf("failed to write file %s %v", file, err)
+			return err
+		}
+		return nil
+	})
+}
+
 func (c *Cli) migrateSyntax(cl client.Client, lang *string, langVer *string, files []string) error {
 	return c.walkPath(files, func(file string) error {
 		if lang == nil || len(*lang) == 0 {
-			actLang, err := LanguageFromExtension(filepath.Ext(file))
+			actLang, err := languageFromExtension(filepath.Ext(file))
 			if err != nil {
 				c.logger.Errorf("skipping unsupported file %s", file)
 				return nil
@@ -330,6 +410,7 @@ func (c *Cli) printGenerateHelp() {
 	fmt.Printf("\n")
 	fmt.Printf("Commands:\n")
 	fmt.Printf(" * docs\n")
+	fmt.Printf(" * unit-tests\n")
 	os.Exit(1)
 }
 

cli/language.go

@@ -8,9 +8,20 @@ var fileExtensions = map[string]string{
 	".java": "JAVA",
 }
 
-func LanguageFromExtension(extension string) (string, error) {
+var testFileSuffixes = map[string]string{
+	"JAVA": "Test.java",
+}
+
+func languageFromExtension(extension string) (string, error) {
 	if lang, ok := fileExtensions[extension]; ok {
 		return lang, nil
 	}
 	return "", fmt.Errorf("the file extension %s is not supported", extension)
 }
+
+func testFileSuffix(language string) (string, error) {
+	if lang, ok := testFileSuffixes[language]; ok {
+		return lang, nil
+	}
+	return "", fmt.Errorf("the language %s is not supported", language)
+}

go.mod

@@ -3,7 +3,7 @@ module github.com/codemakerai/codemaker-cli
 go 1.20
 
 require (
-	github.com/codemakerai/codemaker-sdk-go v0.0.3-0.20230322062931-ad3bd008ba77
+	github.com/codemakerai/codemaker-sdk-go v0.0.3
 	github.com/joho/godotenv v1.5.1
 	go.uber.org/zap v1.24.0
 )

go.sum

@@ -1,8 +1,6 @@
 github.com/benbjohnson/clock v1.1.0 h1:Q92kusRqC1XV2MjkWETPvjJVqKetz1OzxZB7mHJLju8=
-github.com/codemakerai/codemaker-sdk-go v0.0.3-0.20230322062523-70a6d43cfaf9 h1:lZDG+1c1tIEUCcoHODEBmGdAGCMDIXFPmHImuuZGDs8=
-github.com/codemakerai/codemaker-sdk-go v0.0.3-0.20230322062523-70a6d43cfaf9/go.mod h1:drvi/xHywQnOgWWGw3dOWZ4CV0NesAdNNuAtsVdZW2E=
-github.com/codemakerai/codemaker-sdk-go v0.0.3-0.20230322062931-ad3bd008ba77 h1:TBLLWF9zZhmmnHJFz6Io8FtBUcOI4i/YXDGjw7oXYrc=
-github.com/codemakerai/codemaker-sdk-go v0.0.3-0.20230322062931-ad3bd008ba77/go.mod h1:drvi/xHywQnOgWWGw3dOWZ4CV0NesAdNNuAtsVdZW2E=
+github.com/codemakerai/codemaker-sdk-go v0.0.3 h1:+nh2/UUiYuvmDA2hmt+wTp6nN/VS/ykemkYZ+ip9KGc=
+github.com/codemakerai/codemaker-sdk-go v0.0.3/go.mod h1:drvi/xHywQnOgWWGw3dOWZ4CV0NesAdNNuAtsVdZW2E=
 github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
 github.com/joho/godotenv v1.5.1 h1:7eLL/+HRGLY0ldzfGMeQkb7vMd0as4CfYvUVzLqw0N0=
 github.com/joho/godotenv v1.5.1/go.mod h1:f4LDr5Voq0i2e/R5DDNOoa2zzDfwtkZa6DnEwAbqwq4=