fix test

DataDog · Sep 23, 2024 · d09f6f9 · d09f6f9
1 parent c691443
commit d09f6f9
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/packages/dd-trace/test/appsec/index.sequelize.plugin.spec.js b/packages/dd-trace/test/appsec/index.sequelize.plugin.spec.js
@@ -30,7 +30,7 @@ describe('sequelize', () => {
         // close agent
         after(() => {
           appsec.disable()
-          return agent.close()
+          return agent.close({ ritmReset: false })
         })
 
         // init database

diff --git a/packages/dd-trace/test/appsec/rasp/sql_injection.spec.js b/packages/dd-trace/test/appsec/rasp/sql_injection.spec.js
@@ -1,6 +1,6 @@
 'use strict'
 
-const { pgQueryStart, mysql2ConnectionQueryStart } = require('../../../src/appsec/channels')
+const { pgQueryStart, mysql2OuterQueryStart } = require('../../../src/appsec/channels')
 const addresses = require('../../../src/appsec/addresses')
 const proxyquire = require('proxyquire')
 
@@ -122,7 +122,7 @@ describe('RASP - sql_injection', () => {
       const req = {}
       datadogCore.storage.getStore.returns({ req })
 
-      mysql2ConnectionQueryStart.publish(ctx)
+      mysql2OuterQueryStart.publish(ctx)
 
       const persistent = {
         [addresses.DB_STATEMENT]: 'SELECT 1',
@@ -140,7 +140,7 @@ describe('RASP - sql_injection', () => {
       const req = {}
       datadogCore.storage.getStore.returns({ req })
 
-      mysql2ConnectionQueryStart.publish(ctx)
+      mysql2OuterQueryStart.publish(ctx)
 
       sinon.assert.notCalled(waf.run)
     })
@@ -151,7 +151,7 @@ describe('RASP - sql_injection', () => {
       }
       datadogCore.storage.getStore.returns(undefined)
 
-      mysql2ConnectionQueryStart.publish(ctx)
+      mysql2OuterQueryStart.publish(ctx)
 
       sinon.assert.notCalled(waf.run)
     })
@@ -162,7 +162,7 @@ describe('RASP - sql_injection', () => {
       }
       datadogCore.storage.getStore.returns({})
 
-      mysql2ConnectionQueryStart.publish(ctx)
+      mysql2OuterQueryStart.publish(ctx)
 
       sinon.assert.notCalled(waf.run)
     })
@@ -173,7 +173,7 @@ describe('RASP - sql_injection', () => {
       }
       datadogCore.storage.getStore.returns({})
 
-      mysql2ConnectionQueryStart.publish(ctx)
+      mysql2OuterQueryStart.publish(ctx)
 
       sinon.assert.notCalled(waf.run)
     })