Bagaimana untuk menguji Pengesahan Bendera Baris Perintah Tersuai Unit dalam Go?

Mengesahkan Bendera Baris Perintah dalam Ujian Unit Go

Pertimbangkan kod berikut yang menggunakan bendera baris arahan untuk mengkonfigurasi jenis format:

<code class="go">// ... elided for brevity ...</code>

Untuk memastikan bendera -format ditetapkan kepada nilai yang dijangkakan, ujian unit boleh ditulis. Fungsi flag.Var boleh digunakan untuk menyesuaikan pemprosesan bendera, membenarkan pengesahan dan lebih banyak kawalan ke atas nilai bendera.

Pengendalian bendera tersuai dicapai dengan melaksanakan antara muka Nilai:

<code class="go">type formatType string

func (f *formatType) String() string {
    return fmt.Sprint(*f)
}

func (f *formatType) Set(value string) error {
    // Validation logic here
}</code>

Menggunakan ini pada bendera format:

<code class="go">var typeFlag formatType

func init() {
    // ... elided for brevity ...
    flag.Var(&typeFlag, "format", "Format type")
    flag.Var(&typeFlag, "f", "Format type (shorthand)")
}</code>

Untuk menguji unit pengesahan bendera tersuai, pertimbangkan pendekatan berikut yang terdapat dalam flag_test.go:

<code class="go">func TestCustomFlag(t *testing.T) {
    // Setup test environment
    origArgs := os.Args

    // Define custom flag type
    type myFlag int
    flag.Var((*myFlag)(nil), "customflag", "Custom flag")

    tests := []struct {
        origArgs  []string
        expValue  int
        expOutput string
    }{
        // ... test cases ...
    }

    for _, test := range tests {
        os.Args = test.origArgs
        // Parse flags
        flag.Parse()
        // Check flag value
        if flagValue := flag.Lookup("customflag").Value.(myFlag); flagValue != test.expValue {
            t.Errorf("Expected %v, got %v", test.expValue, flagValue)
        }
        // Restore args
        os.Args = origArgs
    }
}</code>

Ringkasnya, fungsi flag.Var membenarkan penyesuaian dan pengesahan nilai bendera, yang boleh diuji unit mengikut corak yang ditetapkan.

Atas ialah kandungan terperinci Bagaimana untuk menguji Pengesahan Bendera Baris Perintah Tersuai Unit dalam Go?. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!