org.jacoco.agent.rt.test/src/org/jacoco/agent/rt/internal/output/MockServerSocketTest.java

/*******************************************************************************
 * Copyright (c) 2009, 2018 Mountainminds GmbH & Co. KG and Contributors
 * 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:
 *    Marc R. Hoffmann - initial API and implementation
 *
 *******************************************************************************/
package org.jacoco.agent.rt.internal.output;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertTrue;

import java.net.InetAddress;
import java.net.ServerSocket;
import java.net.Socket;
import java.net.SocketException;
import java.util.concurrent.Callable;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.Future;

import org.junit.After;
import org.junit.Before;
import org.junit.Test;

/**
 * Unit tests for {@link MockServerSocket}.
 */
public class MockServerSocketTest extends ExecutorTestBase {

	private ServerSocket serverSocket;

	/**
	 * To verify that the tests reflect the behavior of real TCP sockets this
	 * flag can be set to <code>true</code>.
	 */
	private static final boolean REAL_SOCKETS = Boolean
			.getBoolean("MockServerSocketTest.realSockets");

	@Before
	@Override
	public void setup() throws Exception {
		super.setup();
		if (REAL_SOCKETS) {
			System.err.println("Using Real Sockets!");
			final InetAddress addr = InetAddress.getByName("localhost");
			serverSocket = new ServerSocket(16300, 1, addr);
		} else {
			serverSocket = new MockServerSocket();
		}
	}

	@After
	@Override
	public void teardown() throws Exception {
		serverSocket.close();
		super.teardown();
	}

	private Socket connect() throws Exception {
		if (REAL_SOCKETS) {
			final InetAddress addr = InetAddress.getByName("localhost");
			return new Socket(addr, 16300);
		} else {
			return ((MockServerSocket) serverSocket).connect();
		}

	}

	@Test
	public void testClose() throws Exception {
		assertFalse(serverSocket.isClosed());
		serverSocket.close();
		assertTrue(serverSocket.isClosed());
	}

	@Test(expected = SocketException.class)
	public void testCloseWhileAccept() throws Throwable {
		final Future<Socket> f = executor.submit(new Callable<Socket>() {
			public Socket call() throws Exception {
				return serverSocket.accept();
			}
		});
		assertBlocks(f);
		serverSocket.close();
		try {
			f.get();
		} catch (ExecutionException e) {
			throw e.getCause();
		}
	}

	@Test
	public void testAccept() throws Exception {
		final Future<Socket> f = executor.submit(new Callable<Socket>() {
			public Socket call() throws Exception {
				return serverSocket.accept();
			}
		});
		assertBlocks(f);
		connect().getOutputStream().write(123);
		final Socket socket = f.get();
		assertNotNull(socket);
		assertEquals(123, socket.getInputStream().read());
	}

	@Test(expected = SocketException.class)
	public void testAcceptOnClosedServerSocket() throws Exception {
		serverSocket.close();
		serverSocket.accept();
	}

	@Test
	public void testConnect() throws Exception {
		if (!REAL_SOCKETS) {
			final Future<Socket> f = executor.submit(new Callable<Socket>() {
				public Socket call() throws Exception {
					return ((MockServerSocket) serverSocket).connect();
				}
			});
			assertBlocks(f);
			serverSocket.accept().getOutputStream().write(42);
			final Socket socket = f.get();
			assertNotNull(socket);
			assertEquals(42, socket.getInputStream().read());
		}
	}

	@Test
	public void testWaitForAccept() throws Exception {
		if (!REAL_SOCKETS) {
			final Future<Void> f = executor.submit(new Callable<Void>() {
				public Void call() throws Exception {
					((MockServerSocket) serverSocket).waitForAccept();
					connect();
					return null;
				}
			});
			assertBlocks(f);
			serverSocket.accept();
			f.get();
		}
	}

}