woodpecker/server/logging/log_test.go

55 lines
956 B
Go
Raw Normal View History

2019-04-06 13:44:04 +00:00
package logging
import (
"context"
"sync"
"testing"
"time"
"github.com/stretchr/testify/assert"
2019-04-06 13:44:04 +00:00
)
func TestLogging(t *testing.T) {
var (
wg sync.WaitGroup
testPath = "test"
testEntry = &Entry{
Data: []byte("test"),
}
)
ctx, cancel := context.WithCancel(
context.Background(),
)
logger := New()
assert.NoError(t, logger.Open(ctx, testPath))
2019-04-06 13:44:04 +00:00
go func() {
assert.NoError(t, logger.Tail(ctx, testPath, func(entry ...*Entry) { wg.Done() }))
2019-04-06 13:44:04 +00:00
}()
go func() {
assert.NoError(t, logger.Tail(ctx, testPath, func(entry ...*Entry) { wg.Done() }))
2019-04-06 13:44:04 +00:00
}()
<-time.After(500 * time.Millisecond)
2019-04-06 13:44:04 +00:00
wg.Add(4)
go func() {
assert.NoError(t, logger.Write(ctx, testPath, testEntry))
assert.NoError(t, logger.Write(ctx, testPath, testEntry))
2019-04-06 13:44:04 +00:00
}()
wg.Wait()
wg.Add(1)
go func() {
assert.NoError(t, logger.Tail(ctx, testPath, func(entry ...*Entry) { wg.Done() }))
2019-04-06 13:44:04 +00:00
}()
<-time.After(500 * time.Millisecond)
2019-04-06 13:44:04 +00:00
wg.Wait()
cancel()
}