開發程序其中很重要的一點是測試,我們如何保證代碼的質量,如何保證每個函數是可運行,運行結果是正確的,又如何保證寫出來的代碼性能是好的,我們知道單元測試的重點在于發現程序設計或實現的邏輯錯誤,使問題及早暴露,便于問題的定位解決,而性能測試的重點在于發現程序設計上的一些問題,讓線上的程序能夠在高并發的情況下還能保持穩定。本小節將帶著這一連串的問題來講解Go語言中如何來實現單元測試和性能測試。
Go語言中自帶有一個輕量級的測試框架`testing`和自帶的`go test`命令來實現單元測試和性能測試,`testing`框架和其他語言中的測試框架類似,你可以基于這個框架寫針對相應函數的測試用例,也可以基于該框架寫相應的壓力測試用例,那么接下來讓我們一一來看一下怎么寫。
## [](https://github.com/astaxie/build-web-application-with-golang/blob/master/zh/11.3.md#如何編寫測試用例)如何編寫測試用例
由于`go test`命令只能在一個相應的目錄下執行所有文件,所以我們接下來新建一個項目目錄`gotest`,這樣我們所有的代碼和測試代碼都在這個目錄下。
接下來我們在該目錄下面創建兩個文件:gotest.go和gotest_test.go
1. gotest.go:這個文件里面我們是創建了一個包,里面有一個函數實現了除法運算:
~~~
package gotest
import (
"errors"
)
func Division(a, b float64) (float64, error) {
if b == 0 {
return 0, errors.New("除數不能為0")
}
return a / b, nil
}
~~~
2. gotest_test.go:這是我們的單元測試文件,但是記住下面的這些原則:
* 文件名必須是`_test.go`結尾的,這樣在執行`go test`的時候才會執行到相應的代碼
* 你必須import?`testing`這個包
* 所有的測試用例函數必須是`Test`開頭
* 測試用例會按照源代碼中寫的順序依次執行
* 測試函數`TestXxx()`的參數是`testing.T`,我們可以使用該類型來記錄錯誤或者是測試狀態
* 測試格式:`func TestXxx (t *testing.T)`,`Xxx`部分可以為任意的字母數字的組合,但是首字母不能是小寫字母[a-z],例如`Testintdiv`是錯誤的函數名。
* 函數中通過調用`testing.T`的`Error`,?`Errorf`,?`FailNow`,?`Fatal`,?`FatalIf`方法,說明測試不通過,調用`Log`方法用來記錄測試的信息。
下面是我們的測試用例的代碼:
~~~
package gotest
import (
"testing"
)
func Test_Division_1(t *testing.T) {
if i, e := Division(6, 2); i != 3 || e != nil { //try a unit test on function
t.Error("除法函數測試沒通過") // 如果不是如預期的那么就報錯
} else {
t.Log("第一個測試通過了") //記錄一些你期望記錄的信息
}
}
func Test_Division_2(t *testing.T) {
t.Error("就是不通過")
}
~~~
我們在項目目錄下面執行`go test`,就會顯示如下信息:
~~~
--- FAIL: Test_Division_2 (0.00 seconds)
gotest_test.go:16: 就是不通過
FAIL
exit status 1
FAIL gotest 0.013s
~~~
從這個結果顯示測試沒有通過,因為在第二個測試函數中我們寫死了測試不通過的代碼`t.Error`,那么我們的第一個函數執行的情況怎么樣呢?默認情況下執行`go test`是不會顯示測試通過的信息的,我們需要帶上參數`go test -v`,這樣就會顯示如下信息:
~~~
=== RUN Test_Division_1
--- PASS: Test_Division_1 (0.00 seconds)
gotest_test.go:11: 第一個測試通過了
=== RUN Test_Division_2
--- FAIL: Test_Division_2 (0.00 seconds)
gotest_test.go:16: 就是不通過
FAIL
exit status 1
FAIL gotest 0.012s
~~~
上面的輸出詳細的展示了這個測試的過程,我們看到測試函數1`Test_Division_1`測試通過,而測試函數2`Test_Division_2`測試失敗了,最后得出結論測試不通過。接下來我們把測試函數2修改成如下代碼:
~~~
func Test_Division_2(t *testing.T) {
if _, e := Division(6, 0); e == nil { //try a unit test on function
t.Error("Division did not work as expected.") // 如果不是如預期的那么就報錯
} else {
t.Log("one test passed.", e) //記錄一些你期望記錄的信息
}
}
~~~
然后我們執行`go test -v`,就顯示如下信息,測試通過了:
~~~
=== RUN Test_Division_1
--- PASS: Test_Division_1 (0.00 seconds)
gotest_test.go:11: 第一個測試通過了
=== RUN Test_Division_2
--- PASS: Test_Division_2 (0.00 seconds)
gotest_test.go:20: one test passed. 除數不能為0
PASS
ok gotest 0.013s
~~~
## [](https://github.com/astaxie/build-web-application-with-golang/blob/master/zh/11.3.md#如何編寫壓力測試)如何編寫壓力測試
壓力測試用來檢測函數(方法)的性能,和編寫單元功能測試的方法類似,此處不再贅述,但需要注意以下幾點:
* 壓力測試用例必須遵循如下格式,其中XXX可以是任意字母數字的組合,但是首字母不能是小寫字母
~~~
func BenchmarkXXX(b *testing.B) { ... }
~~~
* `go test`不會默認執行壓力測試的函數,如果要執行壓力測試需要帶上參數`-test.bench`,語法:`-test.bench="test_name_regex"`,例如`go test -test.bench=".*"`表示測試全部的壓力測試函數
* 在壓力測試用例中,請記得在循環體內使用`testing.B.N`,以使測試可以正常的運行
* 文件名也必須以`_test.go`結尾
下面我們新建一個壓力測試文件webbench_test.go,代碼如下所示:
~~~
package gotest
import (
"testing"
)
func Benchmark_Division(b *testing.B) {
for i := 0; i < b.N; i++ { //use b.N for looping
Division(4, 5)
}
}
func Benchmark_TimeConsumingFunction(b *testing.B) {
b.StopTimer() //調用該函數停止壓力測試的時間計數
//做一些初始化的工作,例如讀取文件數據,數據庫連接之類的,
//這樣這些時間不影響我們測試函數本身的性能
b.StartTimer() //重新開始時間
for i := 0; i < b.N; i++ {
Division(4, 5)
}
}
~~~
我們執行命令`go test -file webbench_test.go -test.bench=".*"`,可以看到如下結果:
~~~
PASS
Benchmark_Division 500000000 7.76 ns/op
Benchmark_TimeConsumingFunction 500000000 7.80 ns/op
ok gotest 9.364s
~~~
上面的結果顯示我們沒有執行任何`TestXXX`的單元測試函數,顯示的結果只執行了壓力測試函數,第一條顯示了`Benchmark_Division`執行了500000000次,每次的執行平均時間是7.76納秒,第二條顯示了`Benchmark_TimeConsumingFunction`執行了500000000,每次的平均執行時間是7.80納秒。最后一條顯示總共的執行時間。
## [](https://github.com/astaxie/build-web-application-with-golang/blob/master/zh/11.3.md#小結)小結
通過上面對單元測試和壓力測試的學習,我們可以看到`testing`包很輕量,編寫單元測試和壓力測試用例非常簡單,配合內置的`go test`命令就可以非常方便的進行測試,這樣在我們每次修改完代碼,執行一下go test就可以簡單的完成回歸測試了。
- 第一章 Go環境配置
- 1.1 Go安裝
- 1.2 GOPATH 與工作空間
- 1.3 Go 命令
- 1.4 Go開發工具
- 1.5 小結
- 第二章 Go語言基礎
- 2.1 你好,Go
- 2.2 Go基礎
- 2.3 流程和函數
- 2.4 struct類型
- 2.5 面向對象
- 2.6 interface
- 2.7 并發
- 2.8 總結
- 第三章 Web基礎
- 3.1 Web工作方式
- 3.2 Go搭建一個Web服務器
- 3.3 Go如何使得Web工作
- 3.4 Go的http包詳解
- 3.5 小結
- 第四章 表單
- 4.1 處理表單的輸入
- 4.2 驗證表單的輸入
- 4.3 預防跨站腳本
- 4.4 防止多次遞交表單
- 4.5 處理文件上傳
- 4.6 小結
- 第五章 訪問數據庫
- 5.1 database/sql接口
- 5.2 使用MySQL數據庫
- 5.3 使用SQLite數據庫
- 5.4 使用PostgreSQL數據庫
- 5.5 使用beedb庫進行ORM開發
- 5.6 NOSQL數據庫操作
- 5.7 小結
- 第六章 session和數據存儲
- 6.1 session和cookie
- 6.2 Go如何使用session
- 6.3 session存儲
- 6.4 預防session劫持
- 6.5 小結
- 第七章 文本處理
- 7.1 XML處理
- 7.2 JSON處理
- 7.3 正則處理
- 7.4 模板處理
- 7.5 文件操作
- 7.6 字符串處理
- 7.7 小結
- 第八章 Web服務
- 8.1 Socket編程
- 8.2 WebSocket
- 8.3 REST
- 8.4 RPC
- 8.5 小結
- 第九章 安全與加密
- 9.1 預防CSRF攻擊
- 9.2 確保輸入過濾
- 9.3 避免XSS攻擊
- 9.4 避免SQL注入
- 9.5 存儲密碼
- 9.6 加密和解密數據
- 9.7 小結
- 第十章 國際化和本地化
- 10.1 設置默認地區
- 10.2 本地化資源
- 10.3 國際化站點
- 10.4 小結
- 第十一章 錯誤處理,調試和測試
- 11.1 錯誤處理
- 11.2 使用GDB調試
- 11.3 Go怎么寫測試用例
- 11.4 小結
- 第十二章 部署與維護
- 12.1 應用日志
- 12.2 網站錯誤處理
- 12.3 應用部署
- 12.4 備份和恢復
- 12.5 小結
- 第十三章 如何設計一個Web框架
- 13.1 項目規劃
- 13.2 自定義路由器設計
- 13.3 controller設計
- 13.4 日志和配置設計
- 13.5 實現博客的增刪改
- 13.6 小結
- 第十四章 擴展Web框架
- 14.1 靜態文件支持
- 14.2 Session支持
- 14.3 表單及驗證支持
- 14.4 用戶認證
- 14.5 多語言支持
- 14.6 pprof支持
- 14.7 小結
- 附錄A 參考資料