Bagaimana untuk menguji Pengesahan Bendera Baris Perintah Unit dalam Go?

Cara Ujian Unit Bendera Baris Perintah dalam Go

Pengenalan:

Ujian unit ialah penting untuk memastikan kebolehpercayaan dan ketepatan kod anda, termasuk pengendalian bendera baris arahan. Penghuraian bendera ialah tugas biasa dalam aplikasi Go dan adalah penting untuk mengesahkan bahawa bendera disahkan dengan betul. Artikel ini akan meneroka cara untuk menguji bendera baris perintah dalam Go.

Menyediakan Ujian Unit:

Untuk pengesahan bendera ujian unit, kami akan membuat tersuai jenis bendera yang menyemak sama ada nilai bendera berada dalam penghitungan yang dijangkakan. Pertimbangkan contoh kod berikut:

<code class="go">var formatType string

const (
    text = "text"
    json = "json"
    hash = "hash"
)

func init() {
    const (
        defaultFormat = "text"
        formatUsage   = "desired output format"
    )

    flag.StringVar(&amp;formatType, "format", defaultFormat, formatUsage)
    flag.StringVar(&amp;formatType, "f", defaultFormat, formatUsage+" (shorthand)")
}</code>

Pengesahan Bendera Tersuai:

Kami boleh menentukan bendera tersuai. Jenis nilai untuk mengesahkan bendera format:

<code class="go">type formatType string

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

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

Mencipta Ujian Unit:

Sekarang, mari tulis ujian unit yang mengesahkan pengesahan bendera tersuai:

<code class="go">func TestValidateFormatFlag(t *testing.T) {
    testCases := []struct {
        input    string
        expected error
    }{
        {"text", nil},
        {"json", nil},
        {"hash", nil},
        {"invalid", errors.New("Invalid Format Type")},
    }

    for _, tc := range testCases {
        args := []string{"-format", tc.input}
        os.Args = args

        typeFlag = "text"
        flag.CommandLine = flag.NewFlagSet("test", flag.ContinueOnError)
        flag.Var(&amp;typeFlag, "format", "Format type. Must be \"text\", \"json\" or \"hash\". Defaults to \"text\".")
        if err := flag.CommandLine.Parse(args[1:]); err != nil {
            t.Errorf("Error parsing flags: %v", err)
        }

        if tc.expected != nil && err == nil {
            t.Errorf("Expected error: %v, got nil", tc.expected)
        } else if tc.expected == nil && err != nil {
            t.Errorf("Unexpected error: %v", err)
        }
    }
}</code>

Dalam ujian ini, kami sediakan satu siri kes ujian dengan input yang berbeza dan ralat yang dijangkakan. Kami mengejek argumen baris arahan menggunakan os.Args dan menghuraikan bendera menggunakan FlagSet tersuai. Dengan membandingkan ralat sebenar dengan ralat yang dijangkakan untuk setiap kes ujian, kami memastikan bahawa pengesahan bendera berfungsi seperti yang dimaksudkan.

Petua untuk Menjalankan Ujian Bendera:

• Buat ujian unit berasingan untuk setiap keperluan pengesahan bendera.
• Gunakan rangka kerja olok-olok seperti memberi keterangan untuk memudahkan persediaan dan penegasan ujian.
• Elakkan menggunakan penghuraian baris arahan sebenar dalam ujian unit untuk memastikan pengasingan.

Kesimpulan:

Dengan melaksanakan jenis bendera tersuai dan ujian unit penulisan, anda boleh mengesahkan bendera baris perintah dengan berkesan dalam Go. Amalan ini membantu mengekalkan kebolehpercayaan dan keteguhan aplikasi anda.

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