Skip to content

Commit 49c19bb

Browse files
committed
updated e2e tests
1 parent 9e52277 commit 49c19bb

File tree

5 files changed

+13
-13
lines changed

5 files changed

+13
-13
lines changed

Diff for: packages/tracing/tests/e2e/tracer.test.Decorator.ts

+3-3
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
import { Tracer } from '../../src';
22
import { Callback, Context } from 'aws-lambda';
3-
import { DynamoDBClient, ScanCommand } from '@aws-sdk/client-dynamodb';
3+
import { DynamoDBClient, PutItemCommand } from '@aws-sdk/client-dynamodb';
44
// eslint-disable-next-line @typescript-eslint/no-var-requires
55
let AWS = require('aws-sdk');
66

@@ -52,8 +52,8 @@ export class MyFunctionWithDecorator {
5252
}
5353

5454
return Promise.all([
55-
dynamoDBv2.scan({ TableName: testTableName }).promise(),
56-
dynamoDBv3.send(new ScanCommand({ TableName: testTableName })),
55+
dynamoDBv2.put({ TableName: testTableName, Item: { id: `${serviceName}-${event.invocation}-sdkv2` } }).promise(),
56+
dynamoDBv3.send(new PutItemCommand({ TableName: testTableName, Item: { id: { 'S': `${serviceName}-${event.invocation}-sdkv3` } } })),
5757
new Promise((resolve, reject) => {
5858
setTimeout(() => {
5959
const res = this.myMethod();

Diff for: packages/tracing/tests/e2e/tracer.test.DecoratorWithAsyncHandler.ts

+3-3
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
import { Tracer } from '../../src';
22
import { Context } from 'aws-lambda';
3-
import { DynamoDBClient, ScanCommand } from '@aws-sdk/client-dynamodb';
3+
import { DynamoDBClient, PutItemCommand } from '@aws-sdk/client-dynamodb';
44
// eslint-disable-next-line @typescript-eslint/no-var-requires
55
let AWS = require('aws-sdk');
66

@@ -52,13 +52,13 @@ export class MyFunctionWithDecorator {
5252
}
5353

5454
try {
55-
await dynamoDBv2.scan({ TableName: testTableName }).promise();
55+
await dynamoDBv2.put({ TableName: testTableName, Item: { id: `${serviceName}-${event.invocation}-sdkv2` } }).promise();
5656
} catch (err) {
5757
console.error(err);
5858
}
5959

6060
try {
61-
await dynamoDBv3.send(new ScanCommand({ TableName: testTableName }));
61+
await dynamoDBv3.send(new PutItemCommand({ TableName: testTableName, Item: { id: { 'S': `${serviceName}-${event.invocation}-sdkv3` } } }));
6262
} catch (err) {
6363
console.error(err);
6464
}

Diff for: packages/tracing/tests/e2e/tracer.test.Manual.ts

+3-3
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
import { Tracer } from '../../src';
22
import { Context } from 'aws-lambda';
3-
import { DynamoDBClient, ScanCommand } from '@aws-sdk/client-dynamodb';
3+
import { DynamoDBClient, PutItemCommand } from '@aws-sdk/client-dynamodb';
44
// eslint-disable-next-line @typescript-eslint/no-var-requires
55
let AWS = require('aws-sdk');
66

@@ -55,13 +55,13 @@ export const handler = async (event: CustomEvent, _context: Context): Promise<vo
5555
dynamoDBv2 = new AWS.DynamoDB.DocumentClient();
5656
}
5757
try {
58-
await dynamoDBv2.scan({ TableName: testTableName }).promise();
58+
await dynamoDBv2.put({ TableName: testTableName, Item: { id: `${serviceName}-${event.invocation}-sdkv2` } }).promise();
5959
} catch (err) {
6060
console.error(err);
6161
}
6262

6363
try {
64-
await dynamoDBv3.send(new ScanCommand({ TableName: testTableName }));
64+
await dynamoDBv3.send(new PutItemCommand({ TableName: testTableName, Item: { id: { 'S': `${serviceName}-${event.invocation}-sdkv3` } } }));
6565
} catch (err) {
6666
console.error(err);
6767
}

Diff for: packages/tracing/tests/e2e/tracer.test.Middleware.ts

+3-3
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
import middy from '@middy/core';
22
import { captureLambdaHandler, Tracer } from '../../src';
33
import { Context } from 'aws-lambda';
4-
import { DynamoDBClient, ScanCommand } from '@aws-sdk/client-dynamodb';
4+
import { DynamoDBClient, PutItemCommand } from '@aws-sdk/client-dynamodb';
55
// eslint-disable-next-line @typescript-eslint/no-var-requires
66
let AWS = require('aws-sdk');
77

@@ -49,13 +49,13 @@ export const handler = middy(async (event: CustomEvent, _context: Context): Prom
4949
dynamoDBv2 = new AWS.DynamoDB.DocumentClient();
5050
}
5151
try {
52-
await dynamoDBv2.scan({ TableName: testTableName }).promise();
52+
await dynamoDBv2.put({ TableName: testTableName, Item: { id: `${serviceName}-${event.invocation}-sdkv2` } }).promise();
5353
} catch (err) {
5454
console.error(err);
5555
}
5656

5757
try {
58-
await dynamoDBv3.send(new ScanCommand({ TableName: testTableName }));
58+
await dynamoDBv3.send(new PutItemCommand({ TableName: testTableName, Item: { id: { 'S': `${serviceName}-${event.invocation}-sdkv3` } } }));
5959
} catch (err) {
6060
console.error(err);
6161
}

Diff for: packages/tracing/tests/e2e/tracer.test.ts

+1-1
Original file line numberDiff line numberDiff line change
@@ -92,7 +92,7 @@ describe('Tracer integration tests', () => {
9292
},
9393
timeout: Duration.seconds(30),
9494
});
95-
table.grantReadData(fn);
95+
table.grantWriteData(fn);
9696
invocationsMap[functionName] = {
9797
serviceName: expectedServiceName,
9898
resourceArn: `arn:aws:lambda:${region}:${account}:function:${functionName}`, // ARN is still a token at this point, so we construct the ARN manually

0 commit comments

Comments
 (0)