diff options
author | Bjørn Christian Seime <bjorncs@verizonmedia.com> | 2019-05-09 11:29:36 +0200 |
---|---|---|
committer | Bjørn Christian Seime <bjorncs@verizonmedia.com> | 2019-05-09 11:29:36 +0200 |
commit | 5b23cba3f2bc7cda5c6eced3267c75198da9904a (patch) | |
tree | e75a59f02e1ddfb0e7aa941a53bfc95f32905bbb /jdisc-security-filters/src/test | |
parent | c3667718a63a8703bf62833dcb92b7ad5422d0cc (diff) |
Move LocalhostFilter and NoopFilter to jdisc-security-filters
Diffstat (limited to 'jdisc-security-filters/src/test')
-rw-r--r-- | jdisc-security-filters/src/test/java/com/yahoo/jdisc/http/filter/security/misc/LocalhostFilterTest.java | 60 |
1 files changed, 60 insertions, 0 deletions
diff --git a/jdisc-security-filters/src/test/java/com/yahoo/jdisc/http/filter/security/misc/LocalhostFilterTest.java b/jdisc-security-filters/src/test/java/com/yahoo/jdisc/http/filter/security/misc/LocalhostFilterTest.java new file mode 100644 index 00000000000..39c3783caec --- /dev/null +++ b/jdisc-security-filters/src/test/java/com/yahoo/jdisc/http/filter/security/misc/LocalhostFilterTest.java @@ -0,0 +1,60 @@ +// Copyright 2018 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root. +package com.yahoo.jdisc.http.filter.security.misc; + +import com.yahoo.container.jdisc.RequestHandlerTestDriver; +import com.yahoo.jdisc.Response; +import com.yahoo.jdisc.http.filter.DiscFilterRequest; +import org.junit.Test; +import org.mockito.Mockito; + +import java.net.URI; + +import static org.junit.Assert.assertEquals; +import static org.junit.Assert.assertNull; +import static org.mockito.Mockito.when; + +/** + * @author mpolden + * @author bjorncs + */ +public class LocalhostFilterTest { + + @Test + public void filter() { + // Reject from non-loopback + assertUnauthorized(createRequest("1.2.3.4", null)); + + // Allow requests from loopback addresses + assertSuccess(createRequest("127.0.0.1", null)); + assertSuccess(createRequest("127.127.0.1", null)); + assertSuccess(createRequest("0:0:0:0:0:0:0:1", null)); + + // Allow requests originating from self + assertSuccess(createRequest("1.3.3.7", "1.3.3.7")); + } + + private static DiscFilterRequest createRequest(String remoteAddr, String localAddr) { + DiscFilterRequest request = Mockito.mock(DiscFilterRequest.class); + when(request.getRemoteAddr()).thenReturn(remoteAddr); + when(request.getLocalAddr()).thenReturn(localAddr); + when(request.getMethod()).thenReturn("GET"); + when(request.getUri()).thenReturn(URI.create("http://localhost:8080/")); + return request; + } + + private static void assertUnauthorized(DiscFilterRequest request) { + LocalhostFilter filter = new LocalhostFilter(); + RequestHandlerTestDriver.MockResponseHandler handler = new RequestHandlerTestDriver.MockResponseHandler(); + filter.filter(request, handler); + assertEquals(Response.Status.UNAUTHORIZED, handler.getStatus()); + } + + + private static void assertSuccess(DiscFilterRequest request) { + LocalhostFilter filter = new LocalhostFilter(); + RequestHandlerTestDriver.MockResponseHandler handler = new RequestHandlerTestDriver.MockResponseHandler(); + filter.filter(request, handler); + assertNull(handler.getResponse()); + } + +} |