github.com/wangyougui/gf/v2@v2.6.5/util/gmode/gmode_z_unit_test.go (about)

     1  // Copyright GoFrame Author(https://goframe.org). All Rights Reserved.
     2  //
     3  // This Source Code Form is subject to the terms of the MIT License.
     4  // If a copy of the MIT was not distributed with this file,
     5  // You can obtain one at https://github.com/wangyougui/gf.
     6  
     7  // go test *.go -bench=".*"
     8  
     9  package gmode_test
    10  
    11  import (
    12  	"testing"
    13  
    14  	"github.com/wangyougui/gf/v2/test/gtest"
    15  	"github.com/wangyougui/gf/v2/util/gmode"
    16  )
    17  
    18  func Test_AutoCheckSourceCodes(t *testing.T) {
    19  	gtest.C(t, func(t *gtest.T) {
    20  		t.Assert(gmode.IsDevelop(), true)
    21  	})
    22  }
    23  
    24  func Test_Set(t *testing.T) {
    25  	gtest.C(t, func(t *gtest.T) {
    26  		oldMode := gmode.Mode()
    27  		defer gmode.Set(oldMode)
    28  		gmode.SetDevelop()
    29  		t.Assert(gmode.IsDevelop(), true)
    30  		t.Assert(gmode.IsTesting(), false)
    31  		t.Assert(gmode.IsStaging(), false)
    32  		t.Assert(gmode.IsProduct(), false)
    33  	})
    34  	gtest.C(t, func(t *gtest.T) {
    35  		oldMode := gmode.Mode()
    36  		defer gmode.Set(oldMode)
    37  		gmode.SetTesting()
    38  		t.Assert(gmode.IsDevelop(), false)
    39  		t.Assert(gmode.IsTesting(), true)
    40  		t.Assert(gmode.IsStaging(), false)
    41  		t.Assert(gmode.IsProduct(), false)
    42  	})
    43  	gtest.C(t, func(t *gtest.T) {
    44  		oldMode := gmode.Mode()
    45  		defer gmode.Set(oldMode)
    46  		gmode.SetStaging()
    47  		t.Assert(gmode.IsDevelop(), false)
    48  		t.Assert(gmode.IsTesting(), false)
    49  		t.Assert(gmode.IsStaging(), true)
    50  		t.Assert(gmode.IsProduct(), false)
    51  	})
    52  	gtest.C(t, func(t *gtest.T) {
    53  		oldMode := gmode.Mode()
    54  		defer gmode.Set(oldMode)
    55  		gmode.SetProduct()
    56  		t.Assert(gmode.IsDevelop(), false)
    57  		t.Assert(gmode.IsTesting(), false)
    58  		t.Assert(gmode.IsStaging(), false)
    59  		t.Assert(gmode.IsProduct(), true)
    60  	})
    61  }