/*
* JBoss, Home of Professional Open Source.
* Copyright 2014 Red Hat, Inc., and individual contributors
* as indicated by the @author tags.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package io.undertow.server.handlers;
import io.undertow.testutils.category.UnitTest;
import org.junit.Test;
import org.junit.experimental.categories.Category;
import java.net.UnknownHostException;
import static io.undertow.attribute.ExchangeAttributes.requestHeader;
import static io.undertow.util.Headers.USER_AGENT;
import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertTrue;
/**
* Unit tests for peer security handler
*
* @author Andre Dietisheim
*/
@Category(UnitTest.class)
public class UserAgentAccessControlHandlerUnitTestCase {
private static final String PATTERN_IE_ALL = "Mozilla.+\\(compatible; MSIE .+";
private static final String PATTERN_IE_ALL_ABOVE_6 = "Mozilla.+\\(compatible; MSIE ([7-9]|1[0-9]).+";
private static final String PATTERN_FF_ALL = "Mozilla.+\\(.+ Gecko.* Firefox.+";
private static final String IE_6 = "Mozilla/4.0 (compatible; MSIE 6.1; Windows XP; .NET CLR 1.1.4322; .NET CLR 2.0.50727)";
private static final String IE_10 = "Mozilla/5.0 (compatible; MSIE 10.6; Windows NT 6.1; Trident/5.0; InfoPath.2; SLCC1; .NET CLR 3.0.4506.2152; .NET CLR 3.5.30729; .NET CLR 2.0.50727) 3gpp-gba UNTRUSTED/1.0";
private static final String FF_25 = "Mozilla/5.0 (Windows NT 6.1; Win64; x64; rv:25.0) Gecko/20100101 Firefox/25.0";
private static final String SAFARI = "Mozilla/5.0 (iPad; CPU OS 6_0 like Mac OS X) AppleWebKit/536.26 (KHTML, like Gecko) Version/6.0 Mobile/10A5355d Safari/8536.25";
@Test(expected = IllegalArgumentException.class)
public void testInvalidPattern() {
new AccessControlListHandler(requestHeader(USER_AGENT)).addAllow("[bogus");
}
@Test
public void testFalseDefault() {
assertFalse(new AccessControlListHandler(requestHeader(USER_AGENT)).setDefaultAllow(false).isAllowed("some useragent"));
}
@Test
public void testTrueDefault() throws UnknownHostException {
assertTrue(new AccessControlListHandler(requestHeader(USER_AGENT)).setDefaultAllow(true).isAllowed("some useragent"));
}
@Test
public void testNullUserAgent() {
assertTrue(new AccessControlListHandler(requestHeader(USER_AGENT)).setDefaultAllow(true).isAllowed(null));
}
@Test
public void testAllowAllButOne() throws UnknownHostException {
AccessControlListHandler handler = new AccessControlListHandler(requestHeader(USER_AGENT))
.setDefaultAllow(true)
.addDeny(PATTERN_IE_ALL);
assertFalse(handler.isAllowed(IE_6));
assertTrue(handler.isAllowed(FF_25));
}
@Test
public void testDenyAllButOne() throws UnknownHostException {
AccessControlListHandler handler = new AccessControlListHandler(requestHeader(USER_AGENT))
.setDefaultAllow(false)
.addAllow(PATTERN_FF_ALL);
assertTrue(handler.isAllowed(FF_25));
assertFalse(handler.isAllowed(IE_10));
}
@Test
public void testAllowIE6AndAboveAndAllOthers() throws UnknownHostException {
AccessControlListHandler handler = new AccessControlListHandler(requestHeader(USER_AGENT))
.setDefaultAllow(true)
.addAllow(PATTERN_IE_ALL_ABOVE_6)
.addDeny(PATTERN_IE_ALL);
assertFalse(handler.isAllowed(IE_6));
assertTrue(handler.isAllowed(IE_10));
assertTrue(handler.isAllowed(FF_25));
assertTrue(handler.isAllowed(SAFARI));
}
}