From 640dfcf88f7dad0c6ef607b8d02b4b68f202a59a Mon Sep 17 00:00:00 2001
From: Bogdan Ungureanu <ubogdan@gmail.com>
Date: Tue, 2 Nov 2021 22:38:53 +0200
Subject: [PATCH] add unit test

---
 swagger_test.go | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/swagger_test.go b/swagger_test.go
index aa6a0d1..fd23221 100644
--- a/swagger_test.go
+++ b/swagger_test.go
@@ -111,6 +111,18 @@ func TestDisablingCustomWrapHandler(t *testing.T) {
 	assert.Equal(t, 404, w11.Code)
 }
 
+func TestNamedCustomWrapHandler(t *testing.T) {
+	router := gin.New()
+	gin.SetMode(gin.TestMode)
+
+	swag.Register("test", &mockedSwag{})
+
+	router.GET("/swagger/*any", CustomWrapHandler(&Config{InstanceName: "test"}, swaggerFiles.Handler))
+
+	w1 := performRequest("GET", "/swagger/doc.json", router)
+	assert.Equal(t, 200, w1.Code)
+}
+
 func TestWithGzipMiddleware(t *testing.T) {
 	gin.SetMode(gin.TestMode)
 	router := gin.New()