Home | History | Annotate | Download | only in command 
      1 /*
      2  * Copyright 2009 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 import java.net.InetAddress;
     22 
     23 /**
     24  * Tests for the EprtCommandHandler class
     25  *
     26  * @author Chris Mair
     27  * @version $Revision$ - $Date$
     28  */
     29 public final class EprtCommandHandlerTest extends AbstractCommandHandlerTestCase {
     30 
     31     private static final String[] PARAMETERS_INSUFFICIENT = EMPTY;
     32     private static final String[] PARAMETERS_IPV4 = {"|1|132.235.1.2|6275|"};
     33     private static final InetAddress HOST_IPV4 = inetAddress("132.235.1.2");
     34     private static final String[] PARAMETERS_IPV6 = {"|2|1080::8:800:200C:417A|6275|"};
     35     private static final InetAddress HOST_IPV6 = inetAddress("1080::8:800:200C:417A");
     36     private static final int PORT = 6275;
     37 
     38     private EprtCommandHandler commandHandler;
     39 
     40     public void testHandleCommand_IPv4() throws Exception {
     41         final Command COMMAND = new Command(CommandNames.EPRT, PARAMETERS_IPV4);
     42 
     43         session.setClientDataPort(PORT);
     44         session.setClientDataHost(HOST_IPV4);
     45         session.sendReply(ReplyCodes.EPRT_OK, replyTextFor(ReplyCodes.EPRT_OK));
     46         replay(session);
     47 
     48         commandHandler.handleCommand(COMMAND, session);
     49         verify(session);
     50 
     51         verifyNumberOfInvocations(commandHandler, 1);
     52         verifyTwoDataElements(commandHandler.getInvocation(0),
     53                 PortCommandHandler.HOST_KEY, HOST_IPV4,
     54                 PortCommandHandler.PORT_KEY, new Integer(PORT));
     55     }
     56 
     57     public void testHandleCommand_IPv6() throws Exception {
     58         final Command COMMAND = new Command(CommandNames.EPRT, PARAMETERS_IPV6);
     59 
     60         session.setClientDataPort(PORT);
     61         session.setClientDataHost(HOST_IPV6);
     62         session.sendReply(ReplyCodes.EPRT_OK, replyTextFor(ReplyCodes.EPRT_OK));
     63         replay(session);
     64 
     65         commandHandler.handleCommand(COMMAND, session);
     66         verify(session);
     67 
     68         verifyNumberOfInvocations(commandHandler, 1);
     69         verifyTwoDataElements(commandHandler.getInvocation(0),
     70                 PortCommandHandler.HOST_KEY, HOST_IPV6,
     71                 PortCommandHandler.PORT_KEY, new Integer(PORT));
     72     }
     73 
     74     public void testHandleCommand_MissingRequiredParameter() throws Exception {
     75         testHandleCommand_InvalidParameters(commandHandler, CommandNames.EPRT, PARAMETERS_INSUFFICIENT);
     76     }
     77 
     78     /**
     79      * Perform initialization before each test
     80      *
     81      * @see org.mockftpserver.core.command.AbstractCommandHandlerTestCase#setUp()
     82      */
     83     protected void setUp() throws Exception {
     84         super.setUp();
     85         commandHandler = new EprtCommandHandler();
     86         commandHandler.setReplyTextBundle(replyTextBundle);
     87     }
     88 
     89 }