/**
* Logback: the reliable, generic, fast and flexible logging framework.
* Copyright (C) 1999-2013, QOS.ch. All rights reserved.
*
* This program and the accompanying materials are dual-licensed under
* either the terms of the Eclipse Public License v1.0 as published by
* the Eclipse Foundation
*
* or (per the licensee's choosing)
*
* under the terms of the GNU Lesser General Public License version 2.1
* as published by the Free Software Foundation.
*/
package ch.qos.logback.classic.net.server;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertTrue;
import java.io.ObjectOutputStream;
import java.net.InetAddress;
import java.net.ServerSocket;
import java.net.Socket;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.TimeUnit;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import ch.qos.logback.classic.Level;
import ch.qos.logback.classic.Logger;
import ch.qos.logback.classic.LoggerContext;
import ch.qos.logback.classic.net.mock.MockAppender;
import ch.qos.logback.classic.spi.ILoggingEvent;
import ch.qos.logback.classic.spi.LoggingEvent;
import ch.qos.logback.classic.spi.LoggingEventVO;
import ch.qos.logback.core.net.server.ServerSocketUtil;
/**
* A functional test for {@link ServerSocketReceiver}.
* <p>
* In this test we create a SocketServer, connect to it over the local
* network interface, and validate that it receives messages and delivers
* them to its appender.
*/
public class ServerSocketReceiverFunctionalTest {
private static final int EVENT_COUNT = 10;
private static final int SHUTDOWN_DELAY = 10000;
private MockAppender appender;
private Logger logger;
private ServerSocket serverSocket;
private InstrumentedServerSocketReceiver receiver;
private LoggerContext lc;
@Before
public void setUp() throws Exception {
lc = new LoggerContext();
appender = new MockAppender();
appender.start();
logger = lc.getLogger(getClass());
logger.addAppender(appender);
serverSocket = ServerSocketUtil.createServerSocket();
receiver = new InstrumentedServerSocketReceiver(serverSocket);
receiver.setContext(lc);
receiver.start();
}
@After
public void tearDown() throws Exception {
receiver.stop();
ExecutorService executor = lc.getExecutorService();
executor.shutdownNow();
executor.awaitTermination(SHUTDOWN_DELAY, TimeUnit.MILLISECONDS);
assertTrue(executor.isTerminated());
}
@Test
public void testLogEventFromClient() throws Exception {
ILoggingEvent event = new LoggingEvent(logger.getName(), logger,
Level.DEBUG, "test message", null, new Object[0]);
Socket socket = new Socket(InetAddress.getLocalHost(),
serverSocket.getLocalPort());
try {
LoggingEventVO eventVO = LoggingEventVO.build(event);
ObjectOutputStream oos = new ObjectOutputStream(socket.getOutputStream());
for (int i = 0; i < EVENT_COUNT; i++) {
oos.writeObject(eventVO);
}
oos.writeObject(eventVO);
oos.flush();
}
finally {
socket.close();
}
ILoggingEvent rcvdEvent = appender.awaitAppend(SHUTDOWN_DELAY);
assertNotNull(rcvdEvent);
assertEquals(event.getLoggerName(), rcvdEvent.getLoggerName());
assertEquals(event.getLevel(), rcvdEvent.getLevel());
assertEquals(event.getMessage(), rcvdEvent.getMessage());
}
}