diff --git a/plugins/node/opentelemetry-instrumentation-aws-sdk/test/aws-sdk-v3.test.ts b/plugins/node/opentelemetry-instrumentation-aws-sdk/test/aws-sdk-v3.test.ts index 63203e257b..10e2d322e3 100644 --- a/plugins/node/opentelemetry-instrumentation-aws-sdk/test/aws-sdk-v3.test.ts +++ b/plugins/node/opentelemetry-instrumentation-aws-sdk/test/aws-sdk-v3.test.ts @@ -278,7 +278,7 @@ describe('instrumentation-aws-sdk-v3', () => { }); }); - describe.skip('custom service behavior', () => { + describe('custom service behavior', () => { describe('SQS', () => { const sqsClient = new SQS({ region }); @@ -384,7 +384,7 @@ describe('instrumentation-aws-sdk-v3', () => { ); }); - it('sqs receive add messaging attributes and context', done => { + it('sqs receive add messaging attributes', done => { nock(`https://sqs.${region}.amazonaws.com/`) .post('/') .reply( @@ -412,7 +412,27 @@ describe('instrumentation-aws-sdk-v3', () => { 'SQS' ); expect(span.attributes[AttributeNames.AWS_REGION]).toEqual(region); + expect(span.attributes[SemanticAttributes.HTTP_STATUS_CODE]).toEqual( + 200 + ); + done(); + }); + }); + + it.skip('sqs receive context', done => { + nock(`https://sqs.${region}.amazonaws.com/`) + .post('/') + .reply( + 200, + fs.readFileSync('./test/mock-responses/sqs-receive.xml', 'utf8') + ); + const params = { + QueueUrl: + 'https://sqs.us-east-1.amazonaws.com/731241200085/otel-demo-aws-sdk', + MaxNumberOfMessages: 3, + }; + sqsClient.receiveMessage(params).then(res => { const receiveCallbackSpan = trace.getSpan(context.active()); expect(receiveCallbackSpan).toBeDefined(); const attributes = (receiveCallbackSpan as unknown as ReadableSpan) @@ -420,9 +440,6 @@ describe('instrumentation-aws-sdk-v3', () => { expect(attributes[SemanticAttributes.MESSAGING_OPERATION]).toMatch( MessagingOperationValues.RECEIVE ); - expect(span.attributes[SemanticAttributes.HTTP_STATUS_CODE]).toEqual( - 200 - ); done(); }); });