add NewRequestForTest

sjqzhang · Jul 6, 2022 · ece674a · ece674a
1 parent 9fa3de5
commit ece674a
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 3 deletions.
diff --git a/gmock.go b/gmock.go
@@ -1,11 +1,13 @@
 package gmock
 
 import (
-	"github.com/sjqzhang/requests"
 	"github.com/sjqzhang/gmock/mockdb"
 	"github.com/sjqzhang/gmock/mockhttp"
 	"github.com/sjqzhang/gmock/mockredis"
 	"github.com/sjqzhang/gmock/util"
+	"github.com/sjqzhang/requests"
+	"net/http"
+	"net/http/httptest"
 )
 
 func NewMockHttpServer(httpServerPort int, mockJSONDir string, allowProxyHosts []string) *mockhttp.MockHttpServer {
@@ -33,10 +35,17 @@ func NewDBUtil() *util.DBUtil {
 }
 
 func Get(origurl string, args ...interface{}) (resp *requests.Response, err error) {
-
 	return requests.Get(origurl, args...)
 }
 
 func Post(origurl string, args ...interface{}) (resp *requests.Response, err error) {
 	return requests.Post(origurl, args...)
 }
+
+func NewRecorder() *httptest.ResponseRecorder {
+	return requests.NewRecorder()
+}
+
+func NewRequestForTest(method, origurl string, args ...interface{}) (*http.Request, error) {
+	return requests.NewRequestForTest(method,origurl,args...)
+}
diff --git a/go.mod b/go.mod
@@ -36,7 +36,7 @@ require (
 	github.com/pmezard/go-difflib v1.0.0
 	github.com/sirupsen/logrus v1.8.1
 	github.com/sjqzhang/goutil v0.0.2
-	github.com/sjqzhang/requests v1.0.0
+	github.com/sjqzhang/requests v1.0.4
 	github.com/stretchr/testify v1.7.0
 	github.com/syndtr/goleveldb v1.0.0
 	github.com/yuin/gopher-lua v0.0.0-20210529063254-f4c35e4016d9