mirror of
https://github.com/httprunner/httprunner.git
synced 2026-05-12 02:21:29 +08:00
75 lines
2.0 KiB
Go
75 lines
2.0 KiB
Go
package cmd
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
|
|
"github.com/rs/zerolog/log"
|
|
"github.com/spf13/cobra"
|
|
|
|
"github.com/httprunner/httprunner/v4/hrp/internal/builtin"
|
|
"github.com/httprunner/httprunner/v4/hrp/internal/convert"
|
|
"github.com/httprunner/httprunner/v4/hrp/internal/version"
|
|
)
|
|
|
|
var convertCmd = &cobra.Command{
|
|
Use: "convert $path...",
|
|
Short: "convert to JSON/YAML/gotest/pytest testcases",
|
|
Args: cobra.MinimumNArgs(1),
|
|
PreRun: func(cmd *cobra.Command, args []string) {
|
|
setLogLevel(logLevel)
|
|
},
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
var flagCount int
|
|
var outputType convert.OutputType
|
|
if toJSONFlag {
|
|
flagCount++
|
|
}
|
|
if toYAMLFlag {
|
|
flagCount++
|
|
outputType = convert.OutputTypeYAML
|
|
}
|
|
if toGoTestFlag {
|
|
flagCount++
|
|
outputType = convert.OutputTypeGoTest
|
|
}
|
|
if toPyTestFlag {
|
|
flagCount++
|
|
outputType = convert.OutputTypePyTest
|
|
|
|
packages := []string{
|
|
fmt.Sprintf("httprunner==%s", version.VERSION),
|
|
}
|
|
_, err := builtin.EnsurePython3Venv(venv, packages...)
|
|
if err != nil {
|
|
log.Error().Err(err).Msg("python3 venv is not ready")
|
|
return err
|
|
}
|
|
}
|
|
if flagCount > 1 {
|
|
return errors.New("please specify at most one conversion flag")
|
|
}
|
|
convert.Run(outputType, outputDir, profilePath, args)
|
|
return nil
|
|
},
|
|
}
|
|
|
|
var (
|
|
toJSONFlag bool
|
|
toYAMLFlag bool
|
|
toGoTestFlag bool
|
|
toPyTestFlag bool
|
|
outputDir string
|
|
profilePath string
|
|
)
|
|
|
|
func init() {
|
|
rootCmd.AddCommand(convertCmd)
|
|
convertCmd.Flags().BoolVar(&toPyTestFlag, "to-pytest", false, "convert to pytest scripts")
|
|
convertCmd.Flags().BoolVar(&toGoTestFlag, "to-gotest", false, "convert to gotest scripts (TODO)")
|
|
convertCmd.Flags().BoolVar(&toJSONFlag, "to-json", false, "convert to JSON scripts (default)")
|
|
convertCmd.Flags().BoolVar(&toYAMLFlag, "to-yaml", false, "convert to YAML scripts")
|
|
convertCmd.Flags().StringVarP(&outputDir, "output-dir", "d", "", "specify output directory, default to the same dir with har file")
|
|
convertCmd.Flags().StringVarP(&profilePath, "profile", "p", "", "specify profile path to override headers and cookies")
|
|
}
|