前言
前幾話主要講解關于使用golang進行單元測試，在單元測試的上一層就是接口測試，本節主要講使用golang進行接口測試，其中主要以http協議的接口測試來講解
golang中的http請求
golang中擁有一個原生的http依賴庫：net/http，http服務器的建立還是http客戶端的開發，都會使用到這個依賴庫，這里主要講解時client部分，作為請求發起方應用于日常的接口測試，例示代碼如下：
get請求
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
|
package mainimport ( "fmt" "io/ioutil" "net/http")func main() { //模擬一個get提交請求 resp, err := http.Get("http://127.0.0.1:12345/checkon") if err != nil { panic(err) } defer resp.Body.Close() //關閉連接 body, err := ioutil.ReadAll(resp.Body) //讀取body的內容 fmt.Println(string(body))} |
返回結果
|
1
2
3
4
5
6
7
|
E:\go_project>go run testget.go{ "code": 200, "data": "", "msg": "online", "state": "success"} |
post請求：
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
|
package mainimport ( "fmt" "io/ioutil" "net/http" "strings")func main() { //模擬一個post提交請求 resp, err := http.Post("http://www.baidu.com", "application/x-www-form-urlencoded", strings.NewReader("id=1")) if err != nil { panic(err) } //關閉連接 defer resp.Body.Close() //讀取報文中所有內容 body, err := ioutil.ReadAll(resp.Body) //輸出內容 fmt.Println(string(body))} |
上面的post請求以form的方式，最后會返回一個頁面
這里說明一下以下這行代碼
|
1
|
defer resp.Body.Close() |
首先是defer， Go的defer語句用來調度一個函數調用（被延期的函數），使其在執行defer的函數即將返回之前才被運行,被延期執行的函數，它的參數（包括接受者）實在defer執行的時候被求值的，而不是在調用執行的時候。也就是說被延期執行的函數的參數是按正常順序被求值的，簡單理解為，無論defer對應的代碼行放在代碼段的哪個位置，defer是在return前執行的代碼行，但defer代碼行中的參數是需要先聲明再調用的，對應響應中的處理，golang的Response.Body需要被關閉的，body實際上是一個嵌套了多層的net.TCPConn：
- bufio.Reader，這層嘗試將多次小的讀操作替換為一次大的讀操作，減少系統調用的次數，提高性能；
- io.LimitedReader，tcp連接在讀取完body后不會關閉，繼續讀會導致阻塞，所以需要LimitedReader在body讀完后發出eof終止讀取；
- chunkedReader，解析chunked格式編碼（如果不是chunked略過）；
- bodyEOFSignal，在讀到eof，或者是提前關閉body時會對readLoop發出回收連接的通知；
- gzipReader，解析gzip壓縮（如果不是gizp壓縮略過）；
從上面可以看出如果body既沒有被完全讀取，也沒有被關閉，那么這次http事務就沒有完成，除非連接因超時終止了，否則相關資源無法被回收，所以需要我們進行關閉連接的操作，這個是很多golang新手會忽略的一個點，作為client端處理response的時候，body一定要close，否則會造成GC回收不到，繼而產生內存泄露
帶json的post請求
我們大部分應用到的restful接口都是用json格式的請求體，對應的golang的http請求也會有相關的方式post json請求體
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
|
package mainimport ( "fmt" "io/ioutil" "net/http" "bytes" "encoding/json")type HttpData struct { Flag int `json:"flag"` Msg string `json:"msg"`}func main() { url := "http://127.0.0.1:12345/postdata" contentType := "application/json;charset=utf-8" var httpdata HttpData httpdata.Flag = 1 httpdata.Msg = "terrychow" b ,err := json.Marshal(httpdata) if err != nil { fmt.Println("json format error:", err) return } body := bytes.NewBuffer(b) resp, err := http.Post(url, contentType, body) if err != nil { fmt.Println("Post failed:", err) return } defer resp.Body.Close() content, err := ioutil.ReadAll(resp.Body) if err != nil { fmt.Println("Read failed:", err) return } fmt.Println("header:", resp.Header) fmt.Println("content:", string(content))} |
執行結果響應
|
1
2
3
4
5
6
7
8
|
E:\go_project>go run gohttptest.goheader: map[Content-Type:[application/json] Content-Length:[78] Server:[Werkzeug/0.14.1 Python/2.7.15] Date:[Thu, 06 Dec 2018 16:35:11 GMT]]content: { "code": 200, "data": 1, "msg": "terrychow", "state": "success"} |
對于常用的get和post請求基本上就以照上面的版本執行，當然我們現在需要做的是http接口的測試，那就需要引入測試框架進行相關的校驗，本文先講解用之前提到的gocheck來進行斷言
golang中的http接口測試
引入gocheck之后我們得到了以下的腳本：
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
|
package hello_testimport ( "testing" "fmt" "strconv" "io/ioutil" "net/http" "bytes" "encoding/json" . "gopkg.in/check.v1")var a int =1// Hook up gocheck into the "go test" runner.func Test(t *testing.T) { TestingT(t) }type MySuite struct{}type HttpData struct { Flag int `json:"flag"` Msg string `json:"msg"`}var _ = Suite(&MySuite{})var testurl string ="http://127.0.0.1:12345"func (s *MySuite) SetUpSuite(c *C) { str3:="第1次套件開始執行" fmt.Println(str3) //c.Skip("Skip TestSutie")}func (s *MySuite) TearDownSuite(c *C) { str4:="第1次套件執行完成" fmt.Println(str4)}func (s *MySuite) SetUpTest(c *C) { str1:="第"+strconv.Itoa(a)+"條用例開始執行" fmt.Println(str1)}func (s *MySuite) TearDownTest(c *C) { str2:="第"+strconv.Itoa(a)+"條用例執行完成" fmt.Println(str2) a=a+1}func (s *MySuite) TestHttpGet(c *C) { geturl := fmt.Sprintf("%v/checkon", testurl) respget, err := http.Get(geturl) if err != nil { panic(err) } defer respget.Body.Close() //關閉連接 body, err := ioutil.ReadAll(respget.Body) //讀取body的內容 var gdat map[string]interface{} //定義map用于解析resp.body的內容 if err := json.Unmarshal([]byte(string(body)), &gdat); err == nil { fmt.Println(gdat) } else { fmt.Println(err) } var gmsg=gdat["msg"] c.Assert(gmsg, Equals, "terrychow") //模擬失敗的斷言}func (s *MySuite) TestHttpPost(c *C) { url := fmt.Sprintf("%v/postdata", testurl) contentType := "application/json;charset=utf-8" var httpdata HttpData httpdata.Flag = 1 httpdata.Msg = "terrychow" b ,err := json.Marshal(httpdata) if err != nil { fmt.Println("json format error:", err) return } body := bytes.NewBuffer(b) resp, err := http.Post(url, contentType, body) if err != nil { fmt.Println("Post failed:", err) return } defer resp.Body.Close() content, err := ioutil.ReadAll(resp.Body) if err != nil { fmt.Println("Read failed:", err) return } var dat map[string]interface{} //定義map用于解析resp.body的內容 if err := json.Unmarshal([]byte(string(content)), &dat); err == nil { fmt.Println(dat) } else { fmt.Println(err) } var msg=dat["msg"] c.Assert(msg, Equals, "terrychow") //模擬成功的斷言} |
最后的輸出內容：
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
|
E:\go_project>go test -v gocheckhttp_test.go=== RUN Test第1次套件開始執行第1條用例開始執行map[code:200 data: msg:online state:success]第1條用例執行完成----------------------------------------------------------------------FAIL: gocheckhttp_test.go:56: MySuite.TestHttpGetgocheckhttp_test.go:72: c.Assert(gmsg, Equals, "terrychow")... obtained string = "online"... expected string = "terrychow"第2條用例開始執行map[msg:terrychow state:success code:200 data:1]第2條用例執行完成第1次套件執行完成OOPS: 1 passed, 1 FAILED--- FAIL: Test (0.02s)FAILFAIL command-line-arguments 0.613s |
輸出的結果符合預期，這也是比較基本的http接口測試
小結
就上文來說，我們基本可以通過本文掌握如何做http接口測試，其核心還是使用http依賴庫發出請求獲取響應，利用gocheck進行斷言，當然還可以用testing，下一節繼續講一下http接口測試，但會重點講專門做http接口測試的測試框架httpexpect以及用于mock的httptest，希望對大家的學習有所幫助，也希望大家多多支持服務器之家。
原文鏈接：https://studygolang.com/articles/16742
