From 2b148122c8d086f607fb95896f645c54303f00ac Mon Sep 17 00:00:00 2001 From: Kostadin Plachkov Date: Mon, 1 Oct 2018 21:55:33 +0300 Subject: [PATCH] Fix the tests --- mode_test.go | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/mode_test.go b/mode_test.go index cf27acd8..0748de01 100644 --- a/mode_test.go +++ b/mode_test.go @@ -6,6 +6,7 @@ package gin import ( "os" + "sync/atomic" "testing" "github.com/gin-gonic/gin/binding" @@ -17,24 +18,24 @@ func init() { } func TestSetMode(t *testing.T) { - assert.Equal(t, testCode, ginMode) + assert.Equal(t, int32(testCode), atomic.LoadInt32(&ginMode)) assert.Equal(t, TestMode, Mode()) os.Unsetenv(ENV_GIN_MODE) SetMode("") - assert.Equal(t, debugCode, ginMode) + assert.Equal(t, int32(debugCode), atomic.LoadInt32(&ginMode)) assert.Equal(t, DebugMode, Mode()) SetMode(DebugMode) - assert.Equal(t, debugCode, ginMode) + assert.Equal(t, int32(debugCode), atomic.LoadInt32(&ginMode)) assert.Equal(t, DebugMode, Mode()) SetMode(ReleaseMode) - assert.Equal(t, releaseCode, ginMode) + assert.Equal(t, int32(releaseCode), atomic.LoadInt32(&ginMode)) assert.Equal(t, ReleaseMode, Mode()) SetMode(TestMode) - assert.Equal(t, testCode, ginMode) + assert.Equal(t, int32(testCode), atomic.LoadInt32(&ginMode)) assert.Equal(t, TestMode, Mode()) assert.Panics(t, func() { SetMode("unknown") })