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