github.com/blend/go-sdk@v1.20220411.3/r2/opt_header_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 r2
     9  
    10  import (
    11  	"net/http"
    12  	"testing"
    13  
    14  	"github.com/blend/go-sdk/assert"
    15  )
    16  
    17  func TestOptHeader(t *testing.T) {
    18  	assert := assert.New(t)
    19  
    20  	opt := OptHeader(http.Header{"Foo": []string{"bar"}})
    21  
    22  	req := New("https://foo.bar.local")
    23  	assert.Nil(opt(req))
    24  
    25  	assert.NotNil(req.Request.Header)
    26  	assert.Equal("bar", req.Request.Header.Get("foo"))
    27  }
    28  
    29  func TestOptHeaderValue(t *testing.T) {
    30  	assert := assert.New(t)
    31  
    32  	opt := OptHeaderValue("Foo", "bar")
    33  
    34  	req := New("https://foo.bar.local")
    35  	assert.Nil(opt(req))
    36  
    37  	assert.NotNil(req.Request.Header)
    38  	assert.Equal("bar", req.Request.Header.Get("foo"))
    39  }