feat: support set debug for boomer

This commit is contained in:
debugtalk
2021-09-28 15:48:53 +08:00
parent 1c770110d5
commit 6cb2a92b7a
+12 -3
View File
@@ -13,23 +13,32 @@ func Run(testcases ...*TestCase) {
} }
func NewBoomer() *Boomer { func NewBoomer() *Boomer {
return &Boomer{} return &Boomer{
debug: false,
}
} }
type Boomer struct { type Boomer struct {
debug bool
}
func (b *Boomer) SetDebug(debug bool) *Boomer {
b.debug = debug
return b
} }
func (b *Boomer) Run(testcases ...*TestCase) { func (b *Boomer) Run(testcases ...*TestCase) {
var taskSlice []*boomer.Task var taskSlice []*boomer.Task
for _, testcase := range testcases { for _, testcase := range testcases {
task := convertBoomerTask(testcase) task := b.convertBoomerTask(testcase)
taskSlice = append(taskSlice, task) taskSlice = append(taskSlice, task)
} }
boomer.Run(taskSlice...) boomer.Run(taskSlice...)
} }
func convertBoomerTask(testcase *TestCase) *boomer.Task { func (b *Boomer) convertBoomerTask(testcase *TestCase) *boomer.Task {
runner := NewRunner() runner := NewRunner()
runner = runner.SetDebug(b.debug)
return &boomer.Task{ return &boomer.Task{
Name: testcase.Config.Name, Name: testcase.Config.Name,
Weight: testcase.Config.Weight, Weight: testcase.Config.Weight,