UserAgentAccessControlHandlerUnitTestCase.java

/*
 * 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));
    }

}