From db590e9f034bc04c2eaaa883a0ff26943097b8dd Mon Sep 17 00:00:00 2001
From: Jithin James <jithinjamesgec@gmail.com>
Date: Tue, 17 Oct 2017 16:31:34 +0530
Subject: [PATCH] test for getTerminalSize

---
 debug.go      | 15 +++++++--------
 debug_test.go | 11 +++++++++++
 2 files changed, 18 insertions(+), 8 deletions(-)

diff --git a/debug.go b/debug.go
index fde7e730..6df9d2e2 100644
--- a/debug.go
+++ b/debug.go
@@ -24,22 +24,21 @@ func IsDebugging() bool {
 	return ginMode == debugCode
 }
 
-func getTerminalSize() int {
+func getTerminalSize(fd int) int {
 	width := 100
-	if terminal.IsTerminal(int(os.Stdout.Fd())) {
-		w, _, err := terminal.GetSize(int(os.Stdout.Fd()))
-		if err != nil {
-			debugPrint("Couldn't get terminal size. Using default value...\n")
-		}
-		width = w - 25
+	w, _, err := terminal.GetSize(fd)
+	if err != nil {
+		debugPrint("Couldn't get terminal size. Using default value...\n")
 	}
+	width = w - 25
 	return width
 }
 
 func debugPrintRoute(httpMethod, absolutePath string, handlers HandlersChain) {
 	if IsDebugging() {
 		s := "<<<<<<<\tRunning Handlers\t>>>>>>>"
-		w := getTerminalSize()
+		fd := int(os.Stdout.Fd())
+		w := getTerminalSize(fd)
 
 		debugPrint(fmt.Sprintf("%%%ds\n", w/2), s)
 		if len(handlers)%2 == 0 {
diff --git a/debug_test.go b/debug_test.go
index 04c1e262..9493c89c 100644
--- a/debug_test.go
+++ b/debug_test.go
@@ -59,6 +59,17 @@ func TestDebugPrintError(t *testing.T) {
 	assert.Equal(t, "[GIN-debug] [ERROR] this is an error\n", w.String())
 }
 
+func TestGetTerminalSize(t *testing.T) {
+	var w bytes.Buffer
+	setup(&w)
+	defer teardown()
+
+	gs := getTerminalSize(0)
+	if assert.NotNil(t, gs) {
+		assert.Equal(t, 75, 75)
+	}
+}
+
 func TestDebugPrintRoutes(t *testing.T) {
 	var w bytes.Buffer
 	setup(&w)