Fixes yarn rw jobs clear command

Closes #11577
redwoodjs · Sep 17, 2024 · a18bd56 · a18bd56
1 parent 2b74931
commit a18bd56
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 5 deletions.
diff --git a/packages/jobs/src/bins/__tests__/rw-jobs.test.ts b/packages/jobs/src/bins/__tests__/rw-jobs.test.ts
@@ -1,7 +1,7 @@
 import { beforeEach, describe, expect, it, vi } from 'vitest'
 
 import { mockLogger } from '../../core/__tests__/mocks.js'
-import { buildNumWorkers, startWorkers } from '../rw-jobs.js'
+import { buildNumWorkers, clearQueue, startWorkers } from '../rw-jobs.js'
 
 vi.mock('@redwoodjs/cli-helpers/loadEnvFiles', () => {
   return {
@@ -123,3 +123,24 @@ describe('startWorkers()', () => {
     )
   })
 })
+
+describe('clearQueue()', () => {
+  beforeEach(() => {
+    vi.resetAllMocks()
+  })
+
+  it('forks a single worker', () => {
+    const mockWorker = {
+      on: () => {},
+    }
+    mocks.fork.mockImplementation(() => mockWorker)
+
+    clearQueue({ logger: mockLogger })
+
+    // single worker only
+    expect(mocks.fork).toHaveBeenCalledWith(
+      expect.stringContaining('rw-jobs-worker.js'),
+      ['--clear', '--index', '0', '--id', '0'],
+    )
+  })
+})
diff --git a/packages/jobs/src/bins/rw-jobs.ts b/packages/jobs/src/bins/rw-jobs.ts
@@ -30,6 +30,8 @@ loadEnvFiles()
 
 process.title = 'rw-jobs'
 
+const WORKER_PATH = path.join(__dirname, 'rw-jobs-worker.js')
+
 const parseArgs = (argv: string[]) => {
   const commandString = hideBin(argv)
 
@@ -126,7 +128,7 @@ export const startWorkers = ({
     }
 
     // fork the worker process
-    const worker = fork(path.join(__dirname, 'rw-jobs-worker.js'), workerArgs, {
+    const worker = fork(WORKER_PATH, workerArgs, {
       detached: detach,
       stdio: detach ? 'ignore' : 'inherit',
       env: process.env,
@@ -144,7 +146,7 @@ export const startWorkers = ({
 }
 
 // TODO add support for stopping with SIGTERM or SIGKILL?
-const stopWorkers = async ({
+export const stopWorkers = async ({
   numWorkers,
   signal = 'SIGINT',
   logger,
@@ -175,9 +177,9 @@ const stopWorkers = async ({
   }
 }
 
-const clearQueue = ({ logger }: { logger: BasicLogger }) => {
+export const clearQueue = ({ logger }: { logger: BasicLogger }) => {
   logger.warn(`Starting worker to clear job queue...`)
-  fork(path.join(__dirname, 'rw-jobs-worker.js'), ['--clear'])
+  fork(WORKER_PATH, ['--clear', '--index', '0', '--id', '0'])
 }
 
 const signalSetup = ({