xref: /external/mockftpserver/tags/2.0-rc1/src/test/java/org/mockftpserver/stub/command/_AbstractStubDataCommandHandlerTest.java

/*
 * Copyright 2007 the original author or authors.
 *
 * 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 org.mockftpserver.stub.command;

import java.util.ListResourceBundle;
import java.util.ResourceBundle;

import org.apache.log4j.Logger;
import org.mockftpserver.core.command.Command;
import org.mockftpserver.core.command.InvocationRecord;
import org.mockftpserver.core.session.Session;
import org.mockftpserver.core.util.AssertFailedException;
import org.mockftpserver.stub.command.AbstractStubDataCommandHandler;
import org.mockftpserver.test.AbstractTest;

/**
 * Tests for AbstractStubDataCommandHandler. The class name is prefixed with an underscore
 * so that it is not filtered out by Maven's Surefire test plugin.
 *
 * @version $Revision$ - $Date$
 *
 * @author Chris Mair
 */
public final class _AbstractStubDataCommandHandlerTest extends AbstractTest {

    private static final Logger LOG = Logger.getLogger(_AbstractStubDataCommandHandlerTest.class);
    private static final Command COMMAND = new Command("command", EMPTY);
    private static final InvocationRecord INVOCATION_RECORD = new InvocationRecord(COMMAND, DEFAULT_HOST);

    private static final String REPLY_TEXT150 = "reply 150 ... abcdef";
    private static final String REPLY_TEXT226 = "reply 226 ... abcdef";
    private static final String REPLY_TEXT222 = "reply 222 ... abcdef";
    private static final String REPLY_TEXT333 = "reply 333 ... abcdef";
    private static final String REPLY_TEXT444 = "reply 444 ... abcdef";

    private Session session;
    private ResourceBundle replyTextBundle;
    private AbstractStubDataCommandHandler commandHandler;

    /**
     * Test the handleCommand() method
     */
    public void testHandleCommand() throws Exception {

        session.sendReply(150, REPLY_TEXT150);
        session.openDataConnection();
        session.sendReply(222, REPLY_TEXT222);
        session.sendReply(333, REPLY_TEXT333);
        session.sendReply(444, REPLY_TEXT444);
        session.closeDataConnection();
        session.sendReply(226, REPLY_TEXT226);
        replay(session);

        // Define CommandHandler test subclass
        commandHandler = new AbstractStubDataCommandHandler() {
            protected void beforeProcessData(Command c, Session s, InvocationRecord ir) {
                verifyParameters(c, s, ir);
                // Send unique reply code so that we can verify proper method invocation and ordering
                session.sendReply(222, REPLY_TEXT222);
            }

            protected void processData(Command c, Session s, InvocationRecord ir) {
                verifyParameters(c, s, ir);
                // Send unique reply code so that we can verify proper method invocation and ordering
                session.sendReply(333, REPLY_TEXT333);
            }

            protected void afterProcessData(Command c, Session s, InvocationRecord ir) {
                verifyParameters(c, s, ir);
                // Send unique reply code so that we can verify proper method invocation and ordering
                session.sendReply(444, REPLY_TEXT444);
            }

            private void verifyParameters(Command c, Session s, InvocationRecord ir) {
                assertSame("command", COMMAND, c);
                assertSame("session", session, s);
                assertSame("invocationRecord", INVOCATION_RECORD, ir);
            }
        };

        commandHandler.setReplyTextBundle(replyTextBundle);
        commandHandler.handleCommand(COMMAND, session, INVOCATION_RECORD);

        verify(session);
    }

    /**
     * Test the handleCommand() method, overriding the initial reply code and text
     */
    public void testHandleCommand_OverrideInitialReplyCodeAndText() throws Exception {

        final int OVERRIDE_REPLY_CODE = 333;
        final String OVERRIDE_REPLY_TEXT = "reply text";

        session.sendReply(OVERRIDE_REPLY_CODE, OVERRIDE_REPLY_TEXT);
        session.openDataConnection();
        session.closeDataConnection();
        session.sendReply(226, REPLY_TEXT226);
        replay(session);

        commandHandler.setPreliminaryReplyCode(OVERRIDE_REPLY_CODE);
        commandHandler.setPreliminaryReplyText(OVERRIDE_REPLY_TEXT);
        commandHandler.setReplyTextBundle(replyTextBundle);
        commandHandler.handleCommand(COMMAND, session, INVOCATION_RECORD);

        verify(session);
    }

    /**
     * Test the setPreliminaryReplyCode() method, passing in an invalid value
     */
    public void testSetPreliminaryReplyCode_Invalid() {
        try {
            commandHandler.setPreliminaryReplyCode(0);
            fail("Expected AssertFailedException");
        }
        catch (AssertFailedException expected) {
            LOG.info("Expected: " + expected);
        }
    }

    /**
     * Test the setFinalReplyCode() method, passing in an invalid value
     */
    public void testSetFinalReplyCode_Invalid() {
        try {
            commandHandler.setFinalReplyCode(0);
            fail("Expected AssertFailedException");
        }
        catch (AssertFailedException expected) {
            LOG.info("Expected: " + expected);
        }
    }

    //-------------------------------------------------------------------------
    // Test setup
    //-------------------------------------------------------------------------

    /**
     * Perform initialization before each test
     *
     * @see org.mockftpserver.test.AbstractTest#setUp()
     */
    protected void setUp() throws Exception {
        super.setUp();
        session = (Session) createMock(Session.class);
        replyTextBundle = new ListResourceBundle() {
            protected Object[][] getContents() {
                return new Object[][] {
                        { Integer.toString(150), REPLY_TEXT150 },
                        { Integer.toString(222), REPLY_TEXT222 },
                        { Integer.toString(226), REPLY_TEXT226 },
                        { Integer.toString(333), REPLY_TEXT333 },
                        { Integer.toString(444), REPLY_TEXT444 },
                };
            }
        };
        commandHandler = new AbstractStubDataCommandHandler() {
            protected void processData(Command c, Session s, InvocationRecord ir) {
            }
        };
    }

}