diff --git a/boomer.go b/boomer.go index e15bc92b..0161a008 100644 --- a/boomer.go +++ b/boomer.go @@ -6,7 +6,13 @@ import ( "github.com/myzhan/boomer" ) -func HttpBoomer() *Boomer { +var defaultBoomer = NewBoomer() + +func Run(testcases ...*TestCase) { + defaultBoomer.Run(testcases...) +} + +func NewBoomer() *Boomer { return &Boomer{} } diff --git a/boomer_test.go b/boomer_test.go index 6f0a3066..20aa39f4 100644 --- a/boomer_test.go +++ b/boomer_test.go @@ -18,5 +18,5 @@ func TestHttpBoomer(t *testing.T) { }, } - HttpBoomer().Run(testcase1, testcase2) + Run(testcase1, testcase2) } diff --git a/runner.go b/runner.go index 5cc44bba..7872c1df 100644 --- a/runner.go +++ b/runner.go @@ -6,9 +6,13 @@ import ( "github.com/imroc/req" ) -var defaultRunner = HttpRunner() +var defaultRunner = NewRunner() -func HttpRunner() *Runner { +func Test(testcases ...*TestCase) error { + return defaultRunner.Run(testcases...) +} + +func NewRunner() *Runner { return &Runner{ Client: req.New(), } diff --git a/runner_test.go b/runner_test.go index acb969c9..c21c5321 100644 --- a/runner_test.go +++ b/runner_test.go @@ -31,7 +31,7 @@ func TestHttpRunner(t *testing.T) { }, } - err := HttpRunner().Run(testcase1, testcase2) + err := Test(testcase1, testcase2) if err != nil { t.Fatalf("run testcase error: %v", err) }