Skip to content

Commit 50f9403

Browse files
committed
Fix unit test
1 parent 75404b4 commit 50f9403

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

packages/dd-trace/test/appsec/iast/analyzers/sql-injection-analyzer.spec.js

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -203,24 +203,24 @@ describe('sql-injection-analyzer', () => {
203203

204204
it('should call analyze on apm:mysql:query:start', () => {
205205
const onMysqlQueryStart = sqlInjectionAnalyzer._subscriptions[0]._handler
206-
207-
onMysqlQueryStart({ sql: 'SELECT 1', name: 'apm:mysql:query:start' })
206+
const payload = { sql: 'SELECT 1'}
207+
onMysqlQueryStart({ payload, name: 'apm:mysql:query:start' })
208208

209209
expect(analyze).to.be.calledOnceWith('SELECT 1')
210210
})
211211

212212
it('should call analyze on apm:mysql2:query:start', () => {
213-
const onMysql2QueryStart = sqlInjectionAnalyzer._subscriptions[0]._handler
213+
const onMysql2QueryStart = sqlInjectionAnalyzer._subscriptions[1]._handler
214214

215215
onMysql2QueryStart({ sql: 'SELECT 1', name: 'apm:mysql2:query:start' })
216216

217217
expect(analyze).to.be.calledOnceWith('SELECT 1')
218218
})
219219

220220
it('should call analyze on apm:pg:query:start', () => {
221-
const onPgQueryStart = sqlInjectionAnalyzer._subscriptions[0]._handler
221+
const onPgQueryStart = sqlInjectionAnalyzer._subscriptions[2]._handler
222222

223-
onPgQueryStart({ sql: 'SELECT 1', name: 'apm:pg:query:start' })
223+
onPgQueryStart({ query: { text: 'SELECT 1' }, name: 'apm:pg:query:start' })
224224

225225
expect(analyze).to.be.calledOnceWith('SELECT 1')
226226
})

0 commit comments

Comments
 (0)