TUN-7126: Add Management logger io.Writer

management/logger_test.go

@@ -0,0 +1,75 @@
+package management
+
+import (
+	"testing"
+
+	"github.com/rs/zerolog"
+	"github.com/stretchr/testify/assert"
+)
+
+// No listening sessions will not write to the channel
+func TestLoggerWrite_NoSessions(t *testing.T) {
+	logger := NewLogger()
+	zlog := zerolog.New(logger).With().Timestamp().Logger().Level(zerolog.InfoLevel)
+
+	zlog.Info().Msg("hello")
+}
+
+// Validate that the session receives the event
+func TestLoggerWrite_OneSession(t *testing.T) {
+	logger := NewLogger()
+	zlog := zerolog.New(logger).With().Timestamp().Logger().Level(zerolog.InfoLevel)
+
+	session := logger.Listen()
+	defer logger.Close(session)
+	zlog.Info().Int("type", int(HTTP)).Msg("hello")
+	select {
+	case event := <-session.listener:
+		assert.Equal(t, "hello", event.Message)
+		assert.Equal(t, LogLevel("info"), event.Level)
+		assert.Equal(t, HTTP, event.Type)
+	default:
+		assert.Fail(t, "expected an event to be in the listener")
+	}
+}
+
+// Validate all sessions receive the same event
+func TestLoggerWrite_MultipleSessions(t *testing.T) {
+	logger := NewLogger()
+	zlog := zerolog.New(logger).With().Timestamp().Logger().Level(zerolog.InfoLevel)
+
+	session1 := logger.Listen()
+	defer logger.Close(session1)
+	session2 := logger.Listen()
+	zlog.Info().Int("type", int(HTTP)).Msg("hello")
+	for _, session := range []*Session{session1, session2} {
+		select {
+		case event := <-session.listener:
+			assert.Equal(t, "hello", event.Message)
+			assert.Equal(t, LogLevel("info"), event.Level)
+			assert.Equal(t, HTTP, event.Type)
+		default:
+			assert.Fail(t, "expected an event to be in the listener")
+		}
+	}
+
+	// Close session2 and make sure session1 still receives events
+	logger.Close(session2)
+	zlog.Info().Int("type", int(HTTP)).Msg("hello2")
+	select {
+	case event := <-session1.listener:
+		assert.Equal(t, "hello2", event.Message)
+		assert.Equal(t, LogLevel("info"), event.Level)
+		assert.Equal(t, HTTP, event.Type)
+	default:
+		assert.Fail(t, "expected an event to be in the listener")
+	}
+
+	// Make sure a held reference to session2 doesn't receive events after being closed
+	select {
+	case <-session2.listener:
+		assert.Fail(t, "An event was not expected to be in the session listener")
+	default:
+		// pass
+	}
+}