25 lines
450 B
Go
25 lines
450 B
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)
|
|
}
|