Update Logging E2E test to use log buffering.

Author: phipag

powertools-e2e-tests/handlers/logging/src/main/java/software/amazon/lambda/powertools/e2e/Function.java

@@ -14,12 +14,15 @@
 
 package software.amazon.lambda.powertools.e2e;
 
-import com.amazonaws.services.lambda.runtime.Context;
-import com.amazonaws.services.lambda.runtime.RequestHandler;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import org.slf4j.MDC;
+
+import com.amazonaws.services.lambda.runtime.Context;
+import com.amazonaws.services.lambda.runtime.RequestHandler;
+
 import software.amazon.lambda.powertools.logging.Logging;
+import software.amazon.lambda.powertools.logging.PowertoolsLogging;
 
 public class Function implements RequestHandler<Input, String> {
     private static final Logger LOG = LoggerFactory.getLogger(Function.class);
@@ -29,6 +32,9 @@ public String handleRequest(Input input, Context context) {
         input.getKeys().forEach(MDC::put);
         LOG.info(input.getMessage());
 
+        // Flush buffer manually since we buffer at INFO level to test log buffering
+        PowertoolsLogging.flushBuffer();
+
         return "OK";
     }
-}
+}

powertools-e2e-tests/handlers/logging/src/main/resources/log4j2.xml

@@ -4,13 +4,14 @@
         <Console name="JsonAppender" target="SYSTEM_OUT">
             <JsonTemplateLayout eventTemplateUri="classpath:LambdaJsonLayout.json" />
         </Console>
+        <!-- We buffer everything to implicitly test buffer flushing -->
+        <BufferingAppender name="BufferedAppender" bufferAtVerbosity="INFO">
+            <AppenderRef ref="JsonAppender" />
+        </BufferingAppender>
     </Appenders>
     <Loggers>
         <Root level="INFO">
-            <AppenderRef ref="JsonAppender"/>
+            <AppenderRef ref="BufferedAppender" />
         </Root>
-        <Logger name="JsonLogger" level="INFO" additivity="false">
-            <AppenderRef ref="JsonAppender"/>
-        </Logger>
     </Loggers>
-</Configuration>
+</Configuration>