test: middleware/cache

test/middleware/cache.js

@@ -0,0 +1,54 @@
+const supertest = require('supertest');
+const Parser = require('rss-parser');
+const parser = new Parser();
+const wait = require('../../lib/utils/wait');
+let server;
+
+beforeAll(() => {
+    process.env.CACHE_EXPIRE = 1;
+});
+
+afterEach(() => {
+    delete process.env.CACHE_TYPE;
+    jest.resetModules();
+    server.close();
+});
+
+afterAll(() => {
+    delete process.env.CACHE_EXPIRE;
+});
+
+async function check(type) {
+    process.env.CACHE_TYPE = type;
+    server = require('../../lib/index').server;
+    const request = supertest(server);
+
+    const response1 = await request.get('/test/1');
+    const parsed1 = await parser.parseString(response1.text);
+
+    const response2 = await request.get('/test/1');
+    const parsed2 = await parser.parseString(response2.text);
+
+    delete parsed1.lastBuildDate;
+    delete parsed2.lastBuildDate;
+    expect(parsed2).toMatchObject(parsed1);
+
+    expect(response2.status).toBe(200);
+    expect(response2.headers[`x-koa-${type}-cache`]).toBe('true');
+    expect(response3.headers).not.toHaveProperty(`x-koa-${type === 'redis' ? 'memory' : 'redis'}-cache`);
+
+    await wait(1 * 1000 + 10);
+    const response3 = await request.get('/test/1');
+    expect(response3.headers).not.toHaveProperty('x-koa-redis-cache');
+    expect(response3.headers).not.toHaveProperty('x-koa-memory-cache');
+}
+
+describe('cache', () => {
+    it('memory', async () => {
+        check('memory');
+    });
+
+    it('redis', async () => {
+        check('redis');
+    });
+});