From 0933e2f185f12d143cbe0fc7f63df1eb0cafc6b8 Mon Sep 17 00:00:00 2001 From: wanjunlei Date: Wed, 17 Jun 2020 09:48:56 +0800 Subject: [PATCH] add test --- pkg/apiserver/auditing/types_test.go | 26 ++++++++++++++++++++++++++ 1 file changed, 26 insertions(+) diff --git a/pkg/apiserver/auditing/types_test.go b/pkg/apiserver/auditing/types_test.go index bde7146ec..85ccf8f37 100644 --- a/pkg/apiserver/auditing/types_test.go +++ b/pkg/apiserver/auditing/types_test.go @@ -296,3 +296,29 @@ func TestAuditing_LogResponseObject(t *testing.T) { assert.EqualValues(t, string(expectedBs), string(bs)) } + +func TestResponseCapture_WriteHeader(t *testing.T) { + record := httptest.NewRecorder() + resp := NewResponseCapture(record) + + resp.WriteHeader(404) + + assert.EqualValues(t, 404, resp.StatusCode()) + assert.EqualValues(t, 404, record.Code) +} + +func TestResponseCapture_Write(t *testing.T) { + + record := httptest.NewRecorder() + resp := NewResponseCapture(record) + + body := []byte("123") + + _, err := resp.Write(body) + if err != nil { + panic(err) + } + + assert.EqualValues(t, body, resp.Bytes()) + assert.EqualValues(t, body, record.Body.Bytes()) +}