79 lines
1.9 KiB
Go
79 lines
1.9 KiB
Go
|
package e2e
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"fmt"
|
|||
|
|
"log"
|
|||
|
|
"os"
|
|||
|
|
"path/filepath"
|
|||
|
|
"regexp"
|
|||
|
|
|
|||
|
|
. "github.com/onsi/ginkgo"
|
|||
|
|
|
|||
|
|
"github.com/tongjingran/copy"
|
|||
|
|
"gopkg.in/yaml.v3"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
type Sample struct {
|
|||
|
|
Dir string `yaml:"dir"`
|
|||
|
|
Description string `yaml:"description"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// SamplesMgr create and return sample for test case
|
|||
|
|
//
|
|||
|
|
// ginkgo 的运行模型是多进程模型，每一个 test case 是一个独立的进程
|
|||
|
|
// Ginkgo has support for running specs in parallel. It does this by spawning separate go test processes and serving specs to each process off of a shared queue.
|
|||
|
|
// 所以这里会单独在一个临时目录中生成 sample，以便将来测试用例可以并发执行
|
|||
|
|
type SamplesMgr struct {
|
|||
|
|
Samples map[string]Sample `yaml:"samples"`
|
|||
|
|
path string `yaml:"-"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func NewSamplesMgr() *SamplesMgr {
|
|||
|
|
path, _ := os.Getwd()
|
|||
|
|
metaData, err := os.ReadFile(filepath.Join(path, "samples", "meta.yaml"))
|
|||
|
|
if err != nil {
|
|||
|
|
log.Fatalf("fail to read sample meta")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
mgr := SamplesMgr{}
|
|||
|
|
err = yaml.Unmarshal(metaData, &mgr)
|
|||
|
|
if err != nil {
|
|||
|
|
log.Fatalf("fail to parse the meta yaml")
|
|||
|
|
}
|
|||
|
|
mgr.path = path
|
|||
|
|
return &mgr
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// GetSampleByKey return the sample folder location for test use
|
|||
|
|
func (m *SamplesMgr) GetSampleByKey(key string) (string, error) {
|
|||
|
|
sample, ok := m.Samples[key]
|
|||
|
|
if !ok {
|
|||
|
|
return "", fmt.Errorf("no sample found")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
desc := CurrentGinkgoTestDescription()
|
|||
|
|
caseTitle := desc.FullTestText + " " + key
|
|||
|
|
|
|||
|
|
m1 := regexp.MustCompile(`[/\\?%*:|"<>]`)
|
|||
|
|
caseTitle = m1.ReplaceAllString(caseTitle, "-")
|
|||
|
|
|
|||
|
|
dst := filepath.Join(m.path, "tmp", caseTitle)
|
|||
|
|
err := os.RemoveAll(dst)
|
|||
|
|
if err != nil {
|
|||
|
|
log.Fatalf("fail to clean the temp sample: %v", err)
|
|||
|
|
}
|
|||
|
|
err = os.MkdirAll(dst, os.ModePerm)
|
|||
|
|
if err != nil {
|
|||
|
|
log.Fatalf("fail to create sample dir: %v", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
src := filepath.Join(m.path, "samples", sample.Dir)
|
|||
|
|
|
|||
|
|
err = copy.Copy(src, dst)
|
|||
|
|
if err != nil {
|
|||
|
|
log.Fatalf("fail to copy the sample project: %v", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return dst, nil
|
|||
|
|
}
|