opensource/RSSHub
1
0
Fork 0
mirror of https://github.com/DIYgod/RSSHub.git synced 2025-12-05 12:21:31 +08:00
Code Issues Packages Projects Releases Wiki Activity

test: whitelist in middleware/access-control

Browse Source
This commit is contained in:
DIYgod
2019-01-29 16:02:39 +08:00
parent 9f91279296
commit 23ba8aeca4
2 changed files with 32 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

40
test/middleware/access-control.js
View File

@@ -12,26 +12,46 @@ async function checkBlock(response) {
}
afterEach(() => {
process.env.BLACKLIST = undefined;
delete process.env.BLACKLIST;
delete process.env.WHITELIST;
jest.resetModules();
server.close();
});
describe('access-control', () => {
it(`blacklist`, async () => {
process.env.BLACKLIST = '/test/1,/test/2,233.233.233.233';
process.env.BLACKLIST = '/test/1,233.233.233.233';
server = require('../../lib/index').server;
const request = supertest(server);
const response1 = await request.get('/test/1');
checkBlock(response1);
const response11 = await request.get('/test/1');
checkBlock(response11);
const response2 = await request.get('/test/2');
checkBlock(response2);
const response12 = await request.get('/test/1').set('X-Forwarded-For', '233.233.233.233');
checkBlock(response12);
const response31 = await request.get('/test/3');
expect(response31.status).toBe(200);
const response21 = await request.get('/test/2');
expect(response21.status).toBe(200);
const response32 = await request.get('/test/3').set('X-Forwarded-For', '233.233.233.233');
checkBlock(response32);
const response22 = await request.get('/test/2').set('X-Forwarded-For', '233.233.233.233');
checkBlock(response22);
});
it(`whitelist`, async () => {
process.env.WHITELIST = '/test/1,233.233.233.233';
server = require('../../lib/index').server;
const request = supertest(server);
const response11 = await request.get('/test/1');
expect(response11.status).toBe(200);
const response12 = await request.get('/test/1').set('X-Forwarded-For', '233.233.233.233');
expect(response12.status).toBe(200);
const response21 = await request.get('/test/2');
checkBlock(response21);
const response22 = await request.get('/test/2').set('X-Forwarded-For', '233.233.233.233');
expect(response22.status).toBe(200);
});
});