Go语言test之类方法测试

时间:2021-05-27 14:41:56

Go语言提供了完善的单元测试支持,开发人员可以方便的编写测试代码,保证自己代码的质量。在目前的例子中,一般看到都是普通函数的例子。下面我将通过类方法的单元测试例子来展示一下Go语言的魅力。

首先是代码所在的文件xml.go:

package myxml

import (
"encoding/xml"
) type XMLRsq struct {
XMLName xml.Name `xml:"response"`
ResCode string `xml:"res_code"`
ResMessage string `xml:"res_message"`
} func (r *XMLRsq ) ToString() (string, error) {
b, err := r.ToBytes()
return string(b), err
} func (r *XMLRsq ) ToBytes() ([]byte, error) {
b, err := xml.Marshal(r)
if err != nil {
return b, err
} b = append([]byte(xml.Header), b...) return b, err
} func (r *XMLRsq ) Parse(b []byte) error {
return xml.Unmarshal(b, r)
}

接着编写单元测试代码,注意单元测试代码应和被测试的代码在同一个包,且应使用xxx_test.go的规则来命名测试代码所在的文件,例如对上面的代码文件,应将测试文件命名为xml_test.go,包括以下的代码:

package myxml

import "fmt"
import "encoding/xml"
import "testing" func Test_XMLRsp_ToString(t *testing.T) {
in := XMLRsp{xml.Name{Space: "", Local: "XMLRsp"}, "1", "1", "1"}
out := []byte(xml.Header)
out = append(out, []byte(`<XMLRsp><res_code>1</res_code><res_message>1</res_message><can_use>1</can_use></XMLRsp>`)...) r := in b, _ := r.ToString()
if b != string(out) {
t.Errorf("XMLRsp_ToString failed, result is: [%s]\n", b)
fmt.Printf("Expectation is: [%s]\n", out)
} else {
fmt.Printf("XMLRsp_ToString result is: [%s]\n", b)
}
} func Test_XMLRsp_Parse(t *testing.T) {
in := []byte(`<XMLRsp><res_code>1</res_code><res_message>1</res_message><can_use>1</can_use></XMLRsp>`)
out := XMLRsp{xml.Name{Space: "", Local: "XMLRsp"}, "1", "1", "1"} r := new(XMLRsp) _ = r.Parse(in)
if *r != out {
t.Errorf("XMLRsp_Parse failed, result is: [%s]\n", *r)
fmt.Printf("Expectation is: [%s]\n", out)
} else {
fmt.Printf("XMLRsp_Parse result is: [%s]\n", *r)
}
}

测试代码中,函数可以用如下方式命名:Test_T_M,其中T为类型名,M为方法名,这样容易区分,但这不是Go语言的强制要求。

具体测试代码里先构造了一个类XMLRsp的对象,然后通过它去调用相应的类方法,本质上与其他单元测试代码并无不同。

上面测试代码第15行,先用了一个类型转换string(out)来得到一个string类型的out表示,因为Go语言里slice之间不能直接比较。

XMLRsp类的3个方法中,ToString的执行路径已经覆盖了ToBytes,故未单独编写用例。

运行go test命令,可以得到类似如下的结果:

XMLRsp_ToString result is: [<?xml version="1.0" encoding="UTF-8"?>
<XMLRsp><res_code></res_code><res_message></res_message><can_use></can_use></XMLRsp>]
XMLRsp_Parse result is: [{{ XMLRsp} }]
PASS
ok myxml .016s

表示这次测试的结果为PASS,大功告成,你可以忙着写其他的代码去了^-^