/**
    * Logback: the reliable, generic, fast and flexible logging framework.
    * Copyright (C) 1999-2015, 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.jupiter.api.Assertions.assertEquals;
  import static org.junit.jupiter.api.Assertions.assertNotNull;
  import static org.junit.jupiter.api.Assertions.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 ch.qos.logback.classic.util.LogbackMDCAdapter;
  import org.junit.jupiter.api.AfterEach;
  import org.junit.jupiter.api.BeforeEach;
  import org.junit.jupiter.api.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.test.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;
      LogbackMDCAdapter logbackMDCAdapter = new LogbackMDCAdapter();
      @BeforeEach
      public void setUp() throws Exception {
          lc = new LoggerContext();
          lc.setMDCAdapter(logbackMDCAdapter);
  
          appender = new MockAppender();
          appender.start();
  
          logger = lc.getLogger(getClass());
          logger.addAppender(appender);
  
          serverSocket = ServerSocketUtil.createServerSocket();
  
          receiver = new InstrumentedServerSocketReceiver(serverSocket);
  
          receiver.setContext(lc);
          receiver.start();
      }
  
      @AfterEach
      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());
     }
 
 }