/**
 * @Author: aesoper
 * @Description: 请求Id中间件测试
 * @File:  request_id_test
 * @Version: 1.0.0
 * @Date: 2020/5/19 19:13
 */

package gin_middleware

import (
	"gitee.com/gin-ecosystem/gin-middleware/consts"
	"github.com/gin-gonic/gin"
	"github.com/stretchr/testify/assert"
	"net/http"
	"net/http/httptest"
	"testing"
)

func TestNewRequestID(t *testing.T) {
	e := gin.New()
	e.GET("/", NewRequestID(RequestIDConfig{
		Skipper: DefaultSkipper,
		Generator: func() string {
			return "test-header"
		},
	}), func(c *gin.Context) {
		c.String(200, "test")
		return
	})

	req := httptest.NewRequest(http.MethodGet, "/", nil)
	rec := httptest.NewRecorder()

	e.ServeHTTP(rec, req)

	assert.Equal(t, rec.Header().Get(consts.HeaderXRequestID), "test-header")
}