Function TestBatchHandlerWriteAll

tools/logger/batch_handler_test.go:222–253

(t *testing.T)

}

func TestBatchHandlerWriteAll(t *testing.T) {
	ctx := context.Background()

	beforeLogs := []*Log{}
	writeLogs := []*Log{}

	h := NewBatchHandler(BatchOptions{
		BatchSize: 3,
		BeforeAddFunc: func(_ context.Context, log *Log) bool {
			beforeLogs = append(beforeLogs, log)
			return true
		},
		WriteFunc: func(_ context.Context, logs []*Log) error {
			writeLogs = logs
			return nil
		},
	})

	h.Handle(ctx, slog.NewRecord(time.Now(), slog.LevelInfo, "test1", 0))
	h.Handle(ctx, slog.NewRecord(time.Now(), slog.LevelInfo, "test2", 0))

	checkLogMessages([]string{"test1", "test2"}, beforeLogs, t)
	checkLogMessages([]string{"test1", "test2"}, h.logs, t)
	checkLogMessages([]string{}, writeLogs, t) // empty because the batch size hasn't been reached

	// force trigger the batch write
	h.WriteAll(ctx)

	checkLogMessages([]string{"test1", "test2"}, beforeLogs, t)
	checkLogMessages([]string{}, h.logs, t) // reset
	checkLogMessages([]string{"test1", "test2"}, writeLogs, t)
}

func TestBatchHandlerAttrsFormat(t *testing.T) {
	ctx := context.Background()