mirror of
https://github.com/alibaba/anyproxy.git
synced 2025-08-04 21:39:04 +00:00
update to 4.0
This commit is contained in:
91
test/spec_rule/rule_replace_response_header_spec.js
Normal file
91
test/spec_rule/rule_replace_response_header_spec.js
Normal file
@@ -0,0 +1,91 @@
|
||||
/*
|
||||
* test for rule replaceResponseHeader rule
|
||||
*
|
||||
*/
|
||||
const ProxyServerUtil = require('../util/ProxyServerUtil.js');
|
||||
const { proxyGet, generateUrl, directGet } = require('../util/HttpUtil.js');
|
||||
const Server = require('../server/server.js');
|
||||
const { printLog } = require('../util/CommonUtil.js');
|
||||
|
||||
const rule = require('./rule/rule_replace_response_header.js');
|
||||
|
||||
testWrapper('http');
|
||||
testWrapper('https');
|
||||
|
||||
function testWrapper(protocol) {
|
||||
describe('Rule replaceResponseHeader should be working in :' + protocol, () => {
|
||||
let proxyServer;
|
||||
let serverInstance;
|
||||
|
||||
beforeAll((done) => {
|
||||
jasmine.DEFAULT_TIMEOUT_INTERVAL = 50000;
|
||||
printLog('Start server for rule_replace_response_header_spec');
|
||||
|
||||
serverInstance = new Server();
|
||||
|
||||
proxyServer = ProxyServerUtil.proxyServerWithRule(rule);
|
||||
|
||||
setTimeout(() => {
|
||||
done();
|
||||
}, 2000);
|
||||
});
|
||||
|
||||
afterAll(() => {
|
||||
serverInstance && serverInstance.close();
|
||||
proxyServer && proxyServer.close();
|
||||
printLog('Close server for rule_replace_response_header_spec');
|
||||
});
|
||||
|
||||
it('Should replace the header in proxy if assertion is true', done => {
|
||||
const url = generateUrl(protocol, '/test/normal_request1');
|
||||
proxyGet(url)
|
||||
.then(proxyRes => {
|
||||
expect(proxyRes.statusCode).toEqual(200);
|
||||
expect(proxyRes.headers.replacedheaderkey).toEqual('replacedHeader_value_in_rule');
|
||||
expect(proxyRes.body).toEqual('body_normal_request1');
|
||||
|
||||
directGet(url)
|
||||
.then(directRes => {
|
||||
expect(directRes.statusCode).toEqual(200);
|
||||
expect(directRes.headers.replacedheaderkey).toBeUndefined();
|
||||
expect(directRes.body).toEqual(proxyRes.body);
|
||||
done();
|
||||
})
|
||||
.catch(error => {
|
||||
console.error('error happened in direct get: ', error);
|
||||
done.fail('error happened in direct get');
|
||||
});
|
||||
})
|
||||
.catch(error => {
|
||||
console.error('error happened in proxy get: ', error);
|
||||
done.fail('error happened in proxy get');
|
||||
});
|
||||
});
|
||||
|
||||
it('Should not replace the header in proxy if the assertion is false', done => {
|
||||
const url = generateUrl(protocol, '/test/normal_request2');
|
||||
proxyGet(url)
|
||||
.then(proxyRes => {
|
||||
expect(proxyRes.statusCode).toEqual(200);
|
||||
expect(proxyRes.headers.replacedheaderkey).toBeUndefined();
|
||||
expect(proxyRes.body).toEqual('body_normal_request2');
|
||||
|
||||
directGet(url)
|
||||
.then(directRes => {
|
||||
expect(directRes.statusCode).toEqual(200);
|
||||
expect(directRes.headers.replacedheaderkey).toBeUndefined();
|
||||
expect(directRes.body).toEqual(proxyRes.body);
|
||||
done();
|
||||
})
|
||||
.catch(error => {
|
||||
console.error('error happened in direct get: ', error);
|
||||
done.fail('error happened in direct get');
|
||||
});
|
||||
})
|
||||
.catch(error => {
|
||||
console.error('error happened in proxy get: ', error);
|
||||
done.fail('error happened in proxy get');
|
||||
});
|
||||
});
|
||||
});
|
||||
}
|
||||
Reference in New Issue
Block a user