update

Author: jchrostek-dd

integration-tests/lambda/base-python/handler.py

@@ -1,5 +1,10 @@
+import logging
+
+logger = logging.getLogger()
+logger.setLevel(logging.INFO)
+
 def handler(event, context):
-    print('Hello world!')
+    logger.info('Hello world!')
     return {
         'statusCode': 200
     }

integration-tests/lib/base-node-stack.ts

@@ -7,7 +7,7 @@ export class BaseNodeStack extends cdk.Stack {
   constructor(scope: Construct, id: string, props: cdk.StackProps) {
     super(scope, id, props);
 
-    const nodeFunctionName = `${id}-node-function`
+    const nodeFunctionName = `${id}-lambda`
     const nodeFunction = new lambda.Function(this, nodeFunctionName, {
       runtime: lambda.Runtime.NODEJS_20_X,
       architecture: lambda.Architecture.ARM_64,

integration-tests/lib/base-python-stack.ts

@@ -7,7 +7,7 @@ export class BasePythonStack extends cdk.Stack {
   constructor(scope: Construct, id: string, props: cdk.StackProps) {
     super(scope, id, props);
 
-    const pythonFunctionName = `${id}-python-function`
+    const pythonFunctionName = `${id}-lambda`
     const pythonFunction = new lambda.Function(this, pythonFunctionName, {
       runtime: lambda.Runtime.PYTHON_3_12,
       architecture: lambda.Architecture.ARM_64,

integration-tests/tests/base-node.test.ts

@@ -2,7 +2,7 @@ import { invokeLambdaAndGetDatadogData } from './utils/util';
 import { getIdentifier } from './utils/config';
 
 describe('Base Node Lambda Integration Test', () => {
-  const NODE_FUNCTION_NAME = `integ-${getIdentifier()}-base-node-node-function`;
+  const NODE_FUNCTION_NAME = `integ-${getIdentifier()}-base-node-lambda`;
 
   it('should invoke Node.js Lambda successfully and receive logs and traces in Datadog', async () => {
 

integration-tests/tests/base-python.test.ts

@@ -2,7 +2,7 @@ import { invokeLambdaAndGetDatadogData } from './utils/util';
 import { getIdentifier } from './utils/config';
 
 describe('Base Python Lambda Integration Test', () => {
-  const PYTHON_FUNCTION_NAME = `integ-${getIdentifier()}-base-python-python-function`;
+  const PYTHON_FUNCTION_NAME = `integ-${getIdentifier()}-base-python-lambda`;
 
   it('should invoke Python Lambda successfully and receive logs and traces in Datadog', async () => {
 

integration-tests/tests/utils/util.ts

@@ -13,7 +13,7 @@ export async function invokeLambdaAndGetDatadogData(functionName: string, payloa
     const result = await invokeLambda(functionName, payload, coldStart, false);
 
     console.log('Waiting 10 minutes for logs and traces to be indexed in Datadog...');
-    await new Promise(resolve => setTimeout(resolve, 600000));
+    await new Promise(resolve => setTimeout(resolve, 300000));
 
     const traces = await getTraces(functionName, result.requestId);
     const logs = await getLogs(functionName, result.requestId);