From 2bc537d306e3096ccf25b9661281845244cd7572 Mon Sep 17 00:00:00 2001
From: Guangwen Feng <fenggw-fnst@fujitsu.com>
Date: Tue, 25 Jul 2023 16:48:14 +0800
Subject: [PATCH] test: add unit test for Header.get (#2568)

Signed-off-by: Guangwen Feng <fenggw-fnst@fujitsu.com>
---
 pkg/source/header_test.go | 42 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 42 insertions(+)

diff --git a/pkg/source/header_test.go b/pkg/source/header_test.go
index 6e65e3cd0..2cdab9d65 100644
--- a/pkg/source/header_test.go
+++ b/pkg/source/header_test.go
@@ -23,6 +23,48 @@ import (
 	"github.com/stretchr/testify/assert"
 )
 
+func TestHeader_get(t *testing.T) {
+	assert := assert.New(t)
+	tests := []struct {
+		name string
+		h    Header
+		key  string
+		want string
+	}{
+		{
+			name: "",
+			h: Header{
+				"aaa": []string{"ddd"}, "fff": []string{"cccc"},
+			},
+			key:  "aaa",
+			want: "ddd",
+		},
+		{
+			name: "",
+			h: Header{
+				"aaa": []string{"ddd"}, "fff": []string{""},
+			},
+			key:  "fff",
+			want: "",
+		},
+		{
+			name: "",
+			h: Header{
+				"aaa": []string{"ddd"}, "fff": []string{"cccc"},
+			},
+			key:  "bbb",
+			want: "",
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			xx := tt.h.get(tt.key)
+			assert.Equal(tt.want, xx)
+		})
+	}
+}
+
 func TestHeader_has(t *testing.T) {
 	assert := assert.New(t)
 	tests := []struct {