185efb15529d45e32fea8de03c38a968c157c8262chrismair/* 285efb15529d45e32fea8de03c38a968c157c8262chrismair * Copyright 2007 the original author or authors. 385efb15529d45e32fea8de03c38a968c157c8262chrismair * 485efb15529d45e32fea8de03c38a968c157c8262chrismair * Licensed under the Apache License, Version 2.0 (the "License"); 585efb15529d45e32fea8de03c38a968c157c8262chrismair * you may not use this file except in compliance with the License. 685efb15529d45e32fea8de03c38a968c157c8262chrismair * You may obtain a copy of the License at 785efb15529d45e32fea8de03c38a968c157c8262chrismair * 885efb15529d45e32fea8de03c38a968c157c8262chrismair * http://www.apache.org/licenses/LICENSE-2.0 985efb15529d45e32fea8de03c38a968c157c8262chrismair * 1085efb15529d45e32fea8de03c38a968c157c8262chrismair * Unless required by applicable law or agreed to in writing, software 1185efb15529d45e32fea8de03c38a968c157c8262chrismair * distributed under the License is distributed on an "AS IS" BASIS, 1285efb15529d45e32fea8de03c38a968c157c8262chrismair * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 1385efb15529d45e32fea8de03c38a968c157c8262chrismair * See the License for the specific language governing permissions and 1485efb15529d45e32fea8de03c38a968c157c8262chrismair * limitations under the License. 1585efb15529d45e32fea8de03c38a968c157c8262chrismair */ 1685efb15529d45e32fea8de03c38a968c157c8262chrismairpackage org.mockftpserver.core.socket; 1785efb15529d45e32fea8de03c38a968c157c8262chrismair 1885efb15529d45e32fea8de03c38a968c157c8262chrismairimport java.io.IOException; 1985efb15529d45e32fea8de03c38a968c157c8262chrismairimport java.net.ServerSocket; 2085efb15529d45e32fea8de03c38a968c157c8262chrismairimport java.net.Socket; 2185efb15529d45e32fea8de03c38a968c157c8262chrismairimport java.net.SocketTimeoutException; 2285efb15529d45e32fea8de03c38a968c157c8262chrismair 2385efb15529d45e32fea8de03c38a968c157c8262chrismair/** 2485efb15529d45e32fea8de03c38a968c157c8262chrismair * Test (fake) subclass of ServerSocket that performs no network access and allows setting the 2585efb15529d45e32fea8de03c38a968c157c8262chrismair * Socket returned by accept(), and the local port for the ServerSocket. 2685efb15529d45e32fea8de03c38a968c157c8262chrismair * 2785efb15529d45e32fea8de03c38a968c157c8262chrismair * @version $Revision$ - $Date$ 2885efb15529d45e32fea8de03c38a968c157c8262chrismair * 2985efb15529d45e32fea8de03c38a968c157c8262chrismair * @author Chris Mair 3085efb15529d45e32fea8de03c38a968c157c8262chrismair */ 3185efb15529d45e32fea8de03c38a968c157c8262chrismairpublic class StubServerSocket extends ServerSocket { 3285efb15529d45e32fea8de03c38a968c157c8262chrismair private int localPort; 3385efb15529d45e32fea8de03c38a968c157c8262chrismair private Socket socket; 3485efb15529d45e32fea8de03c38a968c157c8262chrismair 3585efb15529d45e32fea8de03c38a968c157c8262chrismair /** 3685efb15529d45e32fea8de03c38a968c157c8262chrismair * Construct a new instance with the specified local port. 3785efb15529d45e32fea8de03c38a968c157c8262chrismair * @param localPort - the local port to be returned from getLocalPort() 3885efb15529d45e32fea8de03c38a968c157c8262chrismair * @throws IOException 3985efb15529d45e32fea8de03c38a968c157c8262chrismair */ 4085efb15529d45e32fea8de03c38a968c157c8262chrismair public StubServerSocket(int localPort) throws IOException { 4185efb15529d45e32fea8de03c38a968c157c8262chrismair this(localPort, null); 4285efb15529d45e32fea8de03c38a968c157c8262chrismair } 4385efb15529d45e32fea8de03c38a968c157c8262chrismair 4485efb15529d45e32fea8de03c38a968c157c8262chrismair /** 4585efb15529d45e32fea8de03c38a968c157c8262chrismair * Construct a new instance with specified local port and accept() socket. 4685efb15529d45e32fea8de03c38a968c157c8262chrismair * @param localPort - the local port to be returned from getLocalPort() 4785efb15529d45e32fea8de03c38a968c157c8262chrismair * @param socket - the socket to be returned from accept(); if null, then accept() throws SocketTimeoutException. 4885efb15529d45e32fea8de03c38a968c157c8262chrismair * @throws IOException 4985efb15529d45e32fea8de03c38a968c157c8262chrismair */ 5085efb15529d45e32fea8de03c38a968c157c8262chrismair public StubServerSocket(int localPort, Socket socket) throws IOException { 5185efb15529d45e32fea8de03c38a968c157c8262chrismair super(0); 5285efb15529d45e32fea8de03c38a968c157c8262chrismair this.localPort = localPort; 5385efb15529d45e32fea8de03c38a968c157c8262chrismair this.socket = socket; 5485efb15529d45e32fea8de03c38a968c157c8262chrismair } 5585efb15529d45e32fea8de03c38a968c157c8262chrismair 5685efb15529d45e32fea8de03c38a968c157c8262chrismair /** 5785efb15529d45e32fea8de03c38a968c157c8262chrismair * Return the predefined local port 5885efb15529d45e32fea8de03c38a968c157c8262chrismair * @see java.net.ServerSocket#getLocalPort() 5985efb15529d45e32fea8de03c38a968c157c8262chrismair */ 6085efb15529d45e32fea8de03c38a968c157c8262chrismair public int getLocalPort() { 6185efb15529d45e32fea8de03c38a968c157c8262chrismair return localPort; 6285efb15529d45e32fea8de03c38a968c157c8262chrismair } 6385efb15529d45e32fea8de03c38a968c157c8262chrismair 6485efb15529d45e32fea8de03c38a968c157c8262chrismair /** 6585efb15529d45e32fea8de03c38a968c157c8262chrismair * If a socket was specified on the constructor, then return that; otherwise, throw a SocketTimeoutException. 6685efb15529d45e32fea8de03c38a968c157c8262chrismair * @see java.net.ServerSocket#accept() 6785efb15529d45e32fea8de03c38a968c157c8262chrismair */ 6885efb15529d45e32fea8de03c38a968c157c8262chrismair public Socket accept() throws IOException { 6985efb15529d45e32fea8de03c38a968c157c8262chrismair if (socket != null) { 7085efb15529d45e32fea8de03c38a968c157c8262chrismair return socket; 7185efb15529d45e32fea8de03c38a968c157c8262chrismair } 7285efb15529d45e32fea8de03c38a968c157c8262chrismair throw new SocketTimeoutException(); 7385efb15529d45e32fea8de03c38a968c157c8262chrismair } 7485efb15529d45e32fea8de03c38a968c157c8262chrismair} 75