forked from flashcat/categraf
92 lines
2.2 KiB
Go
92 lines
2.2 KiB
Go
// Copyright The OpenTelemetry Authors
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package telemetrylogs
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"go.uber.org/zap"
|
|
"go.uber.org/zap/zapcore"
|
|
|
|
"go.opentelemetry.io/collector/config"
|
|
)
|
|
|
|
func TestGRPCLogger(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
cfg config.ServiceTelemetryLogs
|
|
infoLogged bool
|
|
warnLogged bool
|
|
}{
|
|
{
|
|
"collector_info_level_grpc_log_warn",
|
|
config.ServiceTelemetryLogs{
|
|
Level: zapcore.InfoLevel,
|
|
Encoding: "console",
|
|
},
|
|
false,
|
|
true,
|
|
},
|
|
{
|
|
"collector_debug_level_grpc_log_debug",
|
|
config.ServiceTelemetryLogs{
|
|
Level: zapcore.DebugLevel,
|
|
Encoding: "console",
|
|
},
|
|
true,
|
|
true,
|
|
},
|
|
{
|
|
"collector_warn_level_grpc_log_warn",
|
|
config.ServiceTelemetryLogs{
|
|
Development: false, // this must set the grpc loggerV2 to loggerV2
|
|
Level: zapcore.WarnLevel,
|
|
Encoding: "console",
|
|
},
|
|
false,
|
|
true,
|
|
},
|
|
}
|
|
for _, test := range tests {
|
|
t.Run(test.name, func(t *testing.T) {
|
|
obsInfo, obsWarn := false, false
|
|
hook := zap.Hooks(func(entry zapcore.Entry) error {
|
|
switch entry.Level {
|
|
case zapcore.InfoLevel:
|
|
obsInfo = true
|
|
case zapcore.WarnLevel:
|
|
obsWarn = true
|
|
}
|
|
return nil
|
|
})
|
|
|
|
// create new collector zap logger
|
|
logger, err := NewLogger(test.cfg, []zap.Option{hook})
|
|
assert.NoError(t, err)
|
|
|
|
// create colGRPCLogger
|
|
glogger := SetColGRPCLogger(logger, test.cfg.Level)
|
|
assert.NotNil(t, glogger)
|
|
|
|
glogger.Info(test.name)
|
|
glogger.Warning(test.name)
|
|
|
|
assert.Equal(t, obsInfo, test.infoLogged)
|
|
assert.Equal(t, obsWarn, test.warnLogged)
|
|
})
|
|
}
|
|
}
|