Pārlūkot izejas kodu

Merge pull request #978 from thinkerou/add_mode_case

add test case of mode
Javier Provecho Fernandez 8 gadi atpakaļ
vecāks
revīzija
316cb812c3
1 mainītis faili ar 6 papildinājumiem un 1 dzēšanām
  1. 6 1
      mode_test.go

+ 6 - 1
mode_test.go

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