github.com/gogf/gf@v1.16.9/os/glog/glog_z_unit_level_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/gogf/gf.
     6  
     7  package glog
     8  
     9  import (
    10  	"bytes"
    11  	"github.com/gogf/gf/test/gtest"
    12  	"github.com/gogf/gf/text/gstr"
    13  	"testing"
    14  )
    15  
    16  func Test_LevelPrefix(t *testing.T) {
    17  	gtest.C(t, func(t *gtest.T) {
    18  		l := New()
    19  		t.Assert(l.GetLevelPrefix(LEVEL_DEBU), defaultLevelPrefixes[LEVEL_DEBU])
    20  		t.Assert(l.GetLevelPrefix(LEVEL_INFO), defaultLevelPrefixes[LEVEL_INFO])
    21  		t.Assert(l.GetLevelPrefix(LEVEL_NOTI), defaultLevelPrefixes[LEVEL_NOTI])
    22  		t.Assert(l.GetLevelPrefix(LEVEL_WARN), defaultLevelPrefixes[LEVEL_WARN])
    23  		t.Assert(l.GetLevelPrefix(LEVEL_ERRO), defaultLevelPrefixes[LEVEL_ERRO])
    24  		t.Assert(l.GetLevelPrefix(LEVEL_CRIT), defaultLevelPrefixes[LEVEL_CRIT])
    25  		l.SetLevelPrefix(LEVEL_DEBU, "debug")
    26  		t.Assert(l.GetLevelPrefix(LEVEL_DEBU), "debug")
    27  		l.SetLevelPrefixes(map[int]string{
    28  			LEVEL_CRIT: "critical",
    29  		})
    30  		t.Assert(l.GetLevelPrefix(LEVEL_DEBU), "debug")
    31  		t.Assert(l.GetLevelPrefix(LEVEL_INFO), defaultLevelPrefixes[LEVEL_INFO])
    32  		t.Assert(l.GetLevelPrefix(LEVEL_NOTI), defaultLevelPrefixes[LEVEL_NOTI])
    33  		t.Assert(l.GetLevelPrefix(LEVEL_WARN), defaultLevelPrefixes[LEVEL_WARN])
    34  		t.Assert(l.GetLevelPrefix(LEVEL_ERRO), defaultLevelPrefixes[LEVEL_ERRO])
    35  		t.Assert(l.GetLevelPrefix(LEVEL_CRIT), "critical")
    36  	})
    37  	gtest.C(t, func(t *gtest.T) {
    38  		buffer := bytes.NewBuffer(nil)
    39  		l := New()
    40  		l.SetWriter(buffer)
    41  		l.Debug("test1")
    42  		t.Assert(gstr.Contains(buffer.String(), defaultLevelPrefixes[LEVEL_DEBU]), true)
    43  
    44  		buffer.Reset()
    45  
    46  		l.SetLevelPrefix(LEVEL_DEBU, "debug")
    47  		l.Debug("test2")
    48  		t.Assert(gstr.Contains(buffer.String(), defaultLevelPrefixes[LEVEL_DEBU]), false)
    49  		t.Assert(gstr.Contains(buffer.String(), "debug"), true)
    50  
    51  		buffer.Reset()
    52  		l.SetLevelPrefixes(map[int]string{
    53  			LEVEL_ERRO: "error",
    54  		})
    55  		l.Error("test3")
    56  		t.Assert(gstr.Contains(buffer.String(), defaultLevelPrefixes[LEVEL_ERRO]), false)
    57  		t.Assert(gstr.Contains(buffer.String(), "error"), true)
    58  	})
    59  }