test: middleware/debug

lib/middleware/debug.js

@@ -9,16 +9,11 @@ module.exports = async (ctx, next) => {
         ctx.debug.ips[ip] = 0;
     }
     ctx.debug.ips[ip]++;
-
+    ctx.debug.request++;
-    if (ctx.request.path !== '/') {
-        ctx.debug.request++;
-    }
 
     await next();
 
-    if (ctx.request.path !== '/') {
+    if (ctx.response.get('X-Koa-Redis-Cache') || ctx.response.get('X-Koa-Memory-Cache')) {
-        if (ctx.response.get('X-Koa-Redis-Cache') || ctx.response.get('X-Koa-Memory-Cache')) {
+        ctx.debug.hitCache++;
-            ctx.debug.hitCache++;
-        }
     }
 };

test/middleware/debug.js

@@ -0,0 +1,52 @@
+const supertest = require('supertest');
+const { server } = require('../../lib/index');
+const request = supertest(server);
+const cheerio = require('cheerio');
+let gitHash;
+try {
+    gitHash = require('git-rev-sync').short();
+} catch (e) {
+    gitHash = (process.env.HEROKU_SLUG_COMMIT && process.env.HEROKU_SLUG_COMMIT.slice(0, 7)) || 'unknown';
+}
+
+afterAll(() => {
+    server.close();
+});
+
+describe('debug', () => {
+    it(`debug`, async () => {
+        await request.get('/test/1').set('X-Forwarded-For', '233.233.233.233');
+        await request.get('/test/1').set('X-Forwarded-For', '233.233.233.233');
+        await request.get('/test/1').set('X-Forwarded-For', '233.233.233.234');
+        await request.get('/test/2').set('X-Forwarded-For', '233.233.233.233');
+        await request.get('/test/2').set('X-Forwarded-For', '233.233.233.234');
+
+        const response = await request.get('/').set('X-Forwarded-For', '233.233.233.233');
+
+        const $ = cheerio.load(response.text);
+        $('.debug-item').each((index, item) => {
+            const key = $(item)
+                .find('.debug-key')
+                .html()
+                .trim();
+            const value = $(item)
+                .find('.debug-value')
+                .html()
+                .trim();
+            switch (key) {
+                case 'git hash:':
+                    expect(value).toBe(gitHash);
+                    break;
+                case '请求数:':
+                    expect(value).toBe('6');
+                    break;
+                case '热门路由:':
+                    expect(value).toBe(`3&nbsp;&nbsp;/test/1<br>2&nbsp;&nbsp;/test/2<br>1&nbsp;&nbsp;/<br>`);
+                    break;
+                case '热门IP:':
+                    expect(value).toBe(`4&nbsp;&nbsp;233.233.233.233<br>2&nbsp;&nbsp;233.233.233.234<br>`);
+                    break;
+            }
+        });
+    });
+});