| /******************************************************************************* |
| * Copyright (c) 2010 Tasktop Technologies and others. |
| * |
| * This program and the accompanying materials are made available under the |
| * terms of the Eclipse Public License v. 2.0 which is available at |
| * https://www.eclipse.org/legal/epl-2.0 |
| * |
| * SPDX-License-Identifier: EPL-2.0 |
| * |
| * 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(); |
| } |
| |
| // wait 30 seconds for executor to complete |
| long startTime = System.currentTimeMillis(); |
| while (System.currentTimeMillis() - startTime < 30 * 1000 |
| && ((ThreadPoolExecutor) CommonsNetPlugin.getExecutorService()).getActiveCount() > 0) { |
| Thread.sleep(500); |
| } |
| assertEquals(0, ((ThreadPoolExecutor) CommonsNetPlugin.getExecutorService()).getActiveCount()); |
| } |
| |
| } |