feat: data-driven.

This commit is contained in:
徐聪
2021-12-27 15:31:49 +08:00
parent ddc0ec0a0d
commit f43450fa6e
4 changed files with 160 additions and 31 deletions

View File

@@ -2,10 +2,12 @@ package hrp
import (
"bytes"
"encoding/csv"
"encoding/json"
"fmt"
"io/ioutil"
"path/filepath"
"strings"
"github.com/rs/zerolog/log"
"gopkg.in/yaml.v3"
@@ -94,6 +96,36 @@ func loadFromYAML(path string) (*TCase, error) {
return tc, err
}
func loadFromCSV(path string) []map[string]string {
path, err := filepath.Abs(path)
if err != nil {
log.Error().Str("path", path).Err(err).Msg("convert absolute path failed")
return nil
}
log.Info().Str("path", path).Msg("load csv file")
file, err := ioutil.ReadFile(path)
if err != nil {
log.Error().Err(err).Msg("load csv file failed")
return nil
}
r := csv.NewReader(strings.NewReader(string(file)))
content, err := r.ReadAll()
if err != nil {
log.Error().Err(err).Msg("parse csv file failed")
return nil
}
var result []map[string]string
for i := 1; i < len(content); i++ {
row := make(map[string]string)
for j := 0; j < len(content[i]); j++ {
row[content[0][j]] = content[i][j]
}
result = append(result, row)
}
return result
}
func (tc *TCase) ToTestCase() (*TestCase, error) {
testCase := &TestCase{
Config: &Config{cfg: tc.Config},