| /******************************************************************************* |
| * Copyright (c) 2010 Tasktop Technologies and others. |
| * All rights reserved. This program and the accompanying materials |
| * are made available under the terms of the Eclipse Public License v1.0 |
| * which accompanies this distribution, and is available at |
| * http://www.eclipse.org/legal/epl-v10.html |
| * |
| * Contributors: |
| * Tasktop Technologies - initial API and implementation |
| *******************************************************************************/ |
| |
| package org.eclipse.mylyn.commons.tests.net; |
| |
| import java.io.IOException; |
| import java.io.InputStream; |
| import java.io.InterruptedIOException; |
| import java.net.ServerSocket; |
| import java.net.SocketTimeoutException; |
| import java.util.concurrent.ThreadPoolExecutor; |
| |
| import junit.framework.TestCase; |
| |
| import org.eclipse.mylyn.internal.commons.net.CommonsNetPlugin; |
| import org.eclipse.mylyn.internal.commons.net.TimeoutInputStream; |
| |
| /** |
| * @author Steffen Pingel |
| */ |
| public class TimeoutInputStreamTest extends TestCase { |
| |
| volatile int value; |
| |
| volatile IOException e; |
| |
| InputStream stream = new InputStream() { |
| @Override |
| public int read() throws IOException { |
| if (e != null) { |
| throw e; |
| } |
| return value; |
| } |
| }; |
| |
| private ServerSocket server; |
| |
| @Override |
| protected void setUp() throws Exception { |
| assertEquals(0, ((ThreadPoolExecutor) CommonsNetPlugin.getExecutorService()).getActiveCount()); |
| server = new ServerSocket(); |
| new Thread() { |
| @Override |
| public void run() { |
| try { |
| server.accept(); |
| } catch (IOException e) { |
| // ignore |
| } |
| }; |
| }.start(); |
| } |
| |
| @Override |
| protected void tearDown() throws Exception { |
| server.close(); |
| } |
| |
| public void testClose() throws Exception { |
| TimeoutInputStream in = new TimeoutInputStream(stream, 1, 500, 500); |
| assertEquals(0, in.read()); |
| value = -1; |
| // clear buffer |
| in.read(); |
| assertEquals(-1, in.read()); |
| in.close(); |
| Thread.sleep(200); |
| assertEquals(0, ((ThreadPoolExecutor) CommonsNetPlugin.getExecutorService()).getActiveCount()); |
| } |
| |
| public void testCloseTimeout() throws Exception { |
| e = new SocketTimeoutException(); |
| TimeoutInputStream in = new TimeoutInputStream(stream, 1, 500, 500); |
| try { |
| in.read(); |
| fail("expected InterruptedIOException"); |
| } catch (InterruptedIOException e) { |
| // expected |
| } finally { |
| in.close(); |
| } |
| assertEquals(0, ((ThreadPoolExecutor) CommonsNetPlugin.getExecutorService()).getActiveCount()); |
| } |
| |
| } |