@Test public void testValidateBadIP() { assertFalse(authorizer.isAuthorized(MockWebContext.create().setRemoteAddress(BAD_IP), null)); } }
@Test public void testValidateGoodIP() { assertTrue(authorizer.isAuthorized(MockWebContext.create().setRemoteAddress(GOOD_IP), null)); }
@Test public void testRetrieveIpOk() { final MockWebContext context = MockWebContext.create().setRemoteAddress(GOOD_IP); final TokenCredentials credentials = extractor.extract(context); assertEquals(GOOD_IP, credentials.getToken()); }
@Test public void testAuthentication() { final IpClient client = new IpClient(new SimpleTestTokenAuthenticator()); final MockWebContext context = MockWebContext.create(); context.setRemoteAddress(IP); final TokenCredentials credentials = client.getCredentials(context); final CommonProfile profile = client.getUserProfile(credentials, context); assertEquals(IP, profile.getId()); } }
@Test public void testRetrieveIpFromHeaderUsingConstructor() { final MockWebContext context = MockWebContext.create().addRequestHeader(HEADER_NAME, GOOD_IP).setRemoteAddress(LOCALHOST); // test for varargs final IpExtractor ipExtractor = new IpExtractor("fooBar", HEADER_NAME, "barFoo"); final TokenCredentials credentials = ipExtractor.extract(context); assertEquals(GOOD_IP, credentials.getToken()); // test for edge case of 1 header final IpExtractor ipExtractor2 = new IpExtractor(HEADER_NAME); final TokenCredentials credentials2 = ipExtractor2.extract(context); assertEquals(GOOD_IP, credentials2.getToken()); }
@Test public void testRetrieveIpFromHeaderWithProxyIpCheck() { final MockWebContext context = MockWebContext.create().addRequestHeader(HEADER_NAME, GOOD_IP).setRemoteAddress(LOCALHOST); final IpExtractor ipExtractor = new IpExtractor(); ipExtractor.setProxyIp(LOCALHOST); // test for varargs ipExtractor.setAlternateIpHeaders("fooBar", HEADER_NAME, "barFoo"); final TokenCredentials credentials = ipExtractor.extract(context); assertEquals(GOOD_IP, credentials.getToken()); // test for edge case of 1 header ipExtractor.setAlternateIpHeaders(HEADER_NAME); final TokenCredentials credentials2 = ipExtractor.extract(context); assertEquals(GOOD_IP, credentials2.getToken()); }