1 files changed, 133 insertions, 0 deletions
diff --git a/tags/2.1/src/test/java/org/mockftpserver/stub/command/RetrCommandHandlerTest.java b/tags/2.1/src/test/java/org/mockftpserver/stub/command/RetrCommandHandlerTest.java
new file mode 100644
index 0000000..9ec0a17
--- /dev/null
+++ b/tags/2.1/src/test/java/org/mockftpserver/stub/command/RetrCommandHandlerTest.java
@@ -0,0 +1,133 @@
+/*
+ * 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 org.apache.log4j.Logger;
+import org.easymock.MockControl;
+import org.mockftpserver.core.command.AbstractCommandHandlerTest;
+import org.mockftpserver.core.command.Command;
+import org.mockftpserver.core.command.CommandNames;
+import org.mockftpserver.core.command.ReplyCodes;
+import org.mockftpserver.core.util.AssertFailedException;
+
+/**
+ * Tests for the RetrCommandHandler class
+ *
+ * @author Chris Mair
+ * @version $Revision$ - $Date$
+ */
+public final class RetrCommandHandlerTest extends AbstractCommandHandlerTest {
+
+    private static final Logger LOG = Logger.getLogger(RetrCommandHandlerTest.class);
+
+    private RetrCommandHandler commandHandler;
+
+    /**
+     * Test the constructor that takes a String, passing in a null
+     */
+    public void testConstructor_String_Null() {
+        try {
+            new RetrCommandHandler((String) null);
+            fail("Expected AssertFailedException");
+        }
+        catch (AssertFailedException expected) {
+            LOG.info("Expected: " + expected);
+        }
+    }
+
+    /**
+     * Test the constructor that takes a byte[], passing in a null
+     */
+    public void testConstructor_ByteArray_Null() {
+        try {
+            new RetrCommandHandler((byte[]) null);
+            fail("Expected AssertFailedException");
+        }
+        catch (AssertFailedException expected) {
+            LOG.info("Expected: " + expected);
+        }
+    }
+
+    /**
+     * Test the setFileContents(String) method, passing in a null
+     */
+    public void testSetFileContents_String_Null() {
+        try {
+            commandHandler.setFileContents((String) null);
+            fail("Expected AssertFailedException");
+        }
+        catch (AssertFailedException expected) {
+            LOG.info("Expected: " + expected);
+        }
+    }
+
+    /**
+     * Test the setFileContents(byte[]) method, passing in a null
+     */
+    public void testSetFileContents_ByteArray_Null() {
+        try {
+            commandHandler.setFileContents((byte[]) null);
+            fail("Expected AssertFailedException");
+        }
+        catch (AssertFailedException expected) {
+            LOG.info("Expected: " + expected);
+        }
+    }
+
+    /**
+     * Test the handleCommand() method
+     *
+     * @throws Exception
+     */
+    public void testHandleCommand() throws Exception {
+        final String FILE_CONTENTS = "abc_123 456";
+        commandHandler.setFileContents(FILE_CONTENTS);
+
+        session.sendReply(ReplyCodes.TRANSFER_DATA_INITIAL_OK, replyTextFor(ReplyCodes.TRANSFER_DATA_INITIAL_OK));
+        session.openDataConnection();
+        session.sendData(FILE_CONTENTS.getBytes(), FILE_CONTENTS.length());
+        control(session).setMatcher(MockControl.ARRAY_MATCHER);
+        session.closeDataConnection();
+        session.sendReply(ReplyCodes.TRANSFER_DATA_FINAL_OK, replyTextFor(ReplyCodes.TRANSFER_DATA_FINAL_OK));
+        replay(session);
+
+        Command command = new Command(CommandNames.RETR, array(FILENAME1));
+        commandHandler.handleCommand(command, session);
+        verify(session);
+
+        verifyNumberOfInvocations(commandHandler, 1);
+        verifyOneDataElement(commandHandler.getInvocation(0), RetrCommandHandler.PATHNAME_KEY, FILENAME1);
+    }
+
+    /**
+     * Test the handleCommand() method, when no pathname parameter has been specified
+     */
+    public void testHandleCommand_MissingPathnameParameter() throws Exception {
+        testHandleCommand_InvalidParameters(commandHandler, CommandNames.RETR, EMPTY);
+    }
+
+    /**
+     * Perform initialization before each test
+     *
+     * @see org.mockftpserver.core.command.AbstractCommandHandlerTest#setUp()
+     */
+    protected void setUp() throws Exception {
+        super.setUp();
+        commandHandler = new RetrCommandHandler();
+        commandHandler.setReplyTextBundle(replyTextBundle);
+    }
+
+}