Browse Source

Merge pull request #978 from thinkerou/add_mode_case

add test case of mode
Javier Provecho Fernandez 8 years ago
parent
commit
316cb812c3
1 changed files with 6 additions and 1 deletions
  1. 6 1
      mode_test.go

+ 6 - 1
mode_test.go

@@ -5,16 +5,21 @@
 package gin
 
 import (
+	"os"
 	"testing"
 
 	"github.com/stretchr/testify/assert"
 )
 
 func init() {
-	SetMode(TestMode)
+	os.Setenv(ENV_GIN_MODE, TestMode)
 }
 
 func TestSetMode(t *testing.T) {
+	assert.Equal(t, ginMode, testCode)
+	assert.Equal(t, Mode(), TestMode)
+	os.Unsetenv(ENV_GIN_MODE)
+
 	SetMode(DebugMode)
 	assert.Equal(t, ginMode, debugCode)
 	assert.Equal(t, Mode(), DebugMode)