Skip to content

Commit b3fdcb4

Browse files
committed
test(kernel): adjust worker tests
1 parent 0be603a commit b3fdcb4

3 files changed

Lines changed: 29 additions & 13 deletions

File tree

src/worker/WorkerTaskBuilder.ts

Lines changed: 0 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -32,11 +32,6 @@ export class WorkerTaskBuilder {
3232
*/
3333
connection?: string
3434

35-
/**
36-
* The interval instance of the worker task.
37-
*/
38-
interval?: NodeJS.Timeout
39-
4035
/**
4136
* Define if the worker task is registered.
4237
*/
@@ -51,11 +46,6 @@ export class WorkerTaskBuilder {
5146
* The handler of the worker task.
5247
*/
5348
handler?: (ctx: Context) => any | Promise<any>
54-
55-
/**
56-
* Define if the worker task is running.
57-
*/
58-
isRunning?: boolean
5949
} = {}
6050

6151
private timers: NodeJS.Timeout[] = []

tests/unit/drivers/AwsSqsDriverTest.ts

Lines changed: 21 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,8 +11,8 @@ import { Is, Path } from '@athenna/common'
1111
import { EnvHelper } from '@athenna/config'
1212
import { LoggerProvider } from '@athenna/logger'
1313
import { BaseTest } from '#tests/helpers/BaseTest'
14-
import { Queue, QueueProvider, WorkerProvider } from '#src'
15-
import { Test, type Context, BeforeEach, AfterEach, Skip } from '@athenna/test'
14+
import { Queue, WorkerProvider, QueueProvider } from '#src'
15+
import { Test, type Context, BeforeEach, AfterEach, Skip, AfterAll } from '@athenna/test'
1616

1717
export class AwsSqsDriverTest extends BaseTest {
1818
@BeforeEach()
@@ -36,6 +36,25 @@ export class AwsSqsDriverTest extends BaseTest {
3636
Config.clear()
3737
}
3838

39+
@AfterAll()
40+
public async afterAll() {
41+
await Config.loadAll(Path.fixtures('config'))
42+
43+
new QueueProvider().register()
44+
new WorkerProvider().register()
45+
new LoggerProvider().register()
46+
47+
await Queue.connection('aws_sqs').truncate().catch()
48+
49+
await Queue.closeAll()
50+
51+
Queue.worker().close()
52+
53+
ioc.reconstruct()
54+
55+
Config.clear()
56+
}
57+
3958
@Test()
4059
public async shouldBeAbleToConnectToDriver({ assert }: Context) {
4160
Queue.connection('aws_sqs')

tests/unit/providers/WorkerProviderTest.ts

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,14 @@ export class WorkerProviderTest {
4545
new QueueProvider().register()
4646
new WorkerProvider().register()
4747

48-
assert.lengthOf(Queue.worker().getWorkerTasks(), 0)
48+
new QueueProvider().shutdown()
49+
new WorkerProvider().shutdown()
50+
51+
assert.isTrue(
52+
Queue.worker()
53+
.getWorkerTasks()
54+
.every(worker => worker.worker.isRegistered === false)
55+
)
4956
}
5057

5158
@Test()

0 commit comments

Comments
 (0)