omnivore/packages/api/test/util.ts

63 lines
1.6 KiB
TypeScript
Raw Normal View History

2024-01-31 11:10:57 +00:00
import { ConnectionOptions, Job, QueueEvents, Worker } from 'bullmq'
2024-04-18 14:09:04 +00:00
import { createServer } from 'http'
import { nanoid } from 'nanoid'
2022-02-11 17:24:33 +00:00
import supertest from 'supertest'
import { v4 } from 'uuid'
2024-04-09 08:49:03 +00:00
import { makeApolloServer } from '../src/apollo'
2024-01-31 11:10:57 +00:00
import { createWorker, QUEUE_NAME } from '../src/queue-processor'
2023-08-23 04:41:02 +00:00
import { createApp } from '../src/server'
import { corsConfig } from '../src/utils/corsConfig'
2022-02-11 17:24:33 +00:00
2024-04-09 08:49:03 +00:00
const app = createApp()
2024-04-18 14:09:04 +00:00
const httpServer = createServer(app)
const apollo = makeApolloServer(app, httpServer)
2022-02-11 17:24:33 +00:00
export const request = supertest(app)
2024-01-31 11:10:57 +00:00
let worker: Worker
let queueEvents: QueueEvents
2022-02-11 17:24:33 +00:00
export const startApolloServer = async () => {
2022-02-15 22:30:26 +00:00
await apollo.start()
apollo.applyMiddleware({ app, path: '/api/graphql', cors: corsConfig })
}
export const stopApolloServer = async () => {
await apollo.stop()
}
2022-02-15 22:30:26 +00:00
2024-02-22 04:03:58 +00:00
export const startWorker = (connection: ConnectionOptions) => {
2024-01-31 09:45:49 +00:00
worker = createWorker(connection)
2024-01-31 11:10:57 +00:00
queueEvents = new QueueEvents(QUEUE_NAME, {
connection,
})
2024-01-31 09:45:49 +00:00
}
export const stopWorker = async () => {
2024-02-22 04:03:58 +00:00
await queueEvents.close()
await worker.close()
2024-01-31 09:45:49 +00:00
}
2024-01-31 11:10:57 +00:00
export const waitUntilJobsDone = async (jobs: Job[]) => {
await Promise.all(jobs.map((job) => job.waitUntilFinished(queueEvents)))
2024-01-31 09:45:49 +00:00
}
2022-02-11 17:24:33 +00:00
export const graphqlRequest = (
query: string,
2023-10-20 02:06:28 +00:00
authToken: string,
variables?: Record<string, unknown>
2022-02-11 17:24:33 +00:00
): supertest.Test => {
return request
.post(apollo.graphqlPath)
2023-10-20 02:06:28 +00:00
.send({ query, variables })
2022-02-11 17:24:33 +00:00
.set('Accept', 'application/json')
2023-10-20 02:06:28 +00:00
.set('authorization', authToken)
2022-02-11 17:24:33 +00:00
.expect('Content-Type', /json/)
}
export const generateFakeUuid = () => {
return v4()
}
export const generateFakeShortId = () => {
return nanoid(8)
}