github.com/hduhelp/go-zero@v1.4.3/gateway/internal/headerprocessor_test.go (about) 1 package internal 2 3 import ( 4 "net/http" 5 "net/http/httptest" 6 "testing" 7 8 "github.com/stretchr/testify/assert" 9 ) 10 11 func TestBuildHeadersNoValue(t *testing.T) { 12 req := httptest.NewRequest("GET", "/", http.NoBody) 13 req.Header.Add("a", "b") 14 assert.Nil(t, ProcessHeaders(req.Header)) 15 } 16 17 func TestBuildHeadersWithValues(t *testing.T) { 18 req := httptest.NewRequest("GET", "/", http.NoBody) 19 req.Header.Add("grpc-metadata-a", "b") 20 req.Header.Add("grpc-metadata-b", "b") 21 assert.ElementsMatch(t, []string{"gateway-A:b", "gateway-B:b"}, ProcessHeaders(req.Header)) 22 }