github.com/argoproj/argo-cd/v3@v3.2.1/util/http/http_test.go (about)

     1  package http
     2  
     3  import (
     4  	"net/http"
     5  	"strings"
     6  	"testing"
     7  
     8  	"github.com/stretchr/testify/assert"
     9  	"github.com/stretchr/testify/require"
    10  )
    11  
    12  func TestCookieMaxLength(t *testing.T) {
    13  	cookies, err := MakeCookieMetadata("foo", "bar")
    14  	require.NoError(t, err)
    15  	assert.Equal(t, "foo=bar", cookies[0])
    16  
    17  	// keys will be of format foo, foo-1, foo-2 ..
    18  	cookies, err = MakeCookieMetadata("foo", strings.Repeat("_", (maxCookieLength-5)*maxCookieNumber))
    19  	require.EqualError(t, err, "the authentication token is 81760 characters long and requires 21 cookies but the max number of cookies is 20. Contact your Argo CD administrator to increase the max number of cookies")
    20  	assert.Empty(t, cookies)
    21  }
    22  
    23  func TestCookieWithAttributes(t *testing.T) {
    24  	flags := []string{"SameSite=lax", "httpOnly"}
    25  
    26  	cookies, err := MakeCookieMetadata("foo", "bar", flags...)
    27  	require.NoError(t, err)
    28  	assert.Equal(t, "foo=bar; SameSite=lax; httpOnly", cookies[0])
    29  }
    30  
    31  func TestSplitCookie(t *testing.T) {
    32  	cookieValue := strings.Repeat("_", (maxCookieLength-6)*4)
    33  	cookies, err := MakeCookieMetadata("foo", cookieValue)
    34  	require.NoError(t, err)
    35  	assert.Len(t, cookies, 4)
    36  	assert.Len(t, strings.Split(cookies[0], "="), 2)
    37  	token := strings.Split(cookies[0], "=")[1]
    38  	assert.Len(t, strings.Split(token, ":"), 2)
    39  	assert.Equal(t, "4", strings.Split(token, ":")[0])
    40  
    41  	cookies = append(cookies, "bar=this-entry-should-be-filtered")
    42  	var cookieList []*http.Cookie
    43  	for _, cookie := range cookies {
    44  		parts := strings.Split(cookie, "=")
    45  		cookieList = append(cookieList, &http.Cookie{Name: parts[0], Value: parts[1]})
    46  	}
    47  	token, err = JoinCookies("foo", cookieList)
    48  	require.NoError(t, err)
    49  	assert.Equal(t, cookieValue, token)
    50  }
    51  
    52  // TestRoundTripper just copy request headers to the resposne.
    53  type TestRoundTripper struct{}
    54  
    55  func (rt TestRoundTripper) RoundTrip(req *http.Request) (*http.Response, error) {
    56  	resp := http.Response{}
    57  	resp.Header = http.Header{}
    58  	for k, vs := range req.Header {
    59  		for _, v := range vs {
    60  			resp.Header.Add(k, v)
    61  		}
    62  	}
    63  	return &resp, nil
    64  }
    65  
    66  func TestTransportWithHeader(t *testing.T) {
    67  	client := &http.Client{}
    68  	req, _ := http.NewRequest(http.MethodGet, "/foo", http.NoBody)
    69  	req.Header.Set("Bar", "req_1")
    70  	req.Header.Set("Foo", "req_1")
    71  
    72  	// No default headers.
    73  	client.Transport = &TransportWithHeader{
    74  		RoundTripper: &TestRoundTripper{},
    75  	}
    76  	resp, err := client.Do(req)
    77  	require.NoError(t, err)
    78  	assert.Equal(t, http.Header{
    79  		"Bar": []string{"req_1"},
    80  		"Foo": []string{"req_1"},
    81  	}, resp.Header)
    82  
    83  	// with default headers.
    84  	client.Transport = &TransportWithHeader{
    85  		RoundTripper: &TestRoundTripper{},
    86  		Header: http.Header{
    87  			"Foo": []string{"default_1", "default_2"},
    88  		},
    89  	}
    90  	resp, err = client.Do(req)
    91  	require.NoError(t, err)
    92  	assert.Equal(t, http.Header{
    93  		"Bar": []string{"req_1"},
    94  		"Foo": []string{"default_1", "default_2", "req_1"},
    95  	}, resp.Header)
    96  }