autogits/bots-common/log_test.go

package common

import (
	"bytes"
	"regexp"
	"testing"
)

type TestWriter struct {
	str string
}

func TestLogging(t *testing.T) {
	t.Run("basic logging", func(t *testing.T) {
		var strWriter, errWriter *bytes.Buffer

		strWriter = bytes.NewBuffer(make([]byte, 0, 10000))
		errWriter = bytes.NewBuffer(make([]byte, 0, 10000))

		stdLogger, errLogger := CreateStdoutLogger(strWriter, errWriter)
		errLogger.Printf("%d\n", 100)
		stdLogger.Printf("OKA %d Done\n", 77)
		stdLogger.Println("Another line")

		const prefixMatch = `\[\d+\] `

		errStr := errWriter.String()
		if ok, err := regexp.MatchString("^"+prefixMatch+"100\n$", errStr); !ok {
			if err != nil {
				t.Logf("err: %v", err)
			}
			t.Fatal(errStr)
		}
		str := strWriter.String()
		if ok, err := regexp.MatchString("^"+prefixMatch+"OKA 77 Done\n"+prefixMatch+"Another line\n$", str); !ok {
			if err != nil {
				t.Logf("err: %v", err)
			}
			t.Fatal(str)
		}
	})
}