如何針對 Go 中的枚舉對命令列標誌值進行單元測試？

在 Go 中測試命令列標誌

本文探討了 Golang 中命令列標誌的測試技術。具體來說，我們將研究如何針對枚舉對標誌值進行單元測試。

問題陳述

給出以下程式碼：

<code class="go">// Define flag for output format
var formatType string

// Constants representing valid format types
const (
    text = "text"
    json = "json"
    hash = "hash"
)

// Initialize flags
func init() {
    flag.StringVar(&formatType, "format", "text", "Desired output format")
}

// Main function
func main() {
    flag.Parse()
}</code>

我們希望寫一個單元測試來驗證-format 標誌值是否與預定義常量之一匹配。

使用自訂標誌類型的解決方案

要在更多中測試標誌以靈活的方式，我們可以利用flag.Var 函數和實作Value 介面的自訂類型。

<code class="go">package main

import (
    "errors"
    "flag"
    "fmt"
)

// Custom type representing format type
type formatType string

// String() method for Value interface
func (f *formatType) String() string {
    return fmt.Sprint(*f)
}

// Set() method for Value interface
func (f *formatType) Set(value string) error {
    if len(*f) > 0 && *f != "text" {
        return errors.New("format flag already set")
    }
    if value != "text" && value != "json" && value != "hash" {
        return errors.New("Invalid Format Type")
    }
    *f = formatType(value)
    return nil
}

// Initialize flag with custom type
func init() {
    typeFlag := "text" // Default value
    usage := `Format type. Must be "text", "json" or "hash". Defaults to "text".`
    flag.Var(&typeFlag, "format", usage)
}

// Main function
func main() {
    flag.Parse()
    fmt.Println("Format type is", typeFlag)
}
</code>

在此解決方案中，flag.Var 接受一個指向滿足 Value 介面的自訂類型的指針，允許我們在 Set 方法中定義自己的驗證邏輯。

單元測試自訂標誌類型

自訂標誌類型的單元測試可以編寫如下：

<code class="go">// Test unit validates that the format flag is within the enumeration
func TestFormatFlag(t *testing.T) {
    testCases := []struct {
        input       string
        expectedErr string
    }{
        {"text", ""},
        {"json", ""},
        {"hash", ""},
        {"", "Invalid Format Type"},
        {"xml", "Invalid Format Type"},
    }

    for _, tc := range testCases {
        t.Run(tc.input, func(t *testing.T) {
            args := []string{"-format", tc.input}
            flag.CommandLine = flag.NewFlagSet("test", flag.PanicOnError)
            err := flag.CommandLine.Parse(args)

            if err != nil && err.Error() != tc.expectedErr {
                t.Errorf("Unexpected error: %v", err)
                return
            }
        })
    }
}</code>

以上是如何針對 Go 中的枚舉對命令列標誌值進行單元測試？的詳細內容。更多資訊請關注PHP中文網其他相關文章！