/*
* JBoss, Home of Professional Open Source
* Copyright 2005, JBoss Inc., and individual contributors as indicated
* by the @authors tag. See the copyright.txt in the distribution for a
* full listing of individual contributors.
*
* This is free software; you can redistribute it and/or modify it
* under the terms of the GNU Lesser General Public License as
* published by the Free Software Foundation; either version 2.1 of
* the License, or (at your option) any later version.
*
* This software is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this software; if not, write to the Free
* Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
* 02110-1301 USA, or see the FSF site: http://www.fsf.org.
*/

package org.jboss.test.remoting.performance.raw.socket;

import org.apache.log4j.Level;
import org.jboss.jrunit.harness.BenchmarkTestDriver;

import java.io.IOException;

/**
 * @author <a href="mailto:tom.elrod@jboss.com">Tom Elrod</a>
 */
public class SocketPerformanceTestCase extends BenchmarkTestDriver //TestDriver
{
   protected int numberOfClients = 1;

   public static final String REMOTING_TRANSPORT = "remoting.transport";
   public static final String REMOTING_METADATA = "remoting.metadata";
   public static final String REMOTING_SERIALIZATION = "remoting.serialization";
   public static final String PAYLOAD_SIZE = "remoting.payload.size";
   public static final String NUMBER_OF_CLIENTS = "remoting.number_of_clients";
   public static final String NUMBER_OF_CALLS = "remoting.number_of_calls";
   public static final String JVM_MAX_HEAP_SIZE = "jvm.mx";
   public static final String RESULT_TIMEOUT = "jrunit.result_timeout";
   public static final String TEAR_DOWN_TIMEOUT = "jrunit.tear_down_timeout";
   public static final String RUN_TEST_TIMEOUT = "jrunit.run_test_timeout";
   public static final String REMOTING_HOST = "remoting.host";

   public void declareTestClasses()
   {
      //**************** LOGGING ***********************
      org.apache.log4j.BasicConfigurator.configure();
      org.apache.log4j.Category.getRoot().setLevel(Level.DEBUG);
      //org.apache.log4j.Category.getInstance("org.jboss.remoting").setLevel(Level.DEBUG);

      org.apache.log4j.SimpleLayout layout = new org.apache.log4j.SimpleLayout();

      try
      {
         org.apache.log4j.FileAppender fileAppender = new org.apache.log4j.FileAppender(layout, "debug_output.log");
         fileAppender.setThreshold(Level.DEBUG);
         fileAppender.setAppend(false);
         org.apache.log4j.Category.getRoot().addAppender(fileAppender);
      }
      catch(IOException e)
      {
         e.printStackTrace();
      }
      //*************** END LOGGING ***********************


      String numOfClients = System.getProperty(NUMBER_OF_CLIENTS);
      if(numOfClients != null && numOfClients.length() > 0)
      {
         try
         {
            numberOfClients = Integer.parseInt(numOfClients);
         }
         catch(NumberFormatException e)
         {
            e.printStackTrace();
         }
      }

      addTestClasses(getClientTestClass(),
                     numberOfClients,
                     SocketPerformanceServer.class.getName());
   }

   protected String getClientTestClass()
   {
      return SocketPerformanceClient.class.getName();
   }

   protected Level getTestHarnessLogLevel()
   {
      return Level.INFO;
      //return Level.DEBUG;
   }

   /**
    * The log level to run as for the test case.
    *
    * @return
    */
   protected Level getTestLogLevel()
   {
      return Level.INFO;
      //return Level.DEBUG;
   }

   /**
    * Returns the VM arguments to be passed to the vm when creating the client test cases (actually their harness).
    * The default value is null.
    *
    * @return
    */
   protected String getClientJVMArguments()
   {
      return getJVMArguments();
//      String args = "-Xdebug -Xnoagent -Djava.compiler=NONE -Xrunjdwp:transport=dt_socket,server=y,suspend=y,address=5000 ";
//      args = args + getJVMArguments();
//      return args;
   }

   /**
    * Returns the VM arguments to be passed to the vm when creating the server test cases (actually their harness).
    * The default value is null.
    *
    * @return
    */
   protected String getServerJVMArguments()
   {
      return getJVMArguments();
   }

   /**
    * Returns the VM arguments to be passed to the vm when creating the client and server test cases (actually their harness).
    * The default value is null.
    *
    * @return
    */
   private String getJVMArguments()
   {
      String vmArgs = "";

      String transport = System.getProperty(REMOTING_TRANSPORT);
      if(transport != null && transport.length() > 0)
      {
         vmArgs = "-D" + REMOTING_TRANSPORT + "=" + transport;
      }
      String host = System.getProperty(REMOTING_HOST);
      if(host != null && host.length() > 0)
      {
         vmArgs = "-D" + REMOTING_HOST + "=" + host;
      }
      String serialization = System.getProperty(REMOTING_SERIALIZATION);
      if(serialization != null && serialization.length() > 0)
      {
         vmArgs = vmArgs + " -D" + REMOTING_SERIALIZATION + "=" + serialization;
      }
      String metadata = System.getProperty(REMOTING_METADATA);
      if(metadata != null && metadata.length() > 0)
      {
         vmArgs = vmArgs + " -D" + REMOTING_METADATA + "=" + metadata;
      }
      String payloadSize = System.getProperty(PAYLOAD_SIZE);
      if(payloadSize != null && payloadSize.length() > 0)
      {
         vmArgs = vmArgs + " -D" + PAYLOAD_SIZE + "=" + payloadSize;
      }
      String numOfCalls = System.getProperty(NUMBER_OF_CALLS);
      if(numOfCalls != null && numOfCalls.length() > 0)
      {
         vmArgs = vmArgs + " -D" + NUMBER_OF_CALLS + "=" + numOfCalls;
      }
      String jvmMx = System.getProperty(JVM_MAX_HEAP_SIZE);
      if(jvmMx != null && jvmMx.length() > 0)
      {
         vmArgs = vmArgs + " -Xmx" + jvmMx + "m";
      }
      String numOfClients = System.getProperty(NUMBER_OF_CLIENTS);
      if(numOfClients != null && numOfClients.length() > 0)
      {
         try
         {
            numberOfClients = Integer.parseInt(numOfClients);
            vmArgs = vmArgs + " -D" + NUMBER_OF_CLIENTS + "=" + numberOfClients;
         }
         catch(NumberFormatException e)
         {
            e.printStackTrace();
         }
      }

      return vmArgs;
   }

   /**
    * How long to wait for test results to be returned from the client(s).  If goes longer than the
    * specified limit, will throw an exception and kill the running test cases.  Default value is
    * RESULTS_TIMEOUT.
    *
    * @return
    */
   protected long getResultsTimeout()
   {
      long defaultTimeout = 600000; // default to 10 minutes

      String timeout = System.getProperty(RESULT_TIMEOUT);
      if(timeout != null && timeout.length() > 0)
      {
         try
         {
            defaultTimeout = Long.parseLong(timeout);
         }
         catch(NumberFormatException e)
         {
            System.out.println("Can not use " + timeout + " as timeout value as is not a number");
         }
      }
      return defaultTimeout;
   }

   /**
    * How long for the server test case to wait for tear down message.  If exceeds timeout,
    * will throw exception.  The default value is TEARDOWN_TIMEOUT.
    *
    * @return
    */
   protected long getTearDownTimeout()
   {
      long defaultTimeout = 600000; // default to 10 minutes

      String timeout = System.getProperty(TEAR_DOWN_TIMEOUT);
      if(timeout != null && timeout.length() > 0)
      {
         try
         {
            defaultTimeout = Long.parseLong(timeout);
         }
         catch(NumberFormatException e)
         {
            System.out.println("Can not use " + timeout + " as timeout value as is not a number");
         }
      }
      return defaultTimeout;
   }

   /**
    * How long to allow each of the test cases to run their tests.  If exceeds this timeout
    * will throw exception and kill tests.  The default value is RUN_TEST_TIMEOUT.
    *
    * @return
    */
   protected long getRunTestTimeout()
   {
      long defaultTimeout = 600000; // default to 10 minutes

      String timeout = System.getProperty(RUN_TEST_TIMEOUT);
      if(timeout != null && timeout.length() > 0)
      {
         try
         {
            defaultTimeout = Long.parseLong(timeout);
         }
         catch(NumberFormatException e)
         {
            System.out.println("Can not use " + timeout + " as timeout value as is not a number");
         }
      }
      return defaultTimeout;
   }

}
