Files
omnivore/packages/api/test/resolvers/api_key.test.ts
Hongbo Wu d1c2721b95 Add expiration date to api key generation (#495)
* Add exp to jwt token

* Add expiredAt in generate api key input

* Check exp in claim

* Add tests
2022-04-28 09:09:40 +08:00

104 lines
2.5 KiB
TypeScript

import { User } from '../../src/entity/user'
import { createTestUser, deleteTestUser } from '../db'
import { graphqlRequest, request } from '../util'
import { expect } from 'chai'
import supertest from 'supertest'
const testAPIKey = (apiKey: string): supertest.Test => {
const query = `
query {
articles(first: 1) {
... on ArticlesSuccess {
edges {
cursor
}
}
... on ArticlesError {
errorCodes
}
}
}
`
return graphqlRequest(query, apiKey)
}
describe('generate api key', () => {
const username = 'fake_user'
let authToken: string
let user: User
let query: string
let expiredAt: string
before(async () => {
// create test user and login
user = await createTestUser(username)
const res = await request
.post('/local/debug/fake-user-login')
.send({ fakeEmail: user.email })
authToken = res.body.authToken
})
after(async () => {
// clean up
await deleteTestUser(username)
})
beforeEach(() => {
query = `
mutation {
generateApiKey(input: {
expiredAt: "${expiredAt}"
}) {
... on GenerateApiKeySuccess {
apiKey
}
... on GenerateApiKeyError {
errorCodes
}
}
}
`
})
context('when no expiredAt is specified', () => {
before(() => {
expiredAt = ''
})
it('should generate an api key with no expiration date', async () => {
const response = await graphqlRequest(query, authToken)
expect(response.body.data.generateApiKey.apiKey).to.be.a('string')
return testAPIKey(response.body.data.generateApiKey.apiKey).expect(200)
})
})
context('when api key is not expired', () => {
before(() => {
expiredAt = new Date(Date.now() + 1000 * 60 * 60 * 24).toISOString()
})
it('should generate an api key', async () => {
const response = await graphqlRequest(query, authToken)
expect(response.body.data.generateApiKey.apiKey).to.be.a('string')
return testAPIKey(response.body.data.generateApiKey.apiKey).expect(200)
})
})
context('when api key is expired', () => {
before(() => {
expiredAt = new Date(Date.now() - 1000 * 60 * 60 * 24).toISOString()
})
it('should generate an expired api key', async () => {
const response = await graphqlRequest(query, authToken)
expect(response.body.data.generateApiKey.apiKey).to.be.a('string')
return testAPIKey(response.body.data.generateApiKey.apiKey).expect(500)
})
})
})