@Test
public void shouldStripHeaderWithKey() {
request.getHeaders().put("vanish", "begone");
StripHeaderBean shb = new StripHeaderBean();
shb.setPattern("vanish");
shb.setStripType(StripType.KEY);
shb.setWith(With.STRING);
config.getStripHeaders().add(shb);
policy.apply(request, mContext, config, mRequestChain);
assertFalse(request.getHeaders().containsKey("vanish"));
}
@Test
public void shouldStripCaseInsensitively() {
request.getHeaders().put("lord", "Vetinari");
StripHeaderBean shb = new StripHeaderBean();
shb.setPattern("VETINAR\\w+");
shb.setStripType(StripType.VALUE);
shb.setWith(With.REGEX);
config.getStripHeaders().add(shb);
policy.apply(request, mContext, config, mRequestChain);
assertFalse(request.getHeaders().containsKey("vetinari"));
assertTrue(request.getHeaders().isEmpty());
}
@Test
public void shouldStripHeaderWithRegexKey() {
request.getHeaders().put("sybil", "ramkin");
StripHeaderBean shb = new StripHeaderBean();
shb.setPattern("sy.*l");
shb.setStripType(StripType.KEY);
shb.setWith(With.REGEX);
config.getStripHeaders().add(shb);
policy.apply(request, mContext, config, mRequestChain);
assertFalse(request.getHeaders().containsKey("sybil"));
assertTrue(request.getHeaders().isEmpty());
}
@Test
public void shouldStripHeaderWithValue() {
request.getHeaders().put("lu", "tze");
request.getHeaders().put("lu", "tze");
StripHeaderBean shb = new StripHeaderBean();
shb.setPattern("tze");
shb.setStripType(StripType.VALUE);
shb.setWith(With.STRING);
config.getStripHeaders().add(shb);
policy.apply(request, mContext, config, mRequestChain);
policy.apply(response, mContext, config, mResponseChain);
assertFalse(request.getHeaders().containsKey("lu"));
assertFalse(response.getHeaders().containsKey("lu"));
}