1ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair/* 2ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair * Copyright 2007 the original author or authors. 3ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair * 4ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair * Licensed under the Apache License, Version 2.0 (the "License"); 5ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair * you may not use this file except in compliance with the License. 6ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair * You may obtain a copy of the License at 7ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair * 8ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair * http://www.apache.org/licenses/LICENSE-2.0 9ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair * 10ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair * Unless required by applicable law or agreed to in writing, software 11ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair * distributed under the License is distributed on an "AS IS" BASIS, 12ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair * See the License for the specific language governing permissions and 14ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair * limitations under the License. 15ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair */ 16ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismairpackage org.mockftpserver.core.socket; 17ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair 18ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismairimport java.io.IOException; 19ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismairimport java.net.ServerSocket; 20ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismairimport java.net.Socket; 21ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismairimport java.net.SocketTimeoutException; 22ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair 23ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair/** 24ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair * Test (fake) subclass of ServerSocket that performs no network access and allows setting the 25ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair * Socket returned by accept(), and the local port for the ServerSocket. 26ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair * 27ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair * @version $Revision$ - $Date$ 28ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair * 29ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair * @author Chris Mair 30ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair */ 31ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismairpublic class StubServerSocket extends ServerSocket { 32ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair private int localPort; 33ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair private Socket socket; 34ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair 35ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair /** 36ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair * Construct a new instance with the specified local port. 37ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair * @param localPort - the local port to be returned from getLocalPort() 38ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair * @throws IOException 39ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair */ 40ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair public StubServerSocket(int localPort) throws IOException { 41ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair this(localPort, null); 42ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair } 43ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair 44ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair /** 45ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair * Construct a new instance with specified local port and accept() socket. 46ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair * @param localPort - the local port to be returned from getLocalPort() 47ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair * @param socket - the socket to be returned from accept(); if null, then accept() throws SocketTimeoutException. 48ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair * @throws IOException 49ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair */ 50ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair public StubServerSocket(int localPort, Socket socket) throws IOException { 51ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair super(0); 52ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair this.localPort = localPort; 53ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair this.socket = socket; 54ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair } 55ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair 56ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair /** 57ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair * Return the predefined local port 58ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair * @see java.net.ServerSocket#getLocalPort() 59ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair */ 60ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair public int getLocalPort() { 61ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair return localPort; 62ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair } 63ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair 64ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair /** 65ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair * If a socket was specified on the constructor, then return that; otherwise, throw a SocketTimeoutException. 66ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair * @see java.net.ServerSocket#accept() 67ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair */ 68ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair public Socket accept() throws IOException { 69ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair if (socket != null) { 70ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair return socket; 71ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair } 72ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair throw new SocketTimeoutException(); 73ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair } 74ad39334d4c363c6ada5863d0bb3184f5f4699d69chrismair} 75