Bagaimana untuk Menguji Nilai Bendera Baris Perintah Terhadap Penghitungan dalam Go?

Menguji Bendera Baris Perintah dalam Go

Artikel ini meneroka teknik ujian untuk bendera baris arahan di Golang. Khususnya, kami akan mengkaji cara untuk menguji unit nilai bendera terhadap penghitungan.

Pernyataan Masalah

Memandangkan kod berikut:

<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>

Kami ingin menulis ujian unit untuk mengesahkan bahawa nilai bendera -format sepadan dengan salah satu pemalar yang dipratentukan.

Penyelesaian Menggunakan Jenis Bendera Tersuai

Untuk menguji bendera dalam lebih banyak dengan cara yang fleksibel, kita boleh menggunakan fungsi flag.Var dengan jenis tersuai yang melaksanakan antara muka Nilai.

<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>

Dalam penyelesaian ini, flag.Var membawa penuding kepada jenis tersuai yang memenuhi antara muka Value, membenarkan kami untuk menentukan logik pengesahan kami sendiri dalam kaedah Set.

Jenis Bendera Tersuai Pengujian Unit

Ujian unit untuk jenis bendera tersuai boleh ditulis seperti berikut:

<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>

Atas ialah kandungan terperinci Bagaimana untuk Menguji Nilai Bendera Baris Perintah Terhadap Penghitungan dalam Go?. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!