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  }