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