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 }