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.core.socket; 17a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair 18a0ad464efff5f5e2d2523a3522cce6823ce05858chrismairimport java.io.IOException; 19a0ad464efff5f5e2d2523a3522cce6823ce05858chrismairimport java.net.ServerSocket; 20a0ad464efff5f5e2d2523a3522cce6823ce05858chrismairimport java.net.Socket; 21a0ad464efff5f5e2d2523a3522cce6823ce05858chrismairimport java.net.SocketTimeoutException; 22a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair 23a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair/** 24a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair * Test (fake) subclass of ServerSocket that performs no network access and allows setting the 25a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair * Socket returned by accept(), and the local port for the ServerSocket. 26a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair * 27a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair * @version $Revision$ - $Date$ 28a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair * 29a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair * @author Chris Mair 30a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair */ 31a0ad464efff5f5e2d2523a3522cce6823ce05858chrismairpublic class StubServerSocket extends ServerSocket { 32a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair private int localPort; 33a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair private Socket socket; 34a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair 35a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair /** 36a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair * Construct a new instance with the specified local port. 37a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair * @param localPort - the local port to be returned from getLocalPort() 38a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair * @throws IOException 39a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair */ 40a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair public StubServerSocket(int localPort) throws IOException { 41a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair this(localPort, null); 42a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair } 43a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair 44a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair /** 45a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair * Construct a new instance with specified local port and accept() socket. 46a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair * @param localPort - the local port to be returned from getLocalPort() 47a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair * @param socket - the socket to be returned from accept(); if null, then accept() throws SocketTimeoutException. 48a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair * @throws IOException 49a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair */ 50a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair public StubServerSocket(int localPort, Socket socket) throws IOException { 51a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair super(0); 52a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair this.localPort = localPort; 53a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair this.socket = socket; 54a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair } 55a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair 56a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair /** 57a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair * Return the predefined local port 58a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair * @see java.net.ServerSocket#getLocalPort() 59a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair */ 60a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair public int getLocalPort() { 61a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair return localPort; 62a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair } 63a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair 64a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair /** 65a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair * If a socket was specified on the constructor, then return that; otherwise, throw a SocketTimeoutException. 66a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair * @see java.net.ServerSocket#accept() 67a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair */ 68a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair public Socket accept() throws IOException { 69a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair if (socket != null) { 70a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair return socket; 71a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair } 72a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair throw new SocketTimeoutException(); 73a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair } 74a0ad464efff5f5e2d2523a3522cce6823ce05858chrismair} 75