From 1c770110d54a61d647ae2d86ecea092c88f01fba Mon Sep 17 00:00:00 2001
From: debugtalk <mail@debugtalk.com>
Date: Tue, 28 Sep 2021 15:48:13 +0800
Subject: [PATCH] feat: support set debug for runner

---
 runner.go    | 9 ++++++++-
 step_test.go | 4 ++--
 2 files changed, 10 insertions(+), 3 deletions(-)

diff --git a/runner.go b/runner.go
index 7cee988c..d8779c23 100644
--- a/runner.go
+++ b/runner.go
@@ -17,12 +17,14 @@ func Test(t *testing.T, testcases ...*TestCase) error {
 func NewRunner() *Runner {
 	return &Runner{
 		t:      &testing.T{},
+		debug:  false, // default to turn off debug
 		Client: req.New(),
 	}
 }
 
 type Runner struct {
 	t      *testing.T
+	debug  bool
 	Client *req.Req
 }
 
@@ -31,6 +33,11 @@ func (r *Runner) WithTestingT(t *testing.T) *Runner {
 	return r
 }
 
+func (r *Runner) SetDebug(debug bool) *Runner {
+	r.debug = debug
+	return r
+}
+
 func (r *Runner) Run(testcases ...*TestCase) error {
 	for _, testcase := range testcases {
 		if err := r.runCase(testcase); err != nil {
@@ -93,7 +100,7 @@ func (r *Runner) runStepRequest(step *TStep) error {
 	}
 
 	// do request action
-	req.Debug = true
+	req.Debug = r.debug
 	resp, err := r.Client.Do(string(step.Request.Method), step.Request.URL, v...)
 	if err != nil {
 		return err
diff --git a/step_test.go b/step_test.go
index 08f0d922..09d42d31 100644
--- a/step_test.go
+++ b/step_test.go
@@ -77,10 +77,10 @@ func TestRunRequestRun(t *testing.T) {
 	config := &TConfig{
 		BaseURL: "https://postman-echo.com",
 	}
-	if err := defaultRunner.WithTestingT(t).runStep(stepGET, config); err != nil {
+	if err := defaultRunner.WithTestingT(t).SetDebug(true).runStep(stepGET, config); err != nil {
 		t.Fatalf("tStep.Run() error: %s", err)
 	}
-	if err := defaultRunner.WithTestingT(t).runStep(stepPOSTData, config); err != nil {
+	if err := defaultRunner.WithTestingT(t).SetDebug(true).runStep(stepPOSTData, config); err != nil {
 		t.Fatalf("tStepPOSTData.Run() error: %s", err)
 	}
 }