/*
    * MIT License
    * Copyright (c) 2006-2025 JMockit developers
    * See LICENSE file for full license text.
    */
   package otherTests.multicast;
   
   import static java.util.Arrays.asList;
   
  import java.io.ByteArrayOutputStream;
  import java.net.Socket;
  import java.util.ArrayList;
  import java.util.List;
  
  import mockit.Expectations;
  import mockit.Injectable;
  import mockit.Mocked;
  import mockit.Tested;
  import mockit.VerificationsInOrder;
  import mockit.integration.junit5.JMockitExtension;
  
  import org.junit.jupiter.api.Test;
  import org.junit.jupiter.api.extension.ExtendWith;
  
  @ExtendWith(JMockitExtension.class)
  class MessageTest {
      // A general-purpose utility class that waits for background task completion.
      abstract static class TaskExecution implements Runnable {
          {
              List<Thread> tasksBefore = getActiveTasks();
  
              try {
                  // noinspection
                  // OverriddenMethodCallDuringObjectConstruction,OverridableMethodCallDuringObjectConstruction
                  run();
              } finally {
                  waitForCompletionOfNewTasks(tasksBefore);
              }
          }
  
          private List<Thread> getActiveTasks() {
              Thread[] tasks = new Thread[2 * Thread.activeCount()];
              Thread.enumerate(tasks);
              return new ArrayList<>(asList(tasks));
          }
  
          private void waitForCompletionOfNewTasks(List<Thread> tasksBefore) {
              List<Thread> tasksAfter = getActiveTasks();
              tasksAfter.removeAll(tasksBefore);
  
              for (Thread task : tasksAfter) {
                  try {
                      task.join();
                  } catch (InterruptedException ignore) {
                  }
              }
          }
      }
  
      @Injectable
      final String testContents = "hello there";
      @Injectable
      final Client[] testClients = { new Client("client1"), new Client("client2") };
      @Injectable
      StatusListener listener;
      @Tested
      Message message;// = new Message(testClients, testContents, listener);
  
      @Test
      void sendMessageToMultipleClients(@Mocked final Socket con) throws Exception {
          new Expectations() {
              {
                  con.getOutputStream();
                  result = new ByteArrayOutputStream();
                  con.getInputStream();
                  result = "reply1\n reply2\n";
              }
          };
  
          // Code under test:
          new TaskExecution() {
              @Override
              public void run() {
                  message.dispatch();
              }
          };
  
          // Verification that each client received the expected invocations:
          for (final Client client : testClients) {
              new VerificationsInOrder() {
                  {
                      listener.messageSent(client);
                      listener.messageDisplayedByClient(client);
                      listener.messageReadByClient(client);
                  }
              };
          }
      }
  }