fix tests

This commit is contained in:
Hongbo Wu
2024-01-17 19:03:32 +08:00
committed by Jackson Harper
parent 5a7056b7ec
commit d558fe2b17
5 changed files with 16 additions and 24 deletions

View File

@ -1,10 +1,9 @@
import Redis from 'ioredis'
import { DataSource } from 'typeorm'
import { stringToHash } from '../../utils/helpers'
import { RssSubscriptionGroup } from '../../utils/createTask'
import { Job, Queue } from 'bullmq'
import { DataSource } from 'typeorm'
import { QUEUE_NAME } from '../../queue-processor'
import { redisDataSource } from '../../redis_data_source'
import { RssSubscriptionGroup } from '../../utils/createTask'
import { stringToHash } from '../../utils/helpers'
export const refreshAllFeeds = async (db: DataSource): Promise<boolean> => {
const subscriptionGroups = (await db.createEntityManager().query(

View File

@ -36,7 +36,7 @@ export class RedisDataSource {
async shutdown(): Promise<void> {
this.isInitialized = false
try {
// We only call quit on one redis connection as it tears all of them down
await this.redisClient?.quit()
await this.workerRedisClient?.quit()
} catch (err) {
console.error('error while shutting down redis')

View File

@ -1,19 +1,9 @@
/* eslint-disable @typescript-eslint/no-misused-promises */
import express from 'express'
import {
DEFAULT_SUBSCRIPTION_FOLDER,
Subscription,
} from '../../entity/subscription'
import { SubscriptionStatus, SubscriptionType } from '../../generated/graphql'
import { readPushSubscription } from '../../pubsub'
import { getRepository } from '../../repository'
import {
enqueueRssFeedFetch,
RssSubscriptionGroup,
} from '../../utils/createTask'
import { logger } from '../../utils/logger'
import { queueRSSRefreshAllFeedsJob } from '../../jobs/rss/refreshAllFeeds'
import { readPushSubscription } from '../../pubsub'
import { redisDataSource } from '../../redis_data_source'
import { logger } from '../../utils/logger'
export function rssFeedRouter() {
const router = express.Router()

View File

@ -1,4 +1,5 @@
import { appDataSource } from '../src/data_source'
import { env } from '../src/env'
import { redisDataSource } from '../src/redis_data_source'
import { stopApolloServer } from './util'
@ -9,6 +10,8 @@ export const mochaGlobalTeardown = async () => {
await appDataSource.destroy()
console.log('db connection closed')
await redisDataSource.shutdown()
console.log('redis connection closed')
if (env.redis.url) {
await redisDataSource.shutdown()
console.log('redis connection closed')
}
}

View File

@ -4,9 +4,9 @@ import sinon from 'sinon'
import sinonChai from 'sinon-chai'
import { User } from '../../src/entity/user'
import { SubscriptionType } from '../../src/generated/graphql'
import * as refreshAllFeeds from '../../src/jobs/rss/refreshAllFeeds'
import { createRssSubscriptions } from '../../src/services/subscriptions'
import { deleteUser } from '../../src/services/user'
import * as createTask from '../../src/utils/createTask'
import { createTestUser } from '../db'
import { request } from '../util'
@ -80,11 +80,11 @@ describe('Rss feeds Router', () => {
},
}
// fake enqueueRssFeedFetch function
// fake queueRSSRefreshAllFeedsJob function
const fake = sinon.replace(
createTask,
'enqueueRssFeedFetch',
sinon.fake.resolves('task name')
refreshAllFeeds,
'queueRSSRefreshAllFeedsJob',
sinon.fake()
)
const res = await request