onasty: internal/transport/http/reqid/reqid_test.go (c2e1526cc824db72296b1282bee4df957421640a)

onasty/internal/transport/http/reqid/reqid_test.go (view raw)

Oleksandr Smirnov refactor(usersrv): remove unsed dep, 1 year ago

package reqid

import (
	"net/http"
	"net/http/httptest"
	"testing"

	"github.com/gin-gonic/gin"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

//nolint:gochecknoinits
func init() {
	gin.SetMode(gin.TestMode)
}

func testHandler(c *gin.Context) {
	c.Status(http.StatusOK)
}

func TestMiddleware(t *testing.T) {
	r := gin.New()
	r.Use(Middleware())
	r.GET("/", testHandler)

	w := httptest.NewRecorder()
	req, err := http.NewRequest(http.MethodGet, "/", nil)
	require.NoError(t, err)

	r.ServeHTTP(w, req)

	assert.Equal(t, http.StatusOK, w.Code)
	assert.NotEmpty(t, w.Header().Get(headerRequestID))
}

func BenchmarkMiddleware(b *testing.B) {
	r := gin.New()
	r.Use(Middleware())
	r.GET("/", testHandler)

	w := httptest.NewRecorder()
	req, err := http.NewRequest(http.MethodGet, "/", nil)
	require.NoError(b, err)

	for b.Loop() {
		r.ServeHTTP(w, req)
	}
}

1	package reqid
2
3	import (
4	"net/http"
5	"net/http/httptest"
6	"testing"
7
8	"github.com/gin-gonic/gin"
9	"github.com/stretchr/testify/assert"
10	"github.com/stretchr/testify/require"
11	)
12
13	//nolint:gochecknoinits
14	func init() {
15	gin.SetMode(gin.TestMode)
16	}
17
18	func testHandler(c *gin.Context) {
19	c.Status(http.StatusOK)
20	}
21
22	func TestMiddleware(t *testing.T) {
23	r := gin.New()
24	r.Use(Middleware())
25	r.GET("/", testHandler)
26
27	w := httptest.NewRecorder()
28	req, err := http.NewRequest(http.MethodGet, "/", nil)
29	require.NoError(t, err)
30
31	r.ServeHTTP(w, req)
32
33	assert.Equal(t, http.StatusOK, w.Code)
34	assert.NotEmpty(t, w.Header().Get(headerRequestID))
35	}
36
37	func BenchmarkMiddleware(b *testing.B) {
38	r := gin.New()
39	r.Use(Middleware())
40	r.GET("/", testHandler)
41
42	w := httptest.NewRecorder()
43	req, err := http.NewRequest(http.MethodGet, "/", nil)
44	require.NoError(b, err)
45
46	for b.Loop() {
47	r.ServeHTTP(w, req)
48	}
49	}