fix: avoid panic when running tests

This commit is contained in:
debugtalk
2022-04-06 13:34:50 +08:00
parent 399ebd00db
commit 5e40e3692d
3 changed files with 11 additions and 6 deletions

View File

@@ -1,6 +1,7 @@
package hrp
import (
"os"
"sync"
"time"
@@ -42,14 +43,16 @@ func (b *HRPBoomer) Run(testcases ...ITestCase) {
// load all testcases
testCases, err := loadTestCases(testcases...)
if err != nil {
panic(err)
log.Error().Err(err).Msg("failed to load testcases")
os.Exit(1)
}
for _, testcase := range testCases {
cfg := testcase.Config
err = initParameterIterator(cfg, "boomer")
if err != nil {
panic(err)
log.Error().Err(err).Msg("failed to init parameter iterator")
os.Exit(1)
}
rendezvousList := initRendezvous(testcase, int64(b.GetSpawnCount()))
task := b.convertBoomerTask(testcase, rendezvousList)

View File

@@ -227,14 +227,14 @@ func loadFromCSV(path string) []map[string]interface{} {
file, err := readFile(path)
if err != nil {
log.Error().Err(err).Msg("read csv file failed")
panic(err)
os.Exit(1)
}
r := csv.NewReader(strings.NewReader(string(file)))
content, err := r.ReadAll()
if err != nil {
log.Error().Err(err).Msg("parse csv file failed")
panic(err)
os.Exit(1)
}
var result []map[string]interface{}
for i := 1; i < len(content); i++ {

View File

@@ -16,7 +16,8 @@ func buildHashicorpGoPlugin() {
cmd := exec.Command("go", "build",
"-o", templatesDir+"debugtalk.bin", templatesDir+"plugin/debugtalk.go")
if err := cmd.Run(); err != nil {
panic(err)
log.Error().Err(err).Msg("build hashicorp go plugin failed")
os.Exit(1)
}
}
@@ -30,7 +31,8 @@ func buildHashicorpPyPlugin() {
pluginFile := templatesDir + "debugtalk.py"
err := scaffold.CopyFile("templates/plugin/debugtalk.py", pluginFile)
if err != nil {
panic(err)
log.Error().Err(err).Msg("build hashicorp python plugin failed")
os.Exit(1)
}
}