From 5f1e881e3eedc00f81c4b5b66d3dc88ea427976b Mon Sep 17 00:00:00 2001
From: Roc Chan <rockerchen@tencent.com>
Date: Wed, 24 Jul 2019 20:15:11 +0800
Subject: [PATCH] Add unit test for SetServer

---
 gin_test.go | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/gin_test.go b/gin_test.go
index 11bdd79c..5757c8a1 100644
--- a/gin_test.go
+++ b/gin_test.go
@@ -176,6 +176,14 @@ func TestCreateEngine(t *testing.T) {
 	assert.Empty(t, router.Handlers)
 }
 
+func TestSetServer(t *testing.T) {
+	router := New()
+	router.SetServer(&http.Server{IdleTimeout: 5 * time.Second})
+	assert.Equal(t, "/", router.basePath)
+	assert.Equal(t, router.engine, router)
+	assert.Empty(t, router.Handlers)
+}
+
 func TestLoadHTMLFilesTestMode(t *testing.T) {
 	ts := setupHTMLFiles(
 		t,