Use mode in all tests

api/message_test.go

@@ -15,6 +15,7 @@ import (
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/mock"
 	"github.com/stretchr/testify/suite"
+	"github.com/gotify/server/mode"
 )
 
 func TestMessageSuite(t *testing.T) {
@@ -31,7 +32,7 @@ type MessageSuite struct {
 }
 
 func (s *MessageSuite) BeforeTest(suiteName, testName string) {
-	gin.SetMode(gin.TestMode)
+	mode.Set(mode.TestDev)
 	s.recorder = httptest.NewRecorder()
 	s.ctx, _ = gin.CreateTestContext(s.recorder)
 	s.db = &apimock.MockMessageDatabase{}

api/token_test.go

@@ -14,6 +14,7 @@ import (
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/mock"
 	"github.com/stretchr/testify/suite"
+	"github.com/gotify/server/mode"
 )
 
 var (
@@ -36,7 +37,7 @@ type TokenSuite struct {
 }
 
 func (s *TokenSuite) BeforeTest(suiteName, testName string) {
-	gin.SetMode(gin.TestMode)
+	mode.Set(mode.TestDev)
 	rand.Seed(50)
 	s.recorder = httptest.NewRecorder()
 	s.ctx, _ = gin.CreateTestContext(s.recorder)

api/user_test.go

@@ -15,6 +15,7 @@ import (
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/mock"
 	"github.com/stretchr/testify/suite"
+	"github.com/gotify/server/mode"
 )
 
 var (
@@ -37,7 +38,7 @@ type UserSuite struct {
 }
 
 func (s *UserSuite) BeforeTest(suiteName, testName string) {
-	gin.SetMode(gin.TestMode)
+	mode.Set(mode.TestDev)
 	s.recorder = httptest.NewRecorder()
 	s.ctx, _ = gin.CreateTestContext(s.recorder)
 	s.db = &apimock.MockUserDatabase{}

auth/authentication_test.go

@@ -13,6 +13,7 @@ import (
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/mock"
 	"github.com/stretchr/testify/suite"
+	"github.com/gotify/server/mode"
 )
 
 func TestSuite(t *testing.T) {
@@ -26,7 +27,7 @@ type AuthenticationSuite struct {
 }
 
 func (s *AuthenticationSuite) SetupSuite() {
-	gin.SetMode(gin.TestMode)
+	mode.Set(mode.TestDev)
 	s.DB = &authmock.MockDatabase{}
 	s.auth = &Auth{s.DB}
 	s.DB.On("GetClientByToken", "clienttoken").Return(&model.Client{ID: 1, Token: "clienttoken", UserID: 1, Name: "android phone"})

auth/util_test.go

@@ -8,6 +8,7 @@ import (
 	"github.com/gotify/server/model"
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/suite"
+	"github.com/gotify/server/mode"
 )
 
 func TestUtilSuite(t *testing.T) {
@@ -19,7 +20,7 @@ type UtilSuite struct {
 }
 
 func (s *UtilSuite) BeforeTest(suiteName, testName string) {
-	gin.SetMode(gin.TestMode)
+	mode.Set(mode.TestDev)
 }
 
 func (s *UtilSuite) Test_getID() {

docs/swagger_test.go

@@ -6,10 +6,11 @@ import (
 
 	"github.com/gin-gonic/gin"
 	"github.com/stretchr/testify/assert"
+	"github.com/gotify/server/mode"
 )
 
 func TestServe(t *testing.T) {
-	gin.SetMode(gin.TestMode)
+	mode.Set(mode.TestDev)
 	recorder := httptest.NewRecorder()
 	ctx, _ := gin.CreateTestContext(recorder)
 	ctx.Request = httptest.NewRequest("GET", "/swagger", nil)

error/handler_test.go

@@ -10,10 +10,11 @@ import (
 	"github.com/gin-gonic/gin"
 	"github.com/gotify/server/model"
 	"github.com/stretchr/testify/assert"
+	"github.com/gotify/server/mode"
 )
 
 func TestDefaultErrorInternal(t *testing.T) {
-	gin.SetMode(gin.TestMode)
+	mode.Set(mode.TestDev)
 	rec := httptest.NewRecorder()
 	ctx, _ := gin.CreateTestContext(rec)
 	ctx.AbortWithError(500, errors.New("something went wrong"))
@@ -24,7 +25,7 @@ func TestDefaultErrorInternal(t *testing.T) {
 }
 
 func TestBindingErrorDefault(t *testing.T) {
-	gin.SetMode(gin.TestMode)
+	mode.Set(mode.TestDev)
 	rec := httptest.NewRecorder()
 	ctx, _ := gin.CreateTestContext(rec)
 	ctx.AbortWithError(400, errors.New("you need todo something")).SetType(gin.ErrorTypeBind)
@@ -35,7 +36,7 @@ func TestBindingErrorDefault(t *testing.T) {
 }
 
 func TestDefaultErrorBadRequest(t *testing.T) {
-	gin.SetMode(gin.TestMode)
+	mode.Set(mode.TestDev)
 	rec := httptest.NewRecorder()
 	ctx, _ := gin.CreateTestContext(rec)
 	ctx.AbortWithError(400, errors.New("you need todo something"))
@@ -51,7 +52,7 @@ type testValidate struct {
 }
 
 func TestValidationError(t *testing.T) {
-	gin.SetMode(gin.TestMode)
+	mode.Set(mode.TestDev)
 	rec := httptest.NewRecorder()
 	ctx, _ := gin.CreateTestContext(rec)
 	ctx.Request = httptest.NewRequest("GET", "/uri", nil)

error/notfound_test.go

@@ -5,10 +5,11 @@ import (
 	"testing"
 
 	"github.com/gin-gonic/gin"
+	"github.com/gotify/server/mode"
 )
 
 func TestNotFound(t *testing.T) {
-	gin.SetMode(gin.TestMode)
+	mode.Set(mode.TestDev)
 	rec := httptest.NewRecorder()
 	ctx, _ := gin.CreateTestContext(rec)
 

stream/stream_test.go

@@ -21,6 +21,7 @@ import (
 	"github.com/gotify/server/auth"
 	"github.com/gotify/server/model"
 	"github.com/stretchr/testify/assert"
+	"github.com/gotify/server/mode"
 )
 
 func TestFailureOnNormalHttpRequest(t *testing.T) {
@@ -296,7 +297,7 @@ func (c *testingClient) expectNoMessage() {
 }
 
 func bootTestServer(handlerFunc gin.HandlerFunc) (*httptest.Server, *API) {
-	gin.SetMode(gin.TestMode)
+	mode.Set(mode.TestDev)
 
 	r := gin.New()
 	r.Use(handlerFunc)