feat: support set debug for boomer

This commit is contained in:
debugtalk
2021-09-28 15:48:53 +08:00
parent 1c770110d5
commit 6cb2a92b7a

View File

@@ -13,23 +13,32 @@ func Run(testcases ...*TestCase) {
}
func NewBoomer() *Boomer {
return &Boomer{}
return &Boomer{
debug: false,
}
}
type Boomer struct {
debug bool
}
func (b *Boomer) SetDebug(debug bool) *Boomer {
b.debug = debug
return b
}
func (b *Boomer) Run(testcases ...*TestCase) {
var taskSlice []*boomer.Task
for _, testcase := range testcases {
task := convertBoomerTask(testcase)
task := b.convertBoomerTask(testcase)
taskSlice = append(taskSlice, task)
}
boomer.Run(taskSlice...)
}
func convertBoomerTask(testcase *TestCase) *boomer.Task {
func (b *Boomer) convertBoomerTask(testcase *TestCase) *boomer.Task {
runner := NewRunner()
runner = runner.SetDebug(b.debug)
return &boomer.Task{
Name: testcase.Config.Name,
Weight: testcase.Config.Weight,