Apakah kegunaan golang httptest

Dalam beberapa tahun kebelakangan ini, dengan populariti dan kecekapan bahasa Golang, ia telah menarik lebih banyak perhatian dan kasih sayang daripada pembangun. Golang menyediakan rangka kerja ujian yang ringkas dan mudah digunakan, termasuk perpustakaan httptest. Dalam artikel ini, saya akan memperkenalkan konsep asas, penggunaan dan langkah berjaga-jaga httptest, dengan harapan dapat membantu pembangun Golang dalam kerja ujian mereka yang seterusnya.

1. Pengenalan kepada httptest

Dalam rangka kerja ujian Golang, fungsi utama httptest adalah untuk membantu penguji mensimulasikan permintaan dan respons HTTP, supaya dapat menguji bahagian pelayan dengan mudah Ujian antara muka API. Ia tidak bergantung pada rangkaian dan pelayan HTTP sebenar, tetapi menggunakan fungsi dalam pakej net/http yang disediakan oleh Golang untuk mensimulasikan keseluruhan proses permintaan dan tindak balas.

2. Kaedah biasa httptest

(1) Cipta pelayan ujian

Dalam httptest, kita perlu mencipta pelayan ujian melalui kaedah NewServer. Parameter kaedah ini perlu diluluskan dalam HandlerFunc yang melaksanakan antara muka http.Handler. Sebagai contoh, kita boleh mencipta HandlerFunc yang mengendalikan permintaan GET dan mencipta pelayan ujian melaluinya:

func TestGetFoo(t *testing.T) {
    srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        w.WriteHeader(http.StatusOK)
        fmt.Fprintln(w, "Hello, World!")
    }))
    defer srv.Close()

    resp, err := http.Get(srv.URL)
    if err != nil {
        t.Fatal(err)
    }
    defer resp.Body.Close()

    if resp.StatusCode != http.StatusOK {
        t.Errorf("expected %d but got %d", http.StatusOK, resp.StatusCode)
    }
}

(2) Hantar permintaan

Dengan pelayan ujian, kami boleh lulus Klien untuk menghantar permintaan kepada pelayan. Pelanggan ialah klien permintaan HTTP yang disediakan oleh Golang. Sebagai contoh, kita boleh mencipta HandlerFunc yang mengendalikan permintaan POST dan mencipta pelayan ujian daripadanya. Kemudian hantar permintaan POST, termasuk data Json:

func TestPostJson(t *testing.T) {
    srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        body, _ := ioutil.ReadAll(r.Body)
        w.WriteHeader(http.StatusOK)
        fmt.Fprintln(w, string(body))
    }))
    defer srv.Close()

    payload := struct {
        Name string `json:"name"`
        Age  int    `json:"age"`
    }{
        Name: "Robert",
        Age:  30,
    }
    buf, _ := json.Marshal(payload)
    resp, err := http.Post(srv.URL, "application/json", bytes.NewReader(buf))
    if err != nil {
        t.Fatal(err)
    }
    defer resp.Body.Close()

    if resp.StatusCode != http.StatusOK {
        t.Errorf("expected %d but got %d", http.StatusOK, resp.StatusCode)
    }
    body, _ := ioutil.ReadAll(resp.Body)
    if string(body) != `{"name":"Robert","age":30}` {
        t.Errorf("expected %s but got %s", `{"name":"Robert","age":30}`, string(body))
    }
}

(3) Tetapkan pengepala permintaan dan pengepala respons

Dalam ujian, pengepala permintaan dan pengepala respons juga merupakan bahagian yang sangat penting . Kita boleh menetapkan pengepala permintaan dan pengepala respons melalui kaedah Pengepala. Sebagai contoh, kita boleh mencipta HandlerFunc yang mengendalikan permintaan GET dan mencipta pelayan ujian daripadanya. Kemudian tetapkan pengepala permintaan dan pengepala respons:

func TestHeader(t *testing.T) {
    srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        w.Header().Set("X-test", "hello")
        w.WriteHeader(http.StatusOK)
        fmt.Fprintln(w, "Hello, World!")
    }))
    defer srv.Close()

    req, err := http.NewRequest("GET", srv.URL, nil)
    if err != nil {
        t.Fatal(err)
    }
    req.Header.Add("X-test", "world")
    resp, err := http.DefaultClient.Do(req)
    if err != nil {
        t.Fatal(err)
    }
    defer resp.Body.Close()

    if resp.StatusCode != http.StatusOK {
        t.Errorf("expected %d but got %d", http.StatusOK, resp.StatusCode)
    }
    if resp.Header.Get("X-test") != "hello" {
        t.Errorf("expected %s but got %s", "hello", resp.Header.Get("X-test"))
    }
}

(4) Memproses permintaan HTTPS

Dalam projek sebenar, banyak antara muka HTTP memerlukan penghantaran selamat melalui HTTPS. Dalam httptest, kami juga boleh mencipta pelayan ujian yang menyokong HTTPS melalui kaedah NewTLSServer. Pelayan yang baru dibuat menggunakan sijil yang ditandatangani sendiri dan oleh itu tidak boleh digunakan dalam persekitaran pengeluaran. Sebagai contoh, kita boleh mencipta HandlerFunc yang mengendalikan permintaan POST dan menggunakannya untuk mencipta pelayan ujian yang menyokong HTTPS. Kemudian hantar permintaan POST, termasuk data Json:

func TestTlsPostJson(t *testing.T) {
    srv := httptest.NewTLSServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        body, _ := ioutil.ReadAll(r.Body)
        w.WriteHeader(http.StatusOK)
        fmt.Fprintln(w, string(body))
    }))
    defer srv.Close()

    payload := struct {
        Name string `json:"name"`
        Age  int    `json:"age"`
    }{
        Name: "Robert",
        Age:  30,
    }
    buf, _ := json.Marshal(payload)
    client := srv.Client()

    resp, err := client.Post(srv.URL, "application/json", bytes.NewReader(buf))
    if err != nil {
        t.Fatal(err)
    }
    defer resp.Body.Close()

    if resp.StatusCode != http.StatusOK {
        t.Errorf("expected %d but got %d", http.StatusOK, resp.StatusCode)
    }
    body, _ := ioutil.ReadAll(resp.Body)
    if string(body) != `{"name":"Robert","age":30}` {
        t.Errorf("expected %s but got %s", `{"name":"Robert","age":30}`, string(body))
    }
}

3. Nota tentang httptest

Walaupun httptest sangat mudah dan praktikal dalam rangka kerja ujian Golang, ia masih boleh digunakan secara tidak wajar Ini boleh mengakibatkan keputusan ujian tidak tepat. Kita perlu memberi perhatian kepada perkara berikut:

(1) httptest tidak boleh digunakan sebagai pelayan HTTP atau HTTPS dalam persekitaran pengeluaran, hanya untuk tujuan ujian.

(2) Jika pelayan ujian tidak ditutup dengan betul, sumber mungkin kehabisan kerana terlalu banyak suite ujian.

(3) Jika data yang dihantar menggunakan httptest tidak sepadan dengan data yang anda serahkan, anda tidak akan mendapat keputusan ujian yang betul.

4. Ringkasan

Dalam proses ujian Golang, httptest ialah alat yang sangat berguna yang boleh membantu kami mensimulasikan permintaan dan respons HTTP, supaya dapat menguji pelayan dengan mudah- antara muka API sisi untuk ujian. Dalam artikel ini, kami memperkenalkan konsep asas, penggunaan dan langkah berjaga-jaga httptest secara terperinci, dengan harapan dapat membantu pembangun Golang dalam kerja ujian mereka yang seterusnya.

Atas ialah kandungan terperinci Apakah kegunaan golang httptest. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!