restore message queue in tests

rfcx · Oct 20, 2023 · 1e8f833 · 1e8f833
1 parent 68258a8
commit 1e8f833
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/core/classifier-jobs/update.int.test.js b/core/classifier-jobs/update.int.test.js
@@ -7,6 +7,9 @@ const { WAITING, RUNNING, DONE, CANCELLED, ERROR } = require('./classifier-job-s
 const defaultMessageQueue = require('../../common/message-queue/sqs')
 const { CLASSIFIER_JOB_FINISHED } = require('../../common/message-queue/event-names')
 
+const isEnabledOrig = defaultMessageQueue.isEnabled
+const publishOrig = defaultMessageQueue.publish
+
 // Test data
 const CLASSIFIER_1 = { id: 831, name: 'sounds of the underground', version: 1, externalId: '555666', createdById: seedValues.primaryUserId, modelRunner: 'tf2', modelUrl: '???', lastExecutedAt: null, isPublic: true }
 const CLASSIFIERS = [CLASSIFIER_1]
@@ -73,6 +76,8 @@ afterEach(async () => {
 })
 
 afterAll(async () => {
+  defaultMessageQueue.isEnabled = isEnabledOrig
+  defaultMessageQueue.publish = publishOrig
   await models.sequelize.close()
 })
 

diff --git a/core/stream-segments/dao/index.unit.test.js b/core/stream-segments/dao/index.unit.test.js
@@ -3,6 +3,9 @@ jest.mock('../../roles/dao', () => { return {} })
 jest.mock('../../../common/message-queue/sqs')
 const defaultMessageQueue = require('../../../common/message-queue/sqs')
 
+const isEnabledOrig = defaultMessageQueue.isEnabled
+const publishOrig = defaultMessageQueue.publish
+
 const service = require('.')
 const { SEGMENT_CREATED } = require('../../../common/message-queue/event-names')
 
@@ -11,6 +14,11 @@ beforeEach(() => {
   defaultMessageQueue.publish = jest.fn().mockReturnValue(Promise.resolve())
 })
 
+afterAll(() => {
+  defaultMessageQueue.isEnabled = isEnabledOrig
+  defaultMessageQueue.publish = publishOrig
+})
+
 test('Create segment not queued when default queue not enabled', async () => {
   defaultMessageQueue.isEnabled.mockReturnValue(false)
   const segment = { id: 'a5e5', start: '2021-04-18T12:15:00.000Z', stream_id: '13d781bd' }