Bagaimana untuk Menguji Panggilan HTTP dalam Go menggunakan Pakej `httptest`?

Cara Menguji Panggilan HTTP dalam Go dengan httptest

Pengenalan

Ujian unit ialah aspek penting pembangunan perisian, dan untuk program Go, pakej httptest menyediakan alat berguna untuk menguji panggilan HTTP. Artikel ini akan menunjukkan cara menggunakan httptest untuk menulis ujian unit untuk kod Go anda yang membuat permintaan HTTP.

Cabaran

Pertimbangkan kod Go berikut:

package main

import (
    "encoding/json"
    "fmt"
    "io/ioutil"
    "log"
    "net/http"
    "time"
)

type twitterResult struct {
    Results []struct {
        Text     string `json:"text"`
        Ids      string `json:"id_str"`
        Name     string `json:"from_user_name"`
        Username string `json:"from_user"`
        UserId   string `json:"from_user_id_str"`
    }
}

var (
  twitterUrl = "http://search.twitter.com/search.json?q=%23UCL"
  pauseDuration = 5 * time.Second
)

func retrieveTweets(c chan<- *twitterResult) {
    for {
        resp, err := http.Get(twitterUrl)
        if err != nil {
            log.Fatal(err)
        }

        defer resp.Body.Close()
        body, err := ioutil.ReadAll(resp.Body)
        r := new(twitterResult) //or &amp;twitterResult{} which returns *twitterResult
        err = json.Unmarshal(body, &amp;r)
        if err != nil {
            log.Fatal(err)
        }
        c <- r
        time.Sleep(pauseDuration)
    }

}

func displayTweets(c chan *twitterResult) {
    tweets := <-c
    for _, v := range tweets.Results {
        fmt.Printf("%v:%v\n", v.Username, v.Text)
    }

}

func main() {
    c := make(chan *twitterResult)
    go retrieveTweets(c)
    for {
        displayTweets(c)
    }

}

Objektifnya adalah untuk menulis ujian unit untuk kod ini, memfokuskan pada menguji permintaan HTTP yang dibuat untuk mendapatkan tweet daripada carian Twitter API.

Penyelesaian Menggunakan httptest

Pakej httptest menawarkan dua jenis ujian: ujian tindak balas dan ujian pelayan. Untuk senario ini, ujian pelayan lebih sesuai. Begini cara untuk meneruskan:

1. Buat Pelayan Ujian:

func TestRetrieveTweets(t *testing.T){
    ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        w.Header().Set("Content-Type", "application/json")
        fmt.Fprintln(w, `{"fake twitter json string"}`)
    }))
    defer ts.Close()

Kod ini mencipta pelayan ujian yang bertindak balas kepada permintaan HTTP dengan respons JSON yang dipratentukan.

2. Kemas kini yang Asal Kod:

twitterUrl = ts.URL

Dalam kod asal, pembolehubah twitterUrl diubah suai untuk menghala ke URL pelayan ujian dan bukannya titik akhir API Twitter yang sebenar.

3. Buat Saluran:

c := make(chan *twitterResult)

Saluran digunakan untuk berkomunikasi antara goroutine ujian dan goroutine utama dalam kod asal.

4. Mulakan Goroutine untuk Mendapatkan Tweet:

go retrieveTweets(c)

Ini goroutine memulakan proses mendapatkan semula tweet daripada ujian pelayan.

5. Terima Tweet:

tweet := <-c
if tweet != expected1 {
    t.Fail()
}

tweet = <-c
if tweet != expected2 {
    t.Fail()
}

Groutine ujian menerima tweet daripada saluran dan menyemak sama ada ia sepadan dengan hasil yang dijangkakan.

Lagi Pertimbangan

Perlu diingat bahawa ujian itu tidak mengesahkan kandungan respons HTTP. Untuk ujian yang lebih teliti, adalah perlu untuk membandingkan respons sebenar dengan respons yang dijangkakan. Selain itu, pelayan ujian harus mengembalikan struktur tindak balas yang lebih realistik untuk mensimulasikan API Twitter sebenar dengan tepat.

Atas ialah kandungan terperinci Bagaimana untuk Menguji Panggilan HTTP dalam Go menggunakan Pakej `httptest`?. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!