Procházet zdrojové kódy

Merge pull request #978 from thinkerou/add_mode_case

add test case of mode
Javier Provecho Fernandez před 9 roky
rodič
revize
316cb812c3
1 změnil soubory, kde provedl 6 přidání a 1 odebrání
  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)