WaffleAuthenticatorBaseTest.java

/**
 * Waffle (https://github.com/Waffle/waffle)
 *
 * Copyright (c) 2010-2016 Application Security, Inc.
 *
 * All rights reserved. This program and the accompanying materials are made available under the terms of the Eclipse
 * Public License v1.0 which accompanies this distribution, and is available at
 * https://www.eclipse.org/legal/epl-v10.html.
 *
 * Contributors: Application Security, Inc.
 */
package waffle.apache;

import java.io.IOException;

import javax.servlet.http.HttpServletResponse;

import org.apache.catalina.connector.Request;
import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;
import org.slf4j.LoggerFactory;

/**
 * Waffle Authenticator Base Tests.
 * 
 * @author dblock[at]dblock[dot]org
 */
public class WaffleAuthenticatorBaseTest {

    /** The waffle authenticator base. */
    private WaffleAuthenticatorBase waffleAuthenticatorBase;

    /**
     * Inits the.
     */
    @Before
    public void init() {
        this.waffleAuthenticatorBase = new WaffleAuthenticatorBase() {
            {
                this.log = LoggerFactory.getLogger(WaffleAuthenticatorBaseTest.class);
            }

            @Override
            public boolean authenticate(final Request request, final HttpServletResponse response) throws IOException {
                return false;
            }

            @Override
            protected boolean doAuthenticate(Request request, HttpServletResponse response) throws IOException {
                return false;
            }
        };
    }

    /**
     * Should_accept_both_protocols.
     *
     * @throws Exception
     *             the exception
     */
    @Test
    public void should_accept_both_protocols() throws Exception {
        this.waffleAuthenticatorBase.setProtocols("  NTLM , , Negotiate   ");

        Assert.assertEquals("Two protocols added", 2, this.waffleAuthenticatorBase.protocols.size());
        Assert.assertTrue("NTLM has been added", this.waffleAuthenticatorBase.protocols.contains("NTLM"));
        Assert.assertTrue("Negotiate has been added", this.waffleAuthenticatorBase.protocols.contains("Negotiate"));
    }

    /**
     * Should_accept_ negotiate_protocol.
     *
     * @throws Exception
     *             the exception
     */
    @Test
    public void should_accept_Negotiate_protocol() throws Exception {
        this.waffleAuthenticatorBase.setProtocols(" Negotiate  ");

        Assert.assertEquals("One protocol added", 1, this.waffleAuthenticatorBase.protocols.size());
        Assert.assertEquals("Negotiate", this.waffleAuthenticatorBase.protocols.iterator().next());
    }

    /**
     * Should_accept_ ntl m_protocol.
     *
     * @throws Exception
     *             the exception
     */
    @Test
    public void should_accept_NTLM_protocol() throws Exception {
        this.waffleAuthenticatorBase.setProtocols("  NTLM ");

        Assert.assertEquals("One protocol added", 1, this.waffleAuthenticatorBase.protocols.size());
        Assert.assertEquals("NTLM", this.waffleAuthenticatorBase.protocols.iterator().next());
    }

    /**
     * Should_refuse_other_protocol.
     *
     * @throws Exception
     *             the exception
     */
    @Test(expected = RuntimeException.class)
    public void should_refuse_other_protocol() throws Exception {
        this.waffleAuthenticatorBase.setProtocols("  NTLM , OTHER, Negotiate   ");
    }
}