@@ -6,9 +6,12 @@ const { after, afterEach, before, describe, it } = require('mocha')
66const sinon = require ( 'sinon' )
77const semver = require ( 'semver' )
88
9- const { assertObjectContains } = require ( '../../../integration-tests/helpers' )
10- const { withVersions } = require ( '../../dd-trace/test/setup/mocha' )
9+ const { computePathwayHash } = require ( '../../dd-trace/src/datastreams/pathway' )
10+ const { ENTRY_PARENT_HASH } = require ( '../../dd-trace/src/datastreams/processor' )
11+ const propagationHash = require ( '../../dd-trace/src/propagation-hash' )
1112const agent = require ( '../../dd-trace/test/plugins/agent' )
13+ const { withVersions } = require ( '../../dd-trace/test/setup/mocha' )
14+ const { assertObjectContains } = require ( '../../../integration-tests/helpers' )
1215const { setup } = require ( './spec_helpers' )
1316
1417describe ( 'Sns' , function ( ) {
@@ -71,8 +74,8 @@ describe('Sns', function () {
7174 }
7275
7376 describe ( 'Data Streams Monitoring' , ( ) => {
74- const expectedProducerHash = '15386798273908484982'
75- const expectedConsumerHash = '15162998336469814920'
77+ let expectedProducerHash
78+ let expectedConsumerHash
7679 let nowStub
7780
7881 before ( ( ) => {
@@ -84,7 +87,28 @@ describe('Sns', function () {
8487 tracer = require ( '../../dd-trace' )
8588 tracer . use ( 'aws-sdk' , { sns : { dsmEnabled : true } , sqs : { dsmEnabled : true } } )
8689
87- createResources ( 'TestQueueDSM' , 'TestTopicDSM' , done )
90+ createResources ( 'TestQueueDSM' , 'TestTopicDSM' , ( err ) => {
91+ if ( err ) return done ( err )
92+
93+ const phash = propagationHash . getHash ( )
94+ const producerHash = computePathwayHash (
95+ 'test' , 'tester' ,
96+ [ 'direction:out' , `topic:${ TopicArn } ` , 'type:sns' ] ,
97+ ENTRY_PARENT_HASH ,
98+ phash
99+ )
100+ expectedProducerHash = producerHash . readBigUInt64LE ( 0 ) . toString ( )
101+
102+ const queueName = QueueUrl . split ( '/' ) . pop ( )
103+ expectedConsumerHash = computePathwayHash (
104+ 'test' , 'tester' ,
105+ [ 'direction:in' , `topic:${ queueName } ` , 'type:sqs' ] ,
106+ producerHash ,
107+ phash
108+ ) . readBigUInt64LE ( 0 ) . toString ( )
109+
110+ done ( )
111+ } )
88112 } )
89113
90114 after ( done => {
0 commit comments