[Golang]HTTPTESTレスポンスのCookieの値を取得する

7068 ワード
cookie HTTP Go httptest Go テキストリンク

概要

Goのハンドラなどに対するHTTPのテストを行う際に、レスポンスのCookieの値を取り出す方法メモです。

事前知識

参考にした主な記事など

内容

Cookieを返すだけのHogeHandler関数に対して、Cookieの値を取得して表示するテストを書きます。

main.go
package main

import (
    "net/http"
)

func HogeHandler(w http.ResponseWriter, r *http.Request) error {
    http.SetCookie(w, &http.Cookie{
        Name:  "hoge",
        Value: "hogevalue",
    })
    return nil
}
main_test.go
package main

import (
    "net/http"
    "net/http/httptest"
    "testing"
)

func TestHogeHandler(t *testing.T) {
    req := httptest.NewRequest("GET", "/", nil)
    resp := httptest.NewRecorder()
    HogeHandler(resp, req)

    // ここでCookieをパースする
    parser := &http.Request{Header: http.Header{"Cookie": resp.Header()["Set-Cookie"]}}
    hogeCookie, _ := parser.Cookie("hoge")

    t.Log(hogeCookie.Value)
}

Cookie取得部分では、http.Requestを経由してCookie関数を利用することで、Cookieをパースして取得することができます。
本音を言うとhttptest.ResponseRecorderにCookie関数があってほしい...笑
ちなみにパースする必要がないならresp.Header()["Set-Cookie"]で取得できます。
※ HeaderMap関数は非推奨になったらしいです。

実行&出力

テスト実行

$ go test -v path/to/dir

出力結果

=== RUN   TestHogeHandler
--- PASS: TestHogeHandler (0.00s)
    main_test.go:17: hogevalue
PASS

参考

Jonny Reeves – Testing Setting HTTP Cookies in Go
http - The Go Programming Language
httptest - The Go Programming Language
testing - The Go Programming Language

最後に

記事に間違いや不明な点があれば遠慮なくご指摘ください。

Oracleの2つのテーブル関連の更新
HDu 2544最短