Function TestLogHandler_WithFields

lambdacontext/logger_test.go:255–291

(t *testing.T)

}

func TestLogHandler_WithFields(t *testing.T) {
	var buf bytes.Buffer

	opts := &slog.HandlerOptions{
		Level:       slog.LevelInfo,
		ReplaceAttr: ReplaceAttr,
	}
	baseHandler := slog.NewJSONHandler(&buf, opts)

	// Create options with fields
	options := &logOptions{}
	WithFunctionARN()(options)
	WithTenantID()(options)

	handler := &lambdaHandler{
		handler: baseHandler,
		fields:  options.fields,
	}

	lc := &LambdaContext{
		AwsRequestID:       "test-request-123",
		InvokedFunctionArn: "arn:aws:lambda:us-east-1:123456789:function:test",
		TenantID:           "tenant-abc",
	}
	ctx := NewContext(context.Background(), lc)

	logger := slog.New(handler)
	logger.InfoContext(ctx, "test message")

	var logOutput map[string]interface{}
	err := json.Unmarshal(buf.Bytes(), &logOutput)
	require.NoError(t, err)

	assert.Equal(t, "test-request-123", logOutput["requestId"])
	assert.Equal(t, "arn:aws:lambda:us-east-1:123456789:function:test", logOutput["functionArn"])
	assert.Equal(t, "tenant-abc", logOutput["tenantId"])
}

func TestLogHandler_WithFieldFunctionARNOnly(t *testing.T) {
	var buf bytes.Buffer