单元测试的原则,就是你所测试的函数方法,不要受到所依赖环境的影响,比如网络访问等,因为有时候我们运行单元测试的时候,并没有联网,那么总不能让单元测试因为这个失败吧?所以这时候模拟网络访问就有必要了。

对于go的web应用程序中往往需要与其他系统进行交互, 比如通过http访问其他系统, 此时就需要一种方法用于打桩来模拟Web服务端和客户端,httptest包即Go语言针对Web应用提供的解决方案。

httptest 可以方便的模拟各种Web服务器和客户端,以达到测试的目的。

基本使用

假设在server中handler已经写好: main_test.go

package main

import (
"io"
"log"
"net/http"
) func HealthCheckHandler(w http.ResponseWriter, r *http.Request) {
// A very simple health check.
w.WriteHeader(http.StatusOK)
w.Header().Set("Content-Type", "application/json") // In the future we could report back on the status of our DB, or our cache
// (e.g. Redis) by performing a simple PING, and include them in the response.
_, err := io.WriteString(w, `{"alive": true}`)
if err != nil {
log.Printf("reponse err ")
}
} func main() {
// 路由与视图函数绑定
http.HandleFunc("/health-check", HealthCheckHandler) // 启动服务,监听地址
err := http.ListenAndServe(":9999", nil)
if err != nil {
log.Fatal(err)
}
}

测试代码如下:

main.go

import (
"net/http"
"net/http/httptest"
"testing"
) func TestHealthCheckHandler(t *testing.T) {
//创建一个请求
req, err := http.NewRequest("GET", "/health-check", nil)
if err != nil {
t.Fatal(err)
} // 我们创建一个 ResponseRecorder (which satisfies http.ResponseWriter)来记录响应
rr := httptest.NewRecorder() //直接使用HealthCheckHandler,传入参数rr,req
HealthCheckHandler(rr, req) // 检测返回的状态码
if status := rr.Code; status != http.StatusOK {
t.Errorf("handler returned wrong status code: got %v want %v",
status, http.StatusOK)
} // 检测返回的数据
expected := `{"alive": true}`
if rr.Body.String() != expected {
t.Errorf("handler returned unexpected body: got %v want %v",
rr.Body.String(), expected)
}
}

在不启动服务的情况下即可web接口的测试

# go test -v main_test.go main.go

# 输出
root@failymao:/mnt/d/gopath/httptest# go test -v main_test.go main.go
=== RUN TestHealthCheckHandler
--- PASS: TestHealthCheckHandler (0.00s)
PASS
ok command-line-arguments 0.031s

运行这个单元测试,就可以看到访问/health-check的结果里,并且我们没有启动任何HTTP服务就达到了目的。这个主要利用httptest.NewRecorder()创建一个http.ResponseWriter,模拟了真实服务端的响应,这种响应时通过调用http.DefaultServeMux.ServeHTTP方法触发的。

扩展使用

如果Web Server有操作数据库的行为,需要在init函数中进行数据库的连接。

参考官方文档中的样例编写的另外一个测试代码:

func TestHealthCheckHandler2(t *testing.T) {
reqData := struct {
Info string `json:"info"`
}{Info: "P123451"} reqBody, _ := json.Marshal(reqData)
fmt.Println("input:", string(reqBody))
// 使用httptes.NewRequest请求接口
req := httptest.NewRequest(
http.MethodPost,
"/health-check",
bytes.NewReader(reqBody),
) req.Header.Set("userid", "wdt")
req.Header.Set("commpay", "brk") // 使用server,返回的是reponser
rr := httptest.NewRecorder()
HealthCheckHandler(rr, req) result := rr.Result() body, _ := ioutil.ReadAll(result.Body)
fmt.Println(string(body)) if result.StatusCode != http.StatusOK {
t.Errorf("expected status 200,",result.StatusCode)
}
}

不同的地方:

  • http.NewRequest替换为httptest.NewRequest。
  • httptest.NewRequest的第三个参数可以用来传递body数据,必须实现io.Reader接口。
  • httptest.NewRequest不会返回error,无需进行err!=nil检查。
  • 解析响应时没直接使用ResponseRecorder,而是调用了Result函数。

接口context使用

代码如下

func TestGetProjectsHandler(t *testing.T) {
req, err := http.NewRequest("GET", "/api/users", nil)
if err != nil {
t.Fatal(err)
} rr := httptest.NewRecorder()
// e.g. func GetUsersHandler(ctx context.Context, w http.ResponseWriter, r *http.Request)
handler := http.HandlerFunc(GetUsersHandler) // Populate the request's context with our test data.
ctx := req.Context()
ctx = context.WithValue(ctx, "app.auth.token", "abc123")
ctx = context.WithValue(ctx, "app.user",
&YourUser{ID: "qejqjq", Email: "user@example.com"}) // Add our context to the request: note that WithContext returns a copy of
// the request, which we must assign.
req = req.WithContext(ctx)
handler.ServeHTTP(rr, req) // Check the status code is what we expect.
if status := rr.Code; status != http.StatusOK {
t.Errorf("handler returned wrong status code: got %v want %v",
status, http.StatusOK)
}
}

模拟调用

还有一个模拟调用的方式,是真的在测试机上模拟一个服务器,然后进行调用测试。

将上面的代码进行改造

main.go

package main

import (
"encoding/json"
"log"
"net/http"
) type Response struct {
Code int64 `json:"code"`
Data map[string]interface{} `json:"data"`
Msg string `json:"msg"`
} // json 返回
func HealthCheckHandler(w http.ResponseWriter, r *http.Request) {
// A very simple health check.
response := Response{
Code: 200,
Msg: "ok",
Data: map[string]interface{}{"alive": true},
}
// In the future we could report back on the status of our DB, or our cache
// (e.g. Redis) by performing a simple PING, and include them in the response.
res, err := json.Marshal(response)
// _, err := io.WriteString(w, `{"alive": true}`)
if err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
log.Printf("reponse err ")
return
}
w.WriteHeader(http.StatusOK)
w.Header().Set("Content-Type", "application/json") _, err = w.Write(res)
if err != nil {
log.Printf("reponse err ")
}
} func main() {
// 路由与视图函数绑定
http.HandleFunc("/health-check", HealthCheckHandler) // 启动服务,监听地址
err := http.ListenAndServe(":9999", nil)
if err != nil {
log.Fatal(err)
}
}

main_mock_test.go

package main

import (
"encoding/json"
"io/ioutil"
"log"
"net/http"
"net/http/httptest"
"testing"
) func mockServer() *httptest.Server {
// API调用处理函数
healthHandler := func(rw http.ResponseWriter, r *http.Request) {
response := Response{
Code: 200,
Msg: "ok",
Data: map[string]interface{}{"alive": true},
} rw.Header().Set("Content-Type", "application/json")
rw.WriteHeader(http.StatusOK) _ = json.NewEncoder(rw).Encode(response)
} // 适配器转换
return httptest.NewServer(http.HandlerFunc(healthHandler))
} func TestHealthCheck3(t *testing.T) {
// 创建一个模拟的服务器
server := mockServer()
defer server.Close() // Get请求发往模拟服务器的地址
request, err := http.Get(server.URL)
if err != nil {
t.Fatal("创建Get失败")
}
defer request.Body.Close() log.Println("code:", request.StatusCode)
js, err := ioutil.ReadAll(request.Body)
if err != nil {
log.Fatal(err)
} log.Printf("body:%s\n", js)
}

模拟服务器的创建使用的是httptest.NewServer函数,它接收一个http.Handler处理API请求的接口。 代码示例中使用了Hander的适配器模式,http.HandlerFunc是一个函数类型,实现了http.Handler接口,这里是强制类型转换,不是函数的调用

执行测试:

# 指令
go test -v main_mock_test.go main.go # 输出
=== RUN TestHealthCheck3
2021/07/22 23:20:27 code: 200
2021/07/22 23:20:27 body:{"code":200,"data":{"alive":true},"msg":"ok"} --- PASS: TestHealthCheck3 (0.01s)
PASS
ok command-line-arguments 0.032s

测试覆盖率

尽可能的模拟更多的场景来测试我们代码的不同情况,但是有时候的确也有忘记测试的代码,这时候我们就需要测试覆盖率作为参考了。

由单元测试的代码,触发运行到的被测试代码的代码行数占所有代码行数的比例,被称为测试覆盖率,代码覆盖率不一定完全精准,但是可以作为参考,可以帮我们测量和我们预计的覆盖率之间的差距。

main.go

func Tag(tag int){
switch tag {
case 1:
fmt.Println("Android")
case 2:
fmt.Println("Go")
case 3:
fmt.Println("Java")
default:
fmt.Println("C") }
}

main_test

func TestTag(t *testing.T) {
Tag(1)
Tag(2) }

使用go test工具运行单元测试,和前几次不一样的是 ,要显示测试覆盖率,所以要多加一个参数-coverprofile,所以完整的命令为:go test -v -coverprofile=c.out ,-coverprofile是指定生成的覆盖率文件,例子中是c.out,这个文件一会我们会用到。现在看终端输出,已经有了一个覆盖率。

# 执行
$ go test -v -coverprofile=c.out main_test.go main.go # 输出
=== RUN TestTag
Android
Go
--- PASS: TestTag (0.00s)
PASS
coverage: 60.0% of statements
ok command-line-arguments 0.005s coverage: 60.0% of statements

coverage: 60.0% of statements,60%的测试覆盖率,还没有到100%

那么看看还有那些代码没有被测试到。

这就需要我们刚刚生成的测试覆盖率文件c.out生成测试覆盖率报告了。生成报告有go为我们提供的工具,使用go tool cover -html=c.out -o=tag.html ,即可生成一个名字为tag.html的HTML格式的测试覆盖率报告,这里有详细的信息告诉我们哪一行代码测试到了,哪一行代码没有测试到。

上图中可以看到,标记为绿色的代码行已经被测试了;标记为红色的还没有测试到,有2行的,现在我们根据没有测试到的代码逻辑,完善我的单元测试代码即可。

func TestTag(t *testing.T) {
Tag(1)
Tag(2)
Tag(3)
Tag(6) }

单元测试完善为如上代码,再运行单元测试,就可以看到测试覆盖率已经是100%了,大功告成。

参考

httptest 参考

Go进阶--httptest的更多相关文章

  1. nodejs进阶(6)—连接MySQL数据库

    1. 建库连库 连接MySQL数据库需要安装支持 npm install mysql 我们需要提前安装按mysql sever端 建一个数据库mydb1 mysql> CREATE DATABA ...

  2. nodejs进阶(4)—读取图片到页面

    我们先实现从指定路径读取图片然后输出到页面的功能. 先准备一张图片imgs/dog.jpg. file.js里面继续添加readImg方法,在这里注意读写的时候都需要声明'binary'.(file. ...

  3. JavaScript进阶之路(一)初学者的开始

    一:写在前面的问题和话 一个javascript初学者的进阶之路! 背景:3年后端(ASP.NET)工作经验,javascript水平一般般,前端水平一般般.学习资料:犀牛书. 如有误导,或者错误的地 ...

  4. nodejs进阶(3)—路由处理

    1. url.parse(url)解析 该方法将一个URL字符串转换成对象并返回. url.parse(urlStr, [parseQueryString], [slashesDenoteHost]) ...

  5. nodejs进阶(5)—接收请求参数

    1. get请求参数接收 我们简单举一个需要接收参数的例子 如果有个查找功能,查找关键词需要从url里接收,http://localhost:8000/search?keyword=地球.通过前面的进 ...

  6. nodejs进阶(1)—输出hello world

    下面将带领大家一步步学习nodejs,知道怎么使用nodejs搭建服务器,响应get/post请求,连接数据库等. 搭建服务器页面输出hello world var  http  =  require ...

  7. [C#] 进阶 - LINQ 标准查询操作概述

    LINQ 标准查询操作概述 序 “标准查询运算符”是组成语言集成查询 (LINQ) 模式的方法.大多数这些方法都在序列上运行,其中的序列是一个对象,其类型实现了IEnumerable<T> ...

  8. Java 进阶 hello world! - 中级程序员之路

    Java 进阶 hello world! - 中级程序员之路 Java是一种跨平台的语言,号称:"一次编写,到处运行",在世界编程语言排行榜中稳居第二名(TIOBE index). ...

  9. C#进阶系列——WebApi 接口返回值不困惑:返回值类型详解

    前言:已经有一个月没写点什么了,感觉心里空落落的.今天再来篇干货,想要学习Webapi的园友们速速动起来,跟着博主一起来学习吧.之前分享过一篇 C#进阶系列——WebApi接口传参不再困惑:传参详解  ...

随机推荐

  1. Java 获取、删除Word文本框中的表格

    本文介绍如何来获取Word文本框中包含的表格,以及删除表格. 程序测试环境包括: IDEA JDK 1.8.0 Spire.Doc.jar 注:jar导入,可通过创建Maven程序项目,并在pom.x ...

  2. IO流之节点流(字节流)

    节点流可以分为:字节节点流和字符节点流 数据源直接到程序的成为节点流(低级流) 字节流 输入流----InputStream InputStream 是输入流的抽象父类,若创建对象,需new它的子类 ...

  3. go反射原理

    go反射原理 本文基于go1.13.15 1.go汇编 1.1 基本语法 go采用plan9的汇编器完成汇编,有下面几个重要的伪寄存器 FP: Frame pointer: 局部变量访问 PC: Pr ...

  4. informix数据库分页

    需求描述 当查询结果返回大量数据情况下,比如报表查询.需要按一定条件排序提供分页呈现数据. INFORMIX实现方案:Informix 数据库提供了非常便捷.高效的SQL. SELECT SKIP M ...

  5. 使用Visual Studio进行文件差异比较

    启动VS自带的文件差异比较工具,进行代码文本比较,省去安装第三方工具的麻烦. 一.启动VS命令窗口. 依次点击菜单[视图]>>[其它窗口]>>[命令窗口],如下图所示,启动命令 ...

  6. Vue--el-menu 的自动跳转功能与自己的click事件冲突

    一\先看elementUI说明 项目实际 此时点击活导航时以 index 作为 path 进行路由跳转 那么此时不要onclik事件了 如果此时有在有click 就

  7. SPRING.NET FRAMEWORK 3.0 GA啦

    THE SPRING.NET FRAMEWORK http://www.springframework.net/ 1. INTRODUCTION Spring.NET contains: A full ...

  8. C++ //函数调用运算符重载 (仿函数)

    1 //函数调用运算符重载 2 3 #include <iostream> 4 #include <string> 5 using namespace std; 6 7 //函 ...

  9. 只要套路对,薪资直接翻一倍!保姆级Android面试葵花宝典,肝完面试犹如开挂

    跳槽,这在 IT 互联网圈是非常普遍的,也是让自己升职加薪,走上人生巅峰的重要方式.那么作为一个普通的Android程序猿,我们如何才能斩获大厂offer 呢? 疫情向好.面试在即,还在迷茫踌躇中的后 ...

  10. Python之replace()方法失效

    1.背景 Titanic存活率预测案例: # 读取数据 df_train = pd.read_csv("./data/train.csv") df_train.head() OUT ...