X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=E2Manager%2Flogger%2Flogger_test.go;h=d1d00d27563d735076705c1370b4471fe4facb0d;hb=086691323016a409a53df7ca6f52339641a19063;hp=9921ce18996f4b47d38e2ed479fe6c88ff24a490;hpb=2c9d450c93e4f0b312a68d39429efb685ec02022;p=ric-plt%2Fe2mgr.git

diff --git a/E2Manager/logger/logger_test.go b/E2Manager/logger/logger_test.go
index 9921ce1..d1d00d2 100644
--- a/E2Manager/logger/logger_test.go
+++ b/E2Manager/logger/logger_test.go
@@ -44,6 +44,34 @@ func TestInitInfoLoggerSuccess(t *testing.T) {
 	assert.True(t, log.Logger.Core().Enabled(zap.InfoLevel))
 }
 
+func TestInitWarnLoggerSuccess(t *testing.T) {
+	log, err := InitLogger(WarnLevel)
+	assert.Nil(t, err)
+	assert.NotNil(t, log)
+	assert.True(t, log.Logger.Core().Enabled(zap.WarnLevel))
+}
+
+func TestInitErrorLoggerSuccess(t *testing.T) {
+	log, err := InitLogger(ErrorLevel)
+	assert.Nil(t, err)
+	assert.NotNil(t, log)
+	assert.True(t, log.Logger.Core().Enabled(zap.ErrorLevel))
+}
+
+func TestInitDPanicLoggerSuccess(t *testing.T) {
+	log, err := InitLogger(DPanicLevel)
+	assert.Nil(t, err)
+	assert.NotNil(t, log)
+	assert.True(t, log.Logger.Core().Enabled(zap.DPanicLevel))
+}
+
+func TestInitPanicLoggerSuccess(t *testing.T) {
+	log, err := InitLogger(PanicLevel)
+	assert.Nil(t, err)
+	assert.NotNil(t, log)
+	assert.True(t, log.Logger.Core().Enabled(zap.PanicLevel))
+}
+
 func TestInitInfoLoggerFailure(t *testing.T) {
 	log, err := InitLogger(99)
 	assert.NotNil(t, err)
@@ -63,7 +91,7 @@ func TestSyncSuccess(t *testing.T){
 	}
 	err = log.Sync()
 	assert.Nil(t, err)
-	
+
 	os.Stdout = old
 	logFile, err = os.Open("./loggerTest.txt")
 	if err != nil{
@@ -144,6 +172,22 @@ func TestDebugfFatalLevel(t *testing.T)  {
 	assert.False(t,validateRecordExists(FatalLevel, zap.DebugLevel, t))
 }
 
+func TestWarnfWarnLevel(t *testing.T)  {
+	assert.True(t,validateRecordExists(WarnLevel, zap.WarnLevel, t))
+}
+
+func TestWarnfDebugLevel(t *testing.T)  {
+	assert.True(t,validateRecordExists(DebugLevel, zap.WarnLevel, t))
+}
+
+func TestWarnfInfoLevel(t *testing.T)  {
+	assert.True(t,validateRecordExists(InfoLevel, zap.WarnLevel, t))
+}
+
+func TestWarnfFatalLevel(t *testing.T)  {
+	assert.False(t,validateRecordExists(FatalLevel, zap.WarnLevel, t))
+}
+
 func TestLogLevelTokenToLevel(t *testing.T) {
 	level, ok := LogLevelTokenToLevel("deBug")
 	assert.True(t, ok)
@@ -218,14 +262,16 @@ func validateRecordExists(logLevel LogLevel, recordLevel zapcore.Level, t *testi
 		t.Errorf("logger_test.TestSyncFailure - failed to initialize logger, error: %s", err)
 	}
 	switch recordLevel{
-		case  zap.DebugLevel:
-			log.Debugf("%v, %v, %v", 1, "abc", 0.1)
-		case zap.InfoLevel:
-			log.Infof("%v, %v, %v", 1, "abc", 0.1)
-		case zap.ErrorLevel:
-			log.Errorf("%v, %v, %v", 1, "abc", 0.1)
-		case zap.DPanicLevel:
-			log.DPanicf("%v, %v, %v", 1, "abc", 0.1)
+	case  zap.DebugLevel:
+		log.Debugf("%v, %v, %v", 1, "abc", 0.1)
+	case zap.InfoLevel:
+		log.Infof("%v, %v, %v", 1, "abc", 0.1)
+	case zap.WarnLevel:
+		log.Warnf("%v, %v, %v", 1, "abc", 0.1)
+	case zap.ErrorLevel:
+		log.Errorf("%v, %v, %v", 1, "abc", 0.1)
+	case zap.DPanicLevel:
+		log.DPanicf("%v, %v, %v", 1, "abc", 0.1)
 	}
 	err = w.Close()
 	if err != nil {
@@ -244,4 +290,4 @@ func validateRecordExists(logLevel LogLevel, recordLevel zapcore.Level, t *testi
 		s,_= buf.ReadString('\n')
 	}
 	return entryNum == 1
-}
+}
\ No newline at end of file