在go中進(jìn)行單元測(cè)試的案例分享

更新時(shí)間：2024年01月16日 10:34:23 作者：coderyeah

這篇文章主要介紹了使用Go進(jìn)行單元測(cè)試的實(shí)現(xiàn),文中通過(guò)示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧

一. 基礎(chǔ)介紹

1. 創(chuàng)建測(cè)試文件

測(cè)試文件通常與要測(cè)試的代碼文件位于同一個(gè)包中。
測(cè)試文件的名稱應(yīng)該以 _test.go 結(jié)尾。例如，如果你要測(cè)試的文件是 math.go，那么測(cè)試文件可以命名為 math_test.go。

2. 編寫(xiě)測(cè)試函數(shù)

測(cè)試函數(shù)必須導(dǎo)入 testing 包。
每個(gè)測(cè)試函數(shù)必須以 Test 開(kāi)頭，后跟一個(gè)首字母大寫(xiě)的名字，例如 TestSum 或 TestSubtract。
測(cè)試函數(shù)的簽名應(yīng)該接受一個(gè)指向 testing.T 類型的指針：func TestXxx(t *testing.T) { ... }。

3. 使用 t 對(duì)象進(jìn)行斷言和日志記錄

t 對(duì)象用于記錄測(cè)試信息和控制測(cè)試流程。
使用 t.Error 或 t.Errorf 報(bào)告失敗，但繼續(xù)執(zhí)行當(dāng)前測(cè)試。
使用 t.Fatal 或 t.Fatalf 報(bào)告失敗并立即終止當(dāng)前測(cè)試。

4. 運(yùn)行測(cè)試

在命令行中，進(jìn)入包含測(cè)試文件的目錄。
執(zhí)行 go test 命令運(yùn)行所有測(cè)試，或使用 go test -v 以詳細(xì)模式運(yùn)行（打印每個(gè)測(cè)試的名字和運(yùn)行狀態(tài)）。
使用 go test -run 加上正則表達(dá)式來(lái)運(yùn)行特定的測(cè)試。例如，go test -run TestSum 僅運(yùn)行名為 TestSum 的測(cè)試。

示例

假設(shè)有一個(gè)名為 math.go 的文件，其中定義了一個(gè)函數(shù) Sum：

goCopy code// math.go
package math

func Sum(a, b int) int {
    return a + b
}

創(chuàng)建一個(gè)名為 math_test.go 的測(cè)試文件，其中包含以下內(nèi)容：

goCopy code// math_test.go
package math

import "testing"

func TestSum(t *testing.T) {
    total := Sum(5, 5)
    if total != 10 {
        t.Errorf("Sum was incorrect, got: %d, want: %d.", total, 10)
    }
}

然后在終端中運(yùn)行 go test 或 go test -v 來(lái)執(zhí)行測(cè)試。

二. 綜合案例

結(jié)構(gòu)

monster.go

package test_case

import (
	"encoding/json"
	"fmt"
	"os"
)

type Monster struct {
	Name  string `json:"name"`
	Age   int    `json:"age"`
	Skill string `json:"skill"`
}

// Store 將其序列化保存為文件
func (m *Monster) Store() bool {
	data, err := json.Marshal(m)
	if err != nil {
		fmt.Println("json parse Monster err ", err)
		return false
	}
	// 寫(xiě)入文件
	err = os.WriteFile("d:/monster.txt", data, 0666)
	if err != nil {
		fmt.Println("write file err ", err)
		return false
	}
	return true
}

// ReStore 反序列化文件
func (m *Monster) ReStore() bool {
	data, err := os.ReadFile("d:/monster.txt")
	if err != nil {
		fmt.Println("read file err ", err)
		return false
	}
	// 將讀取的數(shù)據(jù)進(jìn)行反序列化
	err = json.Unmarshal(data, m)
	if err != nil {
		fmt.Println("json Unmarshal err ", err)
		return false
	}
	return true
}

monster_test.go

package test_case

import (
    "testing"
)

func TestStore(t *testing.T) {
    monster := &Monster{
       Name:  "小狐貍",
       Age:   200,
       Skill: "魅惑",
    }
    res := monster.Store()
    if !res {
       t.Fatalf("TestStore fail,expected is %v,but got %v", true, res)
    }
    t.Logf("TestStore 測(cè)試通過(guò)")

}

func TestReStore(t *testing.T) {
    monster := &Monster{} //空的結(jié)構(gòu)體

    res := monster.ReStore() // 反序列化后結(jié)構(gòu)體就有數(shù)據(jù)
    if !res {
       t.Fatalf("TestReStore fail,expected is %v,but got %v", true, res)
    }

    if monster.Name != "小狐貍" {
       t.Fatalf("TestStore fail,expected monster.Name is %v,but got %v", "小狐貍", monster.Name)
    }
    t.Logf("TestStore 測(cè)試通過(guò)")
}