在 Golang 單元測(cè)試中使用依賴注入（DI）可以隔離要測(cè)試的程式碼，簡(jiǎn)化測(cè)試設(shè)定和維護(hù)。流行的 DI 庫(kù)包括 wire 和 go-inject，它們可以產(chǎn)生依賴項(xiàng)樁或模擬，供測(cè)試使用。 DI 測(cè)試的步驟包括設(shè)定依賴項(xiàng)、設(shè)定測(cè)試案例和斷言結(jié)果。使用 DI 測(cè)試 HTTP 請(qǐng)求處理函數(shù)的範(fàn)例表明，它可以輕鬆隔離和測(cè)試程式碼，無需實(shí)際依賴項(xiàng)或通訊。

如何在Golang 中使用依賴注入進(jìn)行單元測(cè)試

##依賴注入（DI）是一種設(shè)計(jì)模式，它允許你向物件提供其依賴項(xiàng)，而無需明確地創(chuàng)建它們。在單元測(cè)試中，DI 可以幫助你隔離要測(cè)試的程式碼，並使測(cè)試更容易設(shè)定和維護(hù)。

Golang 中的DI

Golang 中有許多流行的DI 庫(kù)，其中最著名的是[wire](https://github.com/google/ wire) 和[go-inject](https://github.com/go-inject/go-inject)。這些庫(kù)的工作原理是產(chǎn)生可以用作測(cè)試中依賴項(xiàng)的樁或模擬。

設(shè)定DI 測(cè)試

以下是如何使用wire 設(shè)定DI 單元測(cè)試：

import (
    "context"
    "testing"

    "github.com/google/go-cmp/cmp"
)

// Interface we want to test.
type Greeter interface {
    Greet(ctx context.Context, name string) (string, error)
}

// Implementation we want to test.
type DefaultGreeter struct{}

func (g DefaultGreeter) Greet(ctx context.Context, name string) (string, error) {
    return "Hello, " + name, nil
}

func TestGreeter_Greet(t *testing.T) {
    type Fields struct {
        greeter Greeter
    }

    wire.Build(Fields{
        greeter: (*DefaultGreeter)(nil),
    })

    cases := []struct {
        setup    func(t *testing.T, fields Fields)
        expected *string
        wantErr  bool
    }{
        {
            expected: String("Hello, Bob"),
        },
    }

    for _, tc := range cases {
        tc := tc // capture range variable
        t.Run(testName, func(t *testing.T) {
            t.Parallel()

            fields := Fields{}
            tc.setup(t, fields)

            result, err := fields.greeter.Greet(context.Background(), "Bob")

            if (err != nil) != tc.wantErr {
                t.Fatalf("error = %v, wantErr = %v", err, tc.wantErr)
            }
            if tc.wantErr {
                return
            }
            if diff := cmp.Diff(*tc.expected, result); diff != "" {
                t.Fatalf("result mismatch (-want +got):\n%s", diff)
            }
        })
    }
}

使用DI 進(jìn)行測(cè)試

#在上面的測(cè)試中，我們使用

wire.Build 來產(chǎn)生一個(gè)Fields 結(jié)構(gòu)的實(shí)例，該實(shí)例包含要用於測(cè)試的依賴項(xiàng)樁。然後，我們可以像往常一樣設(shè)定測(cè)試案例並斷言結(jié)果。

實(shí)戰(zhàn)案例

以下是如何使用DI 單元測(cè)試一個(gè)處理HTTP 請(qǐng)求的函數(shù)：

import (
    "net/http"
    "net/http/httptest"
    "testing"

    "github.com/gorilla/mux"
    "github.com/stretchr/testify/assert"

    "mypkg/handlers"
)

// Interface we want to test.
type UserService interface {
    GetUser(id int) (*User, error)
}

// Implementation we want to test.
type DefaultUserService struct{}

func (s DefaultUserService) GetUser(id int) (*User, error) {
    return &User{ID: id, Name: "Bob"}, nil
}

type Request struct {
    UserService UserService
}

func (r Request) ServeHTTP(w http.ResponseWriter, req *http.Request) {
    id, err := strconv.Atoi(mux.Vars(req)["id"])
    if err != nil {
        http.Error(w, err.Error(), http.StatusBadRequest)
        return
    }
    user, err := r.UserService.GetUser(id)
    if err != nil {
        http.Error(w, err.Error(), http.StatusInternalServerError)
        return
    }
    fmt.Fprintf(w, "%s", user.Name)
}

func TestHandler_GetUser(t *testing.T) {
    r := &Request{}

    type Fields struct {
        userService UserService
    }

    wire.Build(Fields{
        userService: (*DefaultUserService)(nil),
    })

    cases := []struct {
        name string
        id   int
        body string
        want string
    }{
        {
            body: `{"body":""}`,
            want: `Bob`,
        },
        {
            id:   2,
            body: `{"body":""}`,
            want: `Bob`,
        },
    }

    for _, tc := range cases {
        tc := tc // capture range variable
        t.Run(tc.name, func(t *testing.T) {
            req, _ := http.NewRequest("GET", "/", bytes.NewBuffer([]byte(tc.body)))
            if tc.id != 0 {
                req = mux.SetURLVars(req, map[string]string{"id": strconv.Itoa(tc.id)})
            }
            rr := httptest.NewRecorder()
            handler := http.HandlerFunc(r.ServeHTTP)
            handler.ServeHTTP(rr, req)

            assert.Equal(t, tc.want, rr.Body.String())
        })
    }
}

透過使用DI 和樁，我們能夠輕鬆地隔離和測(cè)試

GetUser 函數(shù)，而無需涉及實(shí)際的資料庫(kù)或HTTP 請(qǐng)求。

以上是如何在 Golang 中使用依賴注入進(jìn)行單元測(cè)試？的詳細(xì)內(nèi)容。更多資訊請(qǐng)關(guān)注PHP中文網(wǎng)其他相關(guān)文章！