コード例 #1
0
ファイル: wrapper_test.go プロジェクト: gourd/kit
func TestWithID(t *testing.T) {
	id := "foobar"
	ctx0 := context.Background()
	ctx1 := gourdctx.WithID(ctx0, id)

	if want, have := "", gourdctx.GetID(ctx0); want != have {
		t.Errorf("expected %#v, got %#v", want, have)
	}
	if want, have := id, gourdctx.GetID(ctx1); want != have {
		t.Errorf("expected %#v, got %#v", want, have)
	}
}
コード例 #2
0
ファイル: middleware_test.go プロジェクト: gourd/kit
func TestUseID_Reuse(t *testing.T) {
	r, _ := http.NewRequest("GET", "/foo/bar", nil)
	id := "hello"
	r.Header.Set(gourdctx.IDHeaderKey, id)
	ctx := gourdctx.UseID(context.Background(), r)

	if want, have := id, gourdctx.GetRequestID(r); want != have {
		t.Errorf("expected %#v, got %#v", want, have)
	}
	if want, have := id, gourdctx.GetID(ctx); want != have {
		t.Errorf("expected %#v, got %#v", want, have)
	}

}
コード例 #3
0
ファイル: middleware_test.go プロジェクト: gourd/kit
func TestUseID(t *testing.T) {
	r, _ := http.NewRequest("GET", "/foo/bar", nil)
	ctx := gourdctx.UseID(context.Background(), r)
	var id string

	if id = gourdctx.GetRequestID(r); id == "" {
		t.Errorf("unexpected empty string")
		return
	}

	if want, have := id, gourdctx.GetID(ctx); want != have {
		t.Errorf("expected %#v, got %#v", want, have)
	}

}