From 7bc0a0f71e9322a24fc1cf5d6688a1426955ac7a Mon Sep 17 00:00:00 2001 From: Guangwen Feng Date: Wed, 19 Jul 2023 16:56:49 +0800 Subject: [PATCH] test: add unit test for Header.has (#2555) Signed-off-by: Guangwen Feng --- pkg/source/header_test.go | 36 ++++++++++++++++++++++++++++++++++++ 1 file changed, 36 insertions(+) diff --git a/pkg/source/header_test.go b/pkg/source/header_test.go index 7a6b0aa68..6e65e3cd0 100644 --- a/pkg/source/header_test.go +++ b/pkg/source/header_test.go @@ -19,8 +19,44 @@ package source import ( "fmt" "testing" + + "github.com/stretchr/testify/assert" ) +func TestHeader_has(t *testing.T) { + assert := assert.New(t) + tests := []struct { + name string + h Header + key string + want bool + }{ + { + name: "", + h: Header{ + "aaa": []string{"ddd"}, "fff": []string{"cccc"}, + }, + key: "aaa", + want: true, + }, + { + name: "", + h: Header{ + "aaa": []string{"ddd"}, "fff": []string{"cccc"}, + }, + key: "bbb", + want: false, + }, + } + + for _, tt := range tests { + t.Run(tt.name, func(t *testing.T) { + xx := tt.h.has(tt.key) + assert.Equal(tt.want, xx) + }) + } +} + func TestHeader_Clone(t *testing.T) { tests := []struct { name string