github.com/blend/go-sdk@v1.20220411.3/webutil/get_port_test.go (about)

     1  /*
     2  
     3  Copyright (c) 2022 - Present. Blend Labs, Inc. All rights reserved
     4  Use of this source code is governed by a MIT license that can be found in the LICENSE file.
     5  
     6  */
     7  
     8  package webutil
     9  
    10  import (
    11  	"net/http"
    12  	"testing"
    13  
    14  	"github.com/blend/go-sdk/assert"
    15  )
    16  
    17  func TestGetPort(t *testing.T) {
    18  	assert := assert.New(t)
    19  
    20  	assert.Empty(GetPort(nil))
    21  	assert.Empty(GetPort(&http.Request{}))
    22  	assert.Equal("8443", GetPort(&http.Request{
    23  		Header: http.Header{
    24  			HeaderXForwardedPort: {"8443"},
    25  		},
    26  	}), "should use existing header if found")
    27  	assert.Equal("8443", GetPort(&http.Request{
    28  		Header: http.Header{
    29  			HeaderXForwardedPort: {"9090,8443"},
    30  		},
    31  	}), "should use existing header last chunk if found")
    32  }