Bagaimana untuk Menguji Kod Go yang Dibaca dari Stdin dan Menulis ke Stdout?

Menulis Ujian Go untuk Kod yang Menulis kepada Stdin

Untuk menulis ujian Go untuk kod yang dibaca dari stdin dan menggemakannya kembali ke stdout , adalah dinasihatkan untuk mengasingkan fungsi ke dalam fungsi berasingan yang mengambil io.Reader dan io.Writer sebagai parameter.

Daripada melaksanakan semua operasi dalam fungsi utama menggunakan stdin dan stdout secara langsung, kod anda harus menentukan fungsi untuk tujuan khusus ini. Contohnya:

<code class="go">func echo(r io.Reader, w io.Writer) {
    reader := bufio.NewReader(r)
    for {
        fmt.Print("> ", w)
        bytes, _, err := reader.ReadLine()
        if err == io.EOF {
            os.Exit(0)
        }
        fmt.Println(string(bytes), w)
    }
}</code>

Dalam fungsi utama anda, anda boleh memanggil fungsi gema:

<code class="go">func main() {
    echo(os.Stdin, os.Stdout)
}</code>

Untuk menguji fungsi ini, anda boleh membuat ujian seperti ini:

<code class="go">import (
    "bytes"
    "io"
    "testing"
)

func TestEcho(t *testing.T) {
    input := "abc\n"
    expected := "abc\n"

    r := bytes.NewBufferString(input)
    w := bytes.NewBufferString("")

    echo(r, w)

    if got := w.String(); got != expected {
        t.Errorf("Expected: %v, Got: %v", expected, got)
    }
}</code>

Ujian ini mencipta pembaca penimbal daripada rentetan input yang ditentukan dan penulis penimbal untuk menangkap output. Ia kemudian memanggil fungsi gema, menghantar pembaca dan penulis. Akhirnya, ia membandingkan output dengan hasil yang dijangkakan.

Atas ialah kandungan terperinci Bagaimana untuk Menguji Kod Go yang Dibaca dari Stdin dan Menulis ke Stdout?. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!