/*
    * Logback: the reliable, generic, fast and flexible logging framework.
    * Copyright (C) 1999-2026, QOS.ch. All rights reserved.
    *
    * This program and the accompanying materials are dual-licensed under
    * either the terms of the Eclipse Public License v2.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.core.net;
  
  import static org.mockito.ArgumentMatchers.anyInt;
  import static org.mockito.Mockito.mock;
  import static org.mockito.Mockito.spy;
  import static org.mockito.Mockito.timeout;
  import static org.mockito.Mockito.verify;
  import static org.mockito.Mockito.when;
  
  import java.io.IOException;
  import java.io.ObjectInputStream;
  import java.io.OutputStream;
  import java.io.Serializable;
  import java.net.ServerSocket;
  import java.net.Socket;
  import java.util.concurrent.LinkedBlockingDeque;
  import java.util.concurrent.ThreadPoolExecutor;
  import java.util.concurrent.TimeUnit;
  
  import org.junit.jupiter.api.AfterEach;
  import org.junit.jupiter.api.Assertions;
  import org.junit.jupiter.api.BeforeEach;
  import org.junit.jupiter.api.Disabled;
  import org.junit.jupiter.api.Test;
  
  import ch.qos.logback.core.net.mock.MockContext;
  import ch.qos.logback.core.spi.PreSerializationTransformer;
  import ch.qos.logback.core.util.ExecutorServiceUtil;
  
  /**
   * Integration tests for {@link ch.qos.logback.core.net.AbstractSocketAppender}.
   *
   * @author Carl Harris
   * @author Sebastian Gröbler
   */
  public class AbstractSocketAppenderIntegrationTest {
  
      private static final int TIMEOUT = 2000;
  
      private ThreadPoolExecutor threadPoolExecutor = ExecutorServiceUtil.newThreadPoolExecutor();
      private MockContext mockContext = new MockContext(threadPoolExecutor);
      private AutoFlushingObjectWriter objectWriter;
      private ObjectWriterFactory objectWriterFactory = new SpyProducingObjectWriterFactory();
      private LinkedBlockingDeque<String> deque = spy(new LinkedBlockingDeque<String>(1));
      private QueueFactory queueFactory = mock(QueueFactory.class);
      private InstrumentedSocketAppender instrumentedAppender = new InstrumentedSocketAppender(queueFactory,
              objectWriterFactory);
  
      @BeforeEach
      public void setUp() throws Exception {
          when(queueFactory.<String>newLinkedBlockingDeque(anyInt())).thenReturn(deque);
          instrumentedAppender.setContext(mockContext);
      }
  
      @AfterEach
      public void tearDown() throws Exception {
          instrumentedAppender.stop();
          Assertions.assertFalse(instrumentedAppender.isStarted());
          threadPoolExecutor.shutdownNow();
          Assertions.assertTrue(threadPoolExecutor.awaitTermination(TIMEOUT, TimeUnit.MILLISECONDS));
      }
  
      @Disabled // JDK 16
      @Test
      public void dispatchesEvents() throws Exception {
  
          // given
          ServerSocket serverSocket = ServerSocketUtil.createServerSocket();
          instrumentedAppender.setRemoteHost(serverSocket.getInetAddress().getHostAddress());
          instrumentedAppender.setPort(serverSocket.getLocalPort());
          instrumentedAppender.start();
  
          Socket appenderSocket = serverSocket.accept();
          serverSocket.close();
  
          // when
          instrumentedAppender.append("some event");
  
          // wait for event to be taken from deque and being written into the stream
          verify(deque, timeout(TIMEOUT).atLeastOnce()).takeFirst();
          verify(objectWriter, timeout(TIMEOUT)).write("some event");
  
          // then
          ObjectInputStream ois = new ObjectInputStream(appenderSocket.getInputStream());
          Assertions.assertEquals( ois.readObject(), "some event");
          appenderSocket.close();
     }
 
     private static class InstrumentedSocketAppender extends AbstractSocketAppender<String> {
 
         public InstrumentedSocketAppender(QueueFactory queueFactory, ObjectWriterFactory objectWriterFactory) {
             super(queueFactory, objectWriterFactory);
         }
 
         @Override
         protected void postProcessEvent(String event) {
         }
 
         @Override
         protected PreSerializationTransformer<String> getPST() {
             return new PreSerializationTransformer<String>() {
                 public Serializable transform(String event) {
                     return event;
                 }
             };
         }
     }
 
     private class SpyProducingObjectWriterFactory extends ObjectWriterFactory {
 
         @Override
         public AutoFlushingObjectWriter newAutoFlushingObjectWriter(OutputStream outputStream) throws IOException {
             objectWriter = spy(super.newAutoFlushingObjectWriter(outputStream));
             return objectWriter;
         }
     }
 }