mirror of https://github.com/XTLS/Xray-core
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
52 lines
1.3 KiB
52 lines
1.3 KiB
package log_test |
|
|
|
import ( |
|
"context" |
|
"testing" |
|
|
|
"github.com/golang/mock/gomock" |
|
"github.com/xtls/xray-core/app/log" |
|
"github.com/xtls/xray-core/common" |
|
clog "github.com/xtls/xray-core/common/log" |
|
"github.com/xtls/xray-core/testing/mocks" |
|
) |
|
|
|
func TestCustomLogHandler(t *testing.T) { |
|
mockCtl := gomock.NewController(t) |
|
defer mockCtl.Finish() |
|
|
|
var loggedValue []string |
|
|
|
mockHandler := mocks.NewLogHandler(mockCtl) |
|
mockHandler.EXPECT().Handle(gomock.Any()).AnyTimes().DoAndReturn(func(msg clog.Message) { |
|
loggedValue = append(loggedValue, msg.String()) |
|
}) |
|
|
|
log.RegisterHandlerCreator(log.LogType_Console, func(lt log.LogType, options log.HandlerCreatorOptions) (clog.Handler, error) { |
|
return mockHandler, nil |
|
}) |
|
|
|
logger, err := log.New(context.Background(), &log.Config{ |
|
ErrorLogLevel: clog.Severity_Debug, |
|
ErrorLogType: log.LogType_Console, |
|
AccessLogType: log.LogType_None, |
|
}) |
|
common.Must(err) |
|
|
|
common.Must(logger.Start()) |
|
|
|
clog.Record(&clog.GeneralMessage{ |
|
Severity: clog.Severity_Debug, |
|
Content: "test", |
|
}) |
|
|
|
if len(loggedValue) < 2 { |
|
t.Fatal("expected 2 log messages, but actually ", loggedValue) |
|
} |
|
|
|
if loggedValue[1] != "[Debug] test" { |
|
t.Fatal("expected '[Debug] test', but actually ", loggedValue[1]) |
|
} |
|
|
|
common.Must(logger.Close()) |
|
}
|
|
|