From 94e4b7a5a243d319f9ab4586a27d0952d7ce9900 Mon Sep 17 00:00:00 2001 From: Guangwen Feng Date: Tue, 18 Jul 2023 17:52:40 +0800 Subject: [PATCH] test: add unit test for Request.Context() (#2554) Signed-off-by: Guangwen Feng --- pkg/source/request_test.go | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) diff --git a/pkg/source/request_test.go b/pkg/source/request_test.go index 3c8e11857..5eabd3097 100644 --- a/pkg/source/request_test.go +++ b/pkg/source/request_test.go @@ -64,6 +64,25 @@ func TestNewRequest(t *testing.T) { assert.Equal(expectedDeadline, gotDeadline) } +func TestRequest_Context(t *testing.T) { + assert := assert.New(t) + + got, err := NewRequest("http://www.dragonfly.io") + assert.Nil(err) + assert.Equal(got.Context(), context.Background()) + + testHeaderMap := map[string]string{"testKey1": "testValue1", "testKey2": "testValue2"} + got, err = NewRequestWithHeader("http://www.dragonfly.io", testHeaderMap) + assert.Nil(err) + assert.Equal(got.Context(), context.Background()) + + testContext, cancelFunc := context.WithTimeout(context.Background(), 3*time.Second) + defer cancelFunc() + got, err = NewRequestWithContext(testContext, "http://www.dragonfly.io", testHeaderMap) + assert.Nil(err) + assert.Equal(got.Context(), testContext) +} + func TestRequest_Clone(t *testing.T) { var testURL, err = url.Parse("http://www.dragonfly.io") testCloneURL, err := url.Parse("http://www.dragonfly.io")