Merge pull request #978 from thinkerou/add_mode_case

add test case of mode
This commit is contained in:
Javier Provecho Fernandez 2017-07-05 17:03:03 +02:00 committed by GitHub
commit 316cb812c3
1 changed files with 6 additions and 1 deletions

View File

@ -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)