Home | History | Annotate | Download | only in command 
      1 /*
      2  * Copyright 2007 the original author or authors.
      3  *
      4  * Licensed under the Apache License, Version 2.0 (the "License");
      5  * you may not use this file except in compliance with the License.
      6  * You may obtain a copy of the License at
      7  *
      8  *      http://www.apache.org/licenses/LICENSE-2.0
      9  *
     10  * Unless required by applicable law or agreed to in writing, software
     11  * distributed under the License is distributed on an "AS IS" BASIS,
     12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
     13  * See the License for the specific language governing permissions and
     14  * limitations under the License.
     15  */
     16 package org.mockftpserver.stub.command;
     17 
     18 import org.mockftpserver.core.command.*;
     19 import org.mockftpserver.core.command.AbstractCommandHandlerTestCase;
     20 
     21 /**
     22  * Tests for the UserCommandHandler class
     23  *
     24  * @version $Revision$ - $Date$
     25  *
     26  * @author Chris Mair
     27  */
     28 public final class UserCommandHandlerTest extends AbstractCommandHandlerTestCase {
     29 
     30     private static final String USERNAME1 = "user1";
     31     private static final String USERNAME2 = "user2";
     32 
     33     private UserCommandHandler commandHandler;
     34     private Command command1;
     35     private Command command2;
     36 
     37     /**
     38      * Test the handleCommand() method
     39      */
     40     public void testHandleCommand() throws Exception {
     41 
     42         session.sendReply(ReplyCodes.USER_NEED_PASSWORD_OK, replyTextFor(ReplyCodes.USER_NEED_PASSWORD_OK));
     43         session.sendReply(ReplyCodes.USER_LOGGED_IN_OK, replyTextFor(ReplyCodes.USER_LOGGED_IN_OK));
     44         replay(session);
     45 
     46         commandHandler.handleCommand(command1, session);
     47         commandHandler.setPasswordRequired(false);
     48         commandHandler.handleCommand(command2, session);
     49         verify(session);
     50 
     51         verifyNumberOfInvocations(commandHandler, 2);
     52         verifyOneDataElement(commandHandler.getInvocation(0), UserCommandHandler.USERNAME_KEY, USERNAME1);
     53         verifyOneDataElement(commandHandler.getInvocation(1), UserCommandHandler.USERNAME_KEY, USERNAME2);
     54     }
     55 
     56     /**
     57      * Test the handleCommand() method, when no username parameter has been specified
     58      */
     59     public void testHandleCommand_MissingUsernameParameter() throws Exception {
     60         testHandleCommand_InvalidParameters(commandHandler, CommandNames.USER, EMPTY);
     61     }
     62 
     63     /**
     64      * Test the setPasswordRequired() and isPasswordRequired() methods
     65      */
     66     public void testSetPasswordRequired() {
     67         assertTrue("initial state", commandHandler.isPasswordRequired());
     68         commandHandler.setPasswordRequired(false);
     69         assertFalse("after set false", commandHandler.isPasswordRequired());
     70     }
     71 
     72     /**
     73      * Perform initialization before each test
     74      *
     75      * @see org.mockftpserver.core.command.AbstractCommandHandlerTestCase#setUp()
     76      */
     77     protected void setUp() throws Exception {
     78         super.setUp();
     79         commandHandler = new UserCommandHandler();
     80         command1 = new Command(CommandNames.USER, array(USERNAME1));
     81         command2 = new Command(CommandNames.USER, array(USERNAME2));
     82         commandHandler.setReplyTextBundle(replyTextBundle);
     83     }
     84 
     85 }
     86