Function TestBatchHandlerHandle

tools/logger/batch_handler_test.go:175–220

(t *testing.T)

}

func TestBatchHandlerHandle(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)

			// skip test2 log
			return log.Message != "test2"
		},
		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))
	h.Handle(ctx, slog.NewRecord(time.Now(), slog.LevelInfo, "test3", 0))

	// no batch write
	{
		checkLogMessages([]string{"test1", "test2", "test3"}, beforeLogs, t)

		checkLogMessages([]string{"test1", "test3"}, h.logs, t)

		// should be empty because no batch write has happened yet
		if totalWriteLogs := len(writeLogs); totalWriteLogs != 0 {
			t.Fatalf("Expected %d writeLogs, got %d", 0, totalWriteLogs)
		}
	}

	// add one more log to trigger the batch write
	{
		h.Handle(ctx, slog.NewRecord(time.Now(), slog.LevelInfo, "test4", 0))

		// should be empty after the batch write
		checkLogMessages([]string{}, h.logs, t)

		checkLogMessages([]string{"test1", "test3", "test4"}, writeLogs, t)
	}
}

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