github.com/lingyao2333/mo-zero@v1.4.1/rest/pathvar/params_test.go (about)

     1  package pathvar
     2  
     3  import (
     4  	"net/http"
     5  	"strings"
     6  	"testing"
     7  
     8  	"github.com/stretchr/testify/assert"
     9  )
    10  
    11  func TestVars(t *testing.T) {
    12  	expect := map[string]string{
    13  		"a": "1",
    14  		"b": "2",
    15  	}
    16  	r, err := http.NewRequest(http.MethodGet, "/", nil)
    17  	assert.Nil(t, err)
    18  	r = WithVars(r, expect)
    19  	assert.EqualValues(t, expect, Vars(r))
    20  }
    21  
    22  func TestVarsNil(t *testing.T) {
    23  	r, err := http.NewRequest(http.MethodGet, "/", nil)
    24  	assert.Nil(t, err)
    25  	assert.Nil(t, Vars(r))
    26  }
    27  
    28  func TestContextKey(t *testing.T) {
    29  	ck := contextKey("hello")
    30  	assert.True(t, strings.Contains(ck.String(), "hello"))
    31  }