1a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair/* 2a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair * Copyright 2007 the original author or authors. 3a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair * 4a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair * Licensed under the Apache License, Version 2.0 (the "License"); 5a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair * you may not use this file except in compliance with the License. 6a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair * You may obtain a copy of the License at 7a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair * 8a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair * http://www.apache.org/licenses/LICENSE-2.0 9a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair * 10a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair * Unless required by applicable law or agreed to in writing, software 11a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair * distributed under the License is distributed on an "AS IS" BASIS, 12a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair * See the License for the specific language governing permissions and 14a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair * limitations under the License. 15a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair */ 16a0ad464efff5f5e2d2523a3522cce6823ce05858chrismairpackage org.mockftpserver.stub.command; 17a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair 18a0ad464efff5f5e2d2523a3522cce6823ce05858chrismairimport org.mockftpserver.core.command.AbstractCommandHandlerTest; 19a0ad464efff5f5e2d2523a3522cce6823ce05858chrismairimport org.mockftpserver.core.command.Command; 20a0ad464efff5f5e2d2523a3522cce6823ce05858chrismairimport org.mockftpserver.core.command.CommandNames; 21a0ad464efff5f5e2d2523a3522cce6823ce05858chrismairimport org.mockftpserver.core.command.ReplyCodes; 22a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair 23a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair/** 24a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair * Tests for the StorCommandHandler class 25a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair * 26a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair * @author Chris Mair 27a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair * @version $Revision$ - $Date$ 28a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair */ 29a0ad464efff5f5e2d2523a3522cce6823ce05858chrismairpublic final class StorCommandHandlerTest extends AbstractCommandHandlerTest { 30a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair 31a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair private StorCommandHandler commandHandler; 32a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair 33a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair /** 34a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair * Perform initialization before each test 35a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair * 36a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair * @see org.mockftpserver.core.command.AbstractCommandHandlerTest#setUp() 37a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair */ 38a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair protected void setUp() throws Exception { 39a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair super.setUp(); 40a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair commandHandler = new StorCommandHandler(); 41a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair commandHandler.setReplyTextBundle(replyTextBundle); 42a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair } 43a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair 44a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair /** 45a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair * Test the handleCommand() method, as well as the getFileContents() and clearFileContents() methods 46a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair */ 47a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair public void testHandleCommand() throws Exception { 48a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair final String DATA = "ABC"; 49a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair 50a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair session.sendReply(ReplyCodes.TRANSFER_DATA_INITIAL_OK, replyTextFor(ReplyCodes.TRANSFER_DATA_INITIAL_OK)); 51a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair session.openDataConnection(); 52a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair session.readData(); 53a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair control(session).setReturnValue(DATA.getBytes()); 54a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair session.closeDataConnection(); 55a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair session.sendReply(ReplyCodes.TRANSFER_DATA_FINAL_OK, replyTextFor(ReplyCodes.TRANSFER_DATA_FINAL_OK)); 56a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair replay(session); 57a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair 58a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair Command command = new Command(CommandNames.STOR, array(FILENAME1)); 59a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair commandHandler.handleCommand(command, session); 60a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair verify(session); 61a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair 62a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair verifyNumberOfInvocations(commandHandler, 1); 63a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair verifyTwoDataElements(commandHandler.getInvocation(0), StorCommandHandler.PATHNAME_KEY, FILENAME1, 64a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair StorCommandHandler.FILE_CONTENTS_KEY, DATA.getBytes()); 65a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair } 66a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair 67a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair /** 68a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair * Test the handleCommand() method, when no pathname parameter has been specified 69a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair */ 70a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair public void testHandleCommand_MissingPathnameParameter() throws Exception { 71a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair testHandleCommand_InvalidParameters(commandHandler, CommandNames.STOR, EMPTY); 72a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair } 73a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair 74a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair} 75