cloudflared/logger/manager_test.go

package logger

import (
	"testing"

	"github.com/stretchr/testify/assert"
)

type outputFunc func(b []byte)

func (f outputFunc) WriteLogLine(data []byte) {
	f(data)
}

func TestWriteManger(t *testing.T) {
	testData := []byte(string("hello Austin, how are you doing?"))
	waitChan := make(chan []byte)
	m := NewWriteManager()
	m.Append(testData, outputFunc(func(b []byte) {
		waitChan <- b
	}))
	resp := <-waitChan
	assert.Equal(t, testData, resp)
}