Go 函數單元測試的重構技巧-Golang-PHP中文網

Go 函數單元測試的重構技巧

WBOY

發布： 2024-05-04 13:04:30

原創

713 人瀏覽過

為了提高 Go 函數單元測試的可維護性和可讀性，我們可以：提取斷言函數簡化程式碼。採用表格驅動的測試組織測試資料。編寫 mocking 介面測試函數與元件的交互作用。執行細粒度的測試隔離和調試問題。應用覆蓋率工具確保測試全面性和指導改進。

Go 函数单元测试的重构技巧

Go 函數單元測試的重構技巧

當我們擁有龐大且複雜的Go 專案時，函數單元測試的維護和可讀性可能成為一大挑戰。為了應對這項挑戰，我們可以採取一些重構技巧來提高測試的可維護性和可讀性。

1. 提取斷言函數

如果測試程式碼中包含許多相同的斷言，則可以提取斷言函數來簡化程式碼。例如，我們可以定義一個AssertEqual 函數來檢查兩個值是否相等：

import "testing"

func AssertEqual(t *testing.T, expected, actual interface{}) {
    if expected != actual {
        t.Errorf("Expected %v, got %v", expected, actual)
    }
}

登入後複製

2. 使用表驅動的測試

表驅動的測試可以幫助組織和簡化測試數據。它允許我們使用一個表來提供不同的輸入和期望輸出，然後對每個輸入執行測試。例如，我們可以編寫一個表格驅動的測試來檢查Max 函數：

import (
    "testing"

    "github.com/stretchr/testify/assert"
)

func TestMax(t *testing.T) {
    tests := []struct {
        name    string
        input   []int
        expected int
    }{
        {"Empty slice", []int{}, 0},
        {"Single element", []int{1}, 1},
        {"Multiple elements", []int{1, 2, 3}, 3},
    }

    for _, tt := range tests {
        actual := Max(tt.input)
        assert.Equal(t, tt.expected, actual)
    }
}

登入後複製

3. 編寫mocking 介面

mocking 介面允許我們測試函數在與其他元件交互時的行為。我們可以使用一個 mocking 框架（如 mockery）來生成 mock 對象，該對象實現了我們關心的接口，但我們可以控制其行為。例如，我們可以寫一個mockDatabase 來測試一個使用資料庫的函數：

package main

import (
    "database/sql"
    "fmt"
    "time"

    "github.com/stretchr/testify/mock"
)

// MockDatabase is a mock database for testing purposes.
type MockDatabase struct {
    mock.Mock
}

// Query implements the Query method of a mock database.
func (m *MockDatabase) Query(query string, args ...interface{}) (*sql.Rows, error) {
    ret := m.Called(query, args)
    return ret.Get(0).(*sql.Rows), ret.Error(1)
}

// GetUserByUsernameAndPassword implements the GetUserByUsernameAndPassword method of a mock database.
func (m *MockDatabase) GetUserByUsernameAndPassword(username, password string) (*User, error) {
    ret := m.Called(username, password)
    return ret.Get(0).(*User), ret.Error(1)
}

// User represents a user in the database.
type User struct {
    Username string
    Password string
    LastLogin time.Time
}

// main is the entry point for the program.
func main() {
    mockDB := &MockDatabase{}
    mockDB.On("GetUserByUsernameAndPassword", "john", "password").Return(&User{
        Username: "john",
        Password: "password",
        LastLogin: time.Now(),
    }, nil)

    user, err := GetUser(mockDB, "john", "password")
    if err != nil {
        fmt.Println("Error getting user:", err)
    } else {
        fmt.Println("Welcome back, ", user.Username)
    }
}

登入後複製

4. 執行細粒度的測試

細粒度的測試專注於測試函數的小部分功能。透過執行細粒度的測試，我們可以更輕鬆地隔離和調試問題。例如，我們可以編寫一個測試來檢查Max 函數是否會傳回最大元素：

import "testing"

func TestMaxElement(t *testing.T) {
    tests := []struct {
        name    string
        input   []int
        expected int
    }{
        {"Empty slice", []int{}, 0},
        {"Single element", []int{1}, 1},
        {"Multiple elements", []int{1, 2, 3}, 3},
    }

    for _, tt := range tests {
        actual := MaxElement(tt.input)
        assert.Equal(t, tt.expected, actual)
    }
}

登入後複製