This commit is contained in:
Hongbo Wu 2022-11-24 13:38:13 +08:00
parent 85d319c3c7
commit 4a3537cdc2
2 changed files with 13 additions and 2 deletions

View file

@ -231,3 +231,13 @@ export const updateTestUser = async (userId: string, update: Partial<User>) => {
await t.getRepository(User).update(userId, update)
})
}
export const deleteTestDeviceTokens = async (
userId: string,
criteria: string[] | FindOptionsWhere<UserDeviceToken>
) => {
await AppDataSource.transaction(async (t) => {
await setClaims(t, userId)
await t.getRepository(UserDeviceToken).delete(criteria)
})
}

View file

@ -1,6 +1,7 @@
import {
createTestDeviceToken,
createTestUser,
deleteTestDeviceTokens,
deleteTestUser,
getDeviceToken,
} from '../db'
@ -66,7 +67,7 @@ describe('Device tokens API', () => {
after(async () => {
// clean up
await getRepository(UserDeviceToken).delete({ user: { id: user.id } })
await deleteTestDeviceTokens(user.id, { user: { id: user.id } })
})
context('when id in input is not null', () => {
@ -174,7 +175,7 @@ describe('Device tokens API', () => {
after(async () => {
// clean up
await getRepository(UserDeviceToken).delete({ token })
await deleteTestDeviceTokens(user.id, { token })
})
it('responds with status code 200 and returns all device tokens', async () => {