add code

.gitignore

@@ -12,6 +12,7 @@ commands/.temp
 
 # never upload the build to git
 armstrong
+armstrong.exe
 
 # test output
 coverage/test_coverage_report*.md

commands/api_test_generate_report.go

@@ -0,0 +1,69 @@
+package commands
+
+import (
+	"flag"
+	"fmt"
+	"github.com/ms-henglu/armstrong/report"
+	"github.com/sirupsen/logrus"
+	"log"
+	"os"
+	"path/filepath"
+	"strings"
+)
+
+type ApiTestGenerateReportCommand struct {
+	workingDir  string
+	swaggerPath string
+}
+
+func (c *ApiTestGenerateReportCommand) flags() *flag.FlagSet {
+	fs := defaultFlagSet("api-test-generate-report")
+	fs.StringVar(&c.workingDir, "working-dir", "", "path that contains all the test cases")
+	fs.StringVar(&c.swaggerPath, "swagger", "", "path to the .json swagger which is being test")
+	fs.Usage = func() { logrus.Error(c.Help()) }
+	return fs
+}
+
+func (c ApiTestGenerateReportCommand) Help() string {
+	helpText := `
+Usage: armstrong api-test-generate-report
+` + c.Synopsis() + "\n\n" + helpForFlags(c.flags())
+
+	return strings.TrimSpace(helpText)
+}
+
+func (c ApiTestGenerateReportCommand) Synopsis() string {
+	return "Generate test report for a set of test results"
+}
+
+func (c ApiTestGenerateReportCommand) Run(args []string) int {
+	f := c.flags()
+	if err := f.Parse(args); err != nil {
+		logrus.Error(fmt.Sprintf("Error parsing command-line flags: %s", err))
+		return 1
+	}
+	return c.Execute()
+}
+
+func (c ApiTestGenerateReportCommand) Execute() int {
+	log.Println("[INFO] ----------- generate API Test Report ---------")
+	wd, err := os.Getwd()
+	if err != nil {
+		logrus.Error(fmt.Sprintf("failed to get working directory: %+v", err))
+		return 1
+	}
+
+	if c.workingDir != "" {
+		wd, err = filepath.Abs(c.workingDir)
+		if err != nil {
+			logrus.Error(fmt.Sprintf("working directory is invalid: %+v", err))
+			return 1
+		}
+	}
+
+	if report.GenerateApiTestReports(wd, c.swaggerPath) != nil {
+		log.Fatalf("[ERROR] failed to generate API Test Report: %+v", err)
+	}
+
+	return 0
+}

commands/test.go

@@ -18,14 +18,18 @@ import (
 )
 
 type TestCommand struct {
-	verbose    bool
-	workingDir string
+	verbose          bool
+	workingDir       string
+	destroyAfterTest bool
+	swaggerPath      string
 }
 
 func (c *TestCommand) flags() *flag.FlagSet {
 	fs := defaultFlagSet("test")
 	fs.BoolVar(&c.verbose, "v", false, "whether show terraform logs")
 	fs.StringVar(&c.workingDir, "working-dir", "", "path to Terraform configuration files")
+	fs.BoolVar(&c.destroyAfterTest, "destroy-after-test", false, "whether to destroy the created resources after each test")
+	fs.StringVar(&c.swaggerPath, "swagger", "", "path to the .json swagger which is being test")
 	fs.Usage = func() { logrus.Error(c.Help()) }
 	return fs
 }
@@ -140,9 +144,27 @@ func (c TestCommand) Execute() int {
 		} else {
 			log.Fatalf("[ERROR] error showing terraform state: %+v", err)
 		}
+
 		return 0
 	}
 
+	if applyErr == nil {
+		if c.destroyAfterTest {
+			destroyErr := terraform.Destroy()
+			if destroyErr != nil {
+				log.Printf("[ERROR] error running terraform destroy: %+v\n", destroyErr)
+			} else {
+				log.Println("[INFO] test resource has been deleted")
+			}
+		}
+	}
+
+	if c.swaggerPath != "" {
+		if report.StoreApiTestReport(wd, c.swaggerPath) != nil {
+			log.Fatalf("[ERROR] error storing api test report: %+v", err)
+		}
+	}
+
 	logs, err := report.ParseLogs(path.Join(wd, "log.txt"))
 	if err != nil {
 		log.Printf("[ERROR] parsing log.txt: %+v", err)

main.go

@@ -31,6 +31,9 @@ func main() {
 		"cleanup": func() (cli.Command, error) {
 			return &commands.CleanupCommand{}, nil
 		},
+		"api-test-generate-report": func() (cli.Command, error) {
+			return &commands.ApiTestGenerateReportCommand{}, nil
+		},
 	}
 
 	exitStatus, err := c.Run()