From aae384e597f98e5c21b151ccee9fb4f1431f8fc9 Mon Sep 17 00:00:00 2001
From: yedf2 <120050102@qq.com>
Date: Sat, 1 Jan 2022 11:37:48 +0800
Subject: [PATCH] cover OnAfterResponse

---
 test/main_test.go | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/test/main_test.go b/test/main_test.go
index 31bf4be..2fa6a1c 100644
--- a/test/main_test.go
+++ b/test/main_test.go
@@ -17,6 +17,7 @@ import (
 	"github.com/dtm-labs/dtm/dtmsvr"
 	"github.com/dtm-labs/dtm/dtmsvr/config"
 	"github.com/dtm-labs/dtm/test/busi"
+	"github.com/go-resty/resty/v2"
 )
 
 func exitIf(code int) {
@@ -36,6 +37,7 @@ func TestMain(m *testing.M) {
 
 	dtmgrpc.AddUnaryInterceptor(busi.SetGrpcHeaderForHeadersYes)
 	dtmcli.OnBeforeRequest(busi.SetHttpHeaderForHeadersYes)
+	dtmcli.OnAfterResponse(func(c *resty.Client, resp *resty.Response) error { return nil })
 
 	tenv := os.Getenv("TEST_STORE")
 	if tenv == "boltdb" {