starlog/starlog_test.go

package starlog

import (
	"bytes"
	"strings"
	"sync"
	"testing"
)

func Test_LOG(t *testing.T) {
	var buf bytes.Buffer
	logger := NewStarlog(&buf)
	logger.SetShowStd(false)
	logger.SetShowColor(false)
	logger.SetShowOriginFile(false)
	logger.SetShowFuncName(false)
	logger.SetShowFlag(false)

	var wait sync.WaitGroup
	wait.Add(2)
	go func() {
		defer wait.Done()
		logger.Debugln("nb")
	}()
	go func() {
		defer wait.Done()
		logger.Debugln("nb2")
	}()
	wait.Wait()

	logStr := buf.String()
	if !strings.Contains(logStr, "nb") || !strings.Contains(logStr, "nb2") {
		t.Fatalf("expected both logs in output, got %q", logStr)
	}
}