如何在單元測試中正確測試Go的「flag」包參數解析？

如何在 Go 中測試參數解析

考慮以下 Go 程式碼，它使用 flag 套件解析命令列參數：

package main

import (
    "flag"
    "fmt"
)

func main() {
    passArguments()
}

func passArguments() string {
    username := flag.String("user", "root", "Username for this server")
    flag.Parse()
    fmt.Printf("Your username is %q.", *username)

    usernameToString := *username
    return usernameToString
}

為了測試這個函數，我們可以將一個參數傳遞給編譯後的程式碼，如圖所示下面：

./args -user=bla

這應該會產生輸出：

Your username is "bla"

但是，如果我們嘗試使用以下測試來測試此行為：

package main

import (
    "os"
    "testing"
)

func TestArgs(t *testing.T) {
    expected := "bla"
    os.Args = []string{"-user=bla"}

    actual := passArguments()

    if actual != expected {
        t.Errorf("Test failed, expected: '%s', got:  '%s'", expected, actual)
    }
}

我們收到一個錯誤：

FAIL
coverage: 87.5% of statements
FAIL    tool    0.008s

此失敗的原因是簡單地將os.Args 設定為新的slice 不將參數傳遞給函數。為了解決這個問題，我們需要確保os.Args 中的第一個值是可執行檔本身（的路徑），如本修訂後的測試所示：

func TestArgs(t *testing.T) {
    expected := "bla"
    oldArgs := os.Args
    defer func() { os.Args = oldArgs }()
    os.Args = []string{"cmd", "-user=bla"}

    actual := passArguments()

    if actual != expected {
        t.Errorf("Test failed, expected: '%s', got:  '%s'", expected, actual)
    }
}

透過延遲os.Args的恢復恢復到其原始狀態，我們可以確保測試不會幹擾其他測試，這些測試可能正在檢查呼叫go test 時傳遞的真實參數。

以上是如何在單元測試中正確測試Go的「flag」包參數解析？的詳細內容。更多資訊請關注PHP中文網其他相關文章！