mirror/advplyr.audiobookshelf
1
0
Fork 0
mirror of https://github.com/advplyr/audiobookshelf.git synced 2025-07-30 15:55:26 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Cache HTTP headers and status

Browse source
This commit is contained in:
mikiher 2023-11-25 23:49:56 +02:00
parent 288a32cc1e
commit 3ff41f2b43
2 changed files with 28 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

30
test/server/managers/ApiCacheManager.test.js
View file

@ -13,14 +13,14 @@ describe('ApiCacheManager', () => {
beforeEach(() => {
cache = { get: sinon.stub(), set: sinon.spy() }
req = { user: { username: 'testUser' }, url: '/test-url' }
res = { send: sinon.spy() }
res = { send: sinon.spy(), getHeaders: sinon.stub(), statusCode: 200, status: sinon.spy(), set: sinon.spy() }
next = sinon.spy()
})
describe('middleware', () => {
it('should send cached data if available', () => {
// Arrange
const cachedData = { data: 'cached data' }
const cachedData = { body: 'cached data', headers: { 'content-type': 'application/json' }, statusCode: 200 }
cache.get.returns(cachedData)
const key = JSON.stringify({ user: req.user.username, url: req.url })
manager = new ApiCacheManager(cache)
@ -31,8 +31,12 @@ describe('ApiCacheManager', () => {
// Assert
expect(cache.get.calledOnce).to.be.true
expect(cache.get.calledWith(key)).to.be.true
expect(res.set.calledOnce).to.be.true
expect(res.set.calledWith(cachedData.headers)).to.be.true
expect(res.status.calledOnce).to.be.true
expect(res.status.calledWith(cachedData.statusCode)).to.be.true
expect(res.send.calledOnce).to.be.true
expect(res.send.calledWith(cachedData)).to.be.true
expect(res.send.calledWith(cachedData.body)).to.be.true
expect(res.originalSend).to.be.undefined
expect(next.called).to.be.false
expect(cache.set.called).to.be.false
@ -41,13 +45,17 @@ describe('ApiCacheManager', () => {
it('should cache and send response if data is not cached', () => {
// Arrange
cache.get.returns(null)
const responseData = { data: 'response data' }
const headers = { 'content-type': 'application/json' }
res.getHeaders.returns(headers)
const body = 'response data'
const statusCode = 200
const responseData = { body, headers, statusCode }
const key = JSON.stringify({ user: req.user.username, url: req.url })
manager = new ApiCacheManager(cache)
// Act
manager.middleware(req, res, next)
res.send(responseData)
res.send(body)
// Assert
expect(cache.get.calledOnce).to.be.true
@ -56,13 +64,17 @@ describe('ApiCacheManager', () => {
expect(cache.set.calledOnce).to.be.true
expect(cache.set.calledWith(key, responseData)).to.be.true
expect(res.originalSend.calledOnce).to.be.true
expect(res.originalSend.calledWith(responseData)).to.be.true
expect(res.originalSend.calledWith(body)).to.be.true
})
it('should cache personalized response with 30 minutes TTL', () => {
// Arrange
cache.get.returns(null)
const responseData = { data: 'personalized data' }
const headers = { 'content-type': 'application/json' }
res.getHeaders.returns(headers)
const body = 'personalized data'
const statusCode = 200
const responseData = { body, headers, statusCode }
req.url = '/libraries/id/personalized'
const key = JSON.stringify({ user: req.user.username, url: req.url })
const ttlOptions = { ttl: 30 * 60 * 1000 }
@ -70,7 +82,7 @@ describe('ApiCacheManager', () => {
// Act
manager.middleware(req, res, next)
res.send(responseData)
res.send(body)
// Assert
expect(cache.get.calledOnce).to.be.true
@ -79,7 +91,7 @@ describe('ApiCacheManager', () => {
expect(cache.set.calledOnce).to.be.true
expect(cache.set.calledWith(key, responseData, ttlOptions)).to.be.true
expect(res.originalSend.calledOnce).to.be.true
expect(res.originalSend.calledWith(responseData)).to.be.true
expect(res.originalSend.calledWith(body)).to.be.true
})
})
})