pspy/internal/logging/logging_test.go

package logging

import (
	"bytes"
	"log"
	"reflect"
	"regexp"
	"testing"
)

const dateFormatPattern = `[\d]{4}/[\d]{2}/[\d]{2} [\d]{2}:[\d]{2}:[\d]{2}`
const ansiPattern = "[\u001B\u009B][[\\]()#;?]*(?:(?:(?:[a-zA-Z\\d]*(?:;[a-zA-Z\\d]*)*)?\u0007)|(?:(?:\\d{1,4}(?:;\\d{0,4})*)?[\\dA-PRZcf-ntqry=><~]))"

var ansiMatcher = regexp.MustCompile(ansiPattern)

var l = NewLogger()

var logTests = []struct {
	logger      *log.Logger
	test        func()
	expectation string
	colors      [][]byte
}{
	{l.infoLogger, func() { l.Infof("Info message no. %d", 1) }, "Info message no. 1\n", nil},
	{l.infoLogger, func() { l.Infof("Info message no. %d with a string %s\n", 2, "appended to it") }, "Info message no. 2 with a string appended to it\n", nil},
	{l.errorLogger, func() { l.Errorf("Error message") }, "Error message\n", nil},
	{l.errorLogger, func() { l.Errorf("Error message\n") }, "Error message\n", nil},
	{l.eventLogger, func() { l.Eventf(ColorNone, "Event message") }, dateFormatPattern + " Event message\n", nil},
	{l.eventLogger, func() { l.Eventf(ColorRed, "Event message") }, dateFormatPattern + " Event message\n", [][]byte{[]byte("\x1b[31;1m"), []byte("\x1b[0m")}},
	{l.eventLogger, func() { l.Eventf(ColorGreen, "Event message") }, dateFormatPattern + " Event message\n", [][]byte{[]byte("\x1b[32;1m"), []byte("\x1b[0m")}},
}

func TestLogging(t *testing.T) {
	for i, tt := range logTests {
		actual := captureOutput(tt.logger, tt.test)

		// check colors and remove afterwards
		colors := ansiMatcher.FindAll(actual, 2)
		if !reflect.DeepEqual(colors, tt.colors) {
			t.Errorf("[%d] Wrong colors: got %+v but want %+v", i, colors, tt.colors)
		}
		actual = ansiMatcher.ReplaceAll(actual, []byte(""))

		// check contents
		matcher := regexp.MustCompile(tt.expectation)
		if !matcher.Match(actual) {
			t.Errorf("[%d] Wrong message logged!: got %s but wanted %v", i, actual, matcher)
		}
	}
}

func captureOutput(logger *log.Logger, f func()) []byte {
	var buf bytes.Buffer
	logger.SetOutput(&buf)
	f()
	return buf.Bytes()
}