httptestを試してみた

goの勉強続き。
testをやってみる

なんとか動いてCoverage 75%になったけど、100%にならないのはなんでなんだろ。
httptestでやるのはなかなか手応えがあったので、
単純な数値計算関数などに戻って、試してみようとおもう。

http2.go

package main

import (
    "io"
    "net/http"
    "strconv"
    "strings"
)

// ResponseWriter is interface as
//  Header()  Header
//  Write([]byte)(int,error)
//  iWriteHeder(statuscoe int)

// WriteString is function as
// func WriteString(w Writer, s string) (n int, err error)

// Writer is interface as
// Write(p []byte) (n int, err error)

func ServeHoge(w http.ResponseWriter, req *http.Request) {
    agent := req.Header.Get("user-agent")
    check := strings.Contains(agent, "537")
    check_string := strconv.FormatBool(check)

    if check {
        io.WriteString(w, agent+"\n537 is error\n"+check_string+"\n")
    }
    io.WriteString(w, "Welcome! this is hoge server\n" + agent+"\n")
}

// testの方法はたぶん、
// requestを作ってuse-agentをセットしてなげる
// それを受けて、結果が書き込まれるのもの（Recorder?）を
// うまく記録して、
// その結果のBodyをチェックするのかな。

//参考：https://qiita.com/umisama/items/3d44560e1c06fa531069

func main() {
    http.HandleFunc("/hoge", ServeHoge)

    http.ListenAndServe(":12345", nil)
}

http2_test.go

package main

import (
    "net/http"
    "net/http/httptest"
    "testing"
    "strings"
    "io/ioutil"
)

func TestCheckAgent(t *testing.T) {
    ts := httptest.NewServer(http.HandlerFunc(ServeHoge))
    defer ts.Close()

    _, err := http.Get(ts.URL)

    var request1 *http.Request

    request1, err = http.NewRequest("GET", ts.URL, nil)

    request1.Header.Add("user-agent","537")

    c := http.DefaultClient
    reply, err := c.Do(request1)

    if err != nil {
        t.Error("unexpected")
        return
    }

    data, err := ioutil.ReadAll(reply.Body)
    if err != nil {
        t.Fatalf("Error:%v",err)
    }
    if reply.StatusCode == 200 && ! strings.Contains(string(data), "537") {
        t.Fatalf("agent is not included 537 in Body:%v",string(data))
    }

    request1.Header.Add("user-agent","532")
    c = http.DefaultClient
    reply, err = c.Do(request1)
    data, err = ioutil.ReadAll(reply.Body)
    if reply.StatusCode == 200 && ! strings.Contains(string(data), "Welcome") {
        t.Fatalf("agent is not included Welcome in Body:%v",string(data))
    }



}

テスト結果

(base) http2 $ go test -v -cover
=== RUN   TestCheckAgent
--- PASS: TestCheckAgent (0.00s)
PASS
coverage: 75.0% of statements
ok      http2   0.330s
(base) http2 $