@Test
public void testBasicUseCase() throws Exception {
URL configUrl = TestUtils.getResourceUrl(this.getClass(), "hostmap.txt");
HostMapper hm = EasyMock.createNiceMock(HostMapper.class);
EasyMock.expect(hm.resolveInboundHostName("test-inbound-host"))
.andReturn("test-inbound-rewritten-host")
.anyTimes();
HostMapperService hms = EasyMock.createNiceMock(HostMapperService.class);
GatewayServices gatewayServices = EasyMock.createNiceMock(GatewayServices.class);
EasyMock.expect(gatewayServices.getService(GatewayServices.HOST_MAPPING_SERVICE))
.andReturn(hms)
.anyTimes();
UrlRewriteEnvironment environment = EasyMock.createNiceMock(UrlRewriteEnvironment.class);
EasyMock.expect(environment.getAttribute(GatewayServices.GATEWAY_SERVICES_ATTRIBUTE))
.andReturn(gatewayServices)
.anyTimes();
EasyMock.expect(environment.resolve("cluster.name"))
.andReturn(Arrays.asList("test-cluster-name"))
.anyTimes();
EasyMock.expect(environment.getResource("/WEB-INF/hostmap.txt"))
.andReturn(configUrl)
.anyTimes();
Resolver resolver = EasyMock.createNiceMock(Resolver.class);
EasyMock.expect(resolver.resolve("host"))
.andReturn(Arrays.asList("test-inbound-host"))
.anyTimes();
EasyMock.replay(gatewayServices, hm, hms, environment, resolver);
UrlRewriteRulesDescriptor descriptor = UrlRewriteRulesDescriptorFactory.create();
UrlRewriteRuleDescriptor rule = descriptor.addRule("test-rule");
rule.pattern("{*}://{host}:{*}/{**}?{**}");
UrlRewriteActionRewriteDescriptorExt rewrite = rule.addStep("rewrite");
rewrite.template("{*}://{$hostmap(host)}:{*}/{**}?{**}");
UrlRewriteProcessor rewriter = new UrlRewriteProcessor();
rewriter.initialize(environment, descriptor);
Template input =
Parser.parseLiteral(
"test-scheme://test-inbound-host:42/test-path/test-file?test-name=test-value");
Template output = rewriter.rewrite(resolver, input, UrlRewriter.Direction.IN, null);
// System.out.println( output );
assertThat(output, notNullValue());
assertThat(output.getHost().getFirstValue().getPattern(), is("test-inbound-rewritten-host"));
}
