/***********************************************************
Copyright (C) 2025 VeriSign, Inc.

This library 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 library 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 library; if not, write to the Free Software
Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA

http://www.verisign.com/nds/naming/namestore/techdocs.html
***********************************************************/
package com.verisign.epp.interfaces.balance.v01;

import java.util.Random;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.verisign.epp.codec.balance.v01.EPPBalanceInfoResp;
import com.verisign.epp.codec.gen.EPPResponse;
import com.verisign.epp.codec.gen.EPPResult;
import com.verisign.epp.interfaces.EPPApplicationSingle;
import com.verisign.epp.interfaces.EPPCommandException;
import com.verisign.epp.interfaces.EPPDomain;
import com.verisign.epp.interfaces.EPPSession;
import com.verisign.epp.transport.EPPClientCon;
import com.verisign.epp.util.Environment;
import com.verisign.epp.util.TestThread;

// JUNIT Imports
import junit.framework.Assert;
import junit.framework.Test;
import junit.framework.TestCase;
import junit.framework.TestSuite;

/**
 * Domain test that will result with the insertion of a Balance Poll Message and 
 * to verify the consumption of the Balance Poll Message.
 */
public class EPPBalanceDomainTst extends TestCase {
  /**
   * Handle to the Singleton EPP Application instance
   * ({@code EPPApplicationSingle})
   */
  private static EPPApplicationSingle app = EPPApplicationSingle.getInstance();

  /** Name of configuration file to use for test (default = epp.config). */
  private static String configFileName = "epp.config";

  /** Logging category */
  private static Logger cat = LoggerFactory.getLogger(EPPBalanceDomainTst.class);

  /** EPP Domain associated with test */
  private EPPDomain domain = null;

  /** EPP Session associated with test */
  private EPPSession session = null;

  /** Connection to the EPP Server. */
  private EPPClientCon connection = null;

  /** Current test iteration */
  private int iteration = 0;

  /**
   * Random instance for the generation of unique objects (hosts, IP addresses,
   * etc.).
   */
  private Random rd = new Random(System.currentTimeMillis());

  /**
   * Allocates an {@code EPPLowBalanceDomainTst} with a logical name. The
   * constructor will initialize the base class {@code TestCase} with the
   * logical name.
   *
   * @param name
   *           Logical name of the test
   */
  public EPPBalanceDomainTst(String name) {
    super(name);
  }

  /**
   * JUNIT test method to implement the
   * {@code EPPLowBalanceDomainTst TestCase}. Each sub-test will be
   * invoked in order to satisfy testing the EPPDomain interface.
   */
  public void testDomain() {
    int numIterations = 1;

    String iterationsStr = System.getProperty("iterations");

    if (iterationsStr != null) {
      numIterations = Integer.parseInt(iterationsStr);
    }

    for (iteration = 0; (numIterations == 0) || (iteration < numIterations); iteration++) {
      printStart("Test Suite");

      domainCreate();

      domainPoll();

      printEnd("Test Suite");
    }
  }

  /**
   * Unit test of {@code EPPDomain.sendCreate}.
   */
  public void domainCreate() {
    printStart("domainCreate");

    EPPResponse response;

    try {
      System.out.println("\ndomainCreate: Domain Create");

      domain.setTransId("ABC-12345-XYZ");

      String myDomainName = this.makeDomainName();

      domain.addDomainName("test.com");

      domain.setAuthString("ClientX");

      // Execute update
      response = domain.sendCreate();

      // -- Output all of the response attributes
      System.out.println("domainCreate: Response = [" + response + "]\n\n");

      /**
       * Result Set
       */
      for (int i = 0; i < response.getResults().size(); i++) {
        EPPResult myResult = (EPPResult) response.getResults().elementAt(i);

        System.out.println("Result Code    : " + myResult.getCode());

        System.out.println("Result Message : " + myResult.getMessage());

        System.out.println("Result Lang    : " + myResult.getLang());

        if (myResult.isSuccess()) {
          System.out.println("Command Passed ");
        }

        else {
          System.out.println("Command Failed ");
        }

        if (myResult.getAllValues() != null) {
          for (int k = 0; k < myResult.getAllValues().size(); k++) {
            System.out.println("Result Values  : " + myResult.getAllValues().elementAt(k));
          }
        }
      }
    }

    catch (EPPCommandException e) {
      handleException(e);
    }

    printEnd("domainCreate");
  }

  /**
   * Unit test of {@code EPPDomain.sendPoll}.
   */
  public void domainPoll() {
    printStart("domainPoll");

    EPPResponse response = null;

    try {
      System.out.println("\ndomainPoll: Domain Poll");

      session.setTransId("AB-12345");

      session.setPollOp(EPPSession.OP_REQ);

      response = session.sendPoll();

      System.out.println("domainPoll: EPPBalanceInfoResponse = [" + response + "]\n\n");

      EPPResult theResult = response.getResult();

      // Got poll message?
      if (theResult.getCode() == EPPResult.SUCCESS_POLL_MSG) {
      	EPPBalanceInfoResp balancePollResponse = (EPPBalanceInfoResp) response;
        // Print the threshold amount
        System.out.println("domainPoll: CreditLimit(Threshold Amount) = ["
              + balancePollResponse.getCreditLimit() + "]\n\n");
        // Print the available credit
        System.out.println(
              "domainPoll: AvailableCredit = [" + balancePollResponse.getAvailableCredit() + "]\n\n");
        System.out.println("domainPoll: Acknowledging poll message");

        // ack poll message
        session.setPollOp(EPPSession.OP_ACK);

        session.setMsgID(response.getMsgQueue().getId());

        response = session.sendPoll();

        System.out.println("domainPoll: Response = [" + response + "]\n\n");
      } // No poll messages?
      else if (theResult.getCode() == EPPResult.SUCCESS_POLL_NO_MSGS) {
        System.out.println("domainPoll: No poll messages");
      }
      else { // Unexpected result?
        Assert.fail("domainPoll: Unexpected result for poll command: " + response);
      }
    }

    catch (EPPCommandException ex) {
      handleException(ex);
    }

    printEnd("domainPoll");
  }

  /**
   * Unit test of {@code EPPSession.initSession}. The session attribute is
   * initialized with the attributes defined in the EPP sample files.
   */
  private void initSession() {
    printStart("initSession");

    // Set attributes for initSession
    session.setTransId("ABC-12345-XYZ");

    session.setVersion("1.0");

    session.setLang("en");

    // Initialize the session
    try {
      session.initSession();
    }

    catch (EPPCommandException e) {
      EPPResponse response = session.getResponse();

      // Is a server specified error?
      if ((response != null) && (!response.isSuccess())) {
        Assert.fail("Server Error : " + response);
      }
      else {
        e.printStackTrace();

        Assert.fail("initSession Error : " + e);
      }
    }

    printEnd("initSession");
  }

  /**
   * Unit test of {@code EPPSession.endSession}. The session with the EPP
   * Server will be terminated.
   */
  private void endSession() {
    printStart("endSession");

    session.setTransId("ABC-12345-XYZ");

    // End the session
    try {
      session.endSession();
    }

    catch (EPPCommandException e) {
      EPPResponse response = session.getResponse();

      // Is a server specified error?
      if ((response != null) && (!response.isSuccess())) {
        Assert.fail("Server Error : " + response);
      }

      else // Other error
      {
        e.printStackTrace();

        Assert.fail("initSession Error : " + e);
      }
    }

    printEnd("endSession");
  }

  /**
   * JUNIT {@code setUp} method, which sets the default client Id to
   * "theRegistrar".
   */
  protected void setUp() {
    try {
      String theSessionClassName = System.getProperty("EPP.SessionClass");

      if (theSessionClassName != null) {
        try {
          Class theSessionClass = Class.forName(theSessionClassName);

          if (!EPPSession.class.isAssignableFrom((theSessionClass))) {
            Assert.fail(theSessionClassName + " is not a subclass of EPPSession");
          }

          session = (EPPSession) theSessionClass.getDeclaredConstructor().newInstance();
        }
        catch (Exception ex) {
          Assert.fail("Exception instantiating EPP.SessionClass value " + theSessionClassName + ": " + ex);
        }
      }
      else {
        session = new EPPSession();
      }

      session.setClientID(Environment.getProperty("EPP.Test.clientId", "ClientX"));
      session.setPassword(Environment.getProperty("EPP.Test.password", "foo-BAR2"));
    }

    catch (Exception e) {
      e.printStackTrace();

      Assert.fail("Error initializing the session: " + e);
    }

    initSession();

    // System.out.println("out init");
    domain = new EPPDomain(session);
  }

  /**
   * JUNIT {@code tearDown}, which currently does nothing.
   */
  protected void tearDown() {
    endSession();
  }

  /**
   * JUNIT {@code suite} static method, which returns the tests associated
   * with {@code EPPLowBalanceDomainTst}.
   *
   * @return DOCUMENT ME!
   */
  public static Test suite() {
    TestSuite suite = new TestSuite(EPPBalanceDomainTst.class);

    String theConfigFileName = System.getProperty("EPP.ConfigFile");
    if (theConfigFileName != null)
      configFileName = theConfigFileName;

    try {
      app.initialize(configFileName);
    }

    catch (EPPCommandException e) {
      e.printStackTrace();

      Assert.fail("Error initializing the EPP Application: " + e);
    }

    return suite;
  }

  /**
   * Handle an {@code EPPCommandException}, which can be either a server
   * generated error or a general exception. If the exception was caused by a
   * server error, "Server Error :&lt;Response XML&gt;" will be specified. If
   * the exception was caused by a general algorithm error, "General Error
   * :&lt;Exception Description&gt;" will be specified.
   *
   * @param aException
   *           Exception thrown during test
   */
  public void handleException(Exception aException) {
    EPPResponse response = session.getResponse();

    aException.printStackTrace();

    // Is a server specified error?
    if ((response != null) && (!response.isSuccess())) {
      Assert.fail("Server Error : " + response);
    }

    else {
      Assert.fail("General Error : " + aException);
    }
  }

  /**
   * Unit test main, which accepts the following system property options: <br>
   *
   * <ul>
   * <li>iterations Number of unit test iterations to run</li>
   * <li>validate Turn XML validation on ({@code true}) or off
   * ({@code false}). If validate is not specified, validation will be
   * off.</li>
   * </ul>
   *
   *
   * @param args
   *           DOCUMENT ME!
   */
  public static void main(String[] args) {
    // Override the default configuration file name?
    if (args.length > 0) {
      configFileName = args[0];
    }

    // Number of Threads
    int numThreads = 1;

    String threadsStr = System.getProperty("threads");

    if (threadsStr != null) {
      numThreads = Integer.parseInt(threadsStr);

      // Run test suite in multiple threads?
    }

    if (numThreads > 1) {
      // Spawn each thread passing in the Test Suite
      for (int i = 0; i < numThreads; i++) {
        TestThread thread = new TestThread("EPPSessionTst Thread " + i, EPPBalanceDomainTst.suite());

        thread.start();
      }
    }

    else { // Single threaded mode.
      junit.textui.TestRunner.run(EPPBalanceDomainTst.suite());
    }

    try {
      app.endApplication();
    }

    catch (EPPCommandException e) {
      e.printStackTrace();

      Assert.fail("Error ending the EPP Application: " + e);
    }
  }

  /**
   * This method tries to generate a unique String as Domain Name and Name
   * Server
   *
   * @return DOCUMENT ME!
   */
  public String makeDomainName() {
    long tm = System.currentTimeMillis();

    return new String(Thread.currentThread() + String.valueOf(tm + rd.nextInt(12)).substring(10) + ".com");
  }

  /**
   * Makes a unique IP address based off of the current time.
   *
   * @return Unique IP address {@code String}
   */
  public String makeIP() {
    long tm = System.currentTimeMillis();

    return new String(
          String.valueOf(tm + rd.nextInt(50)).substring(10) + "." + String.valueOf(tm + rd.nextInt(50)).substring(10)
                + "." + String.valueOf(tm + rd.nextInt(50)).substring(10) + "."
                + String.valueOf(tm + rd.nextInt(50)).substring(10));
  }

  /**
   * Makes a unique host name for a domain using the current time.
   *
   * @param newDomainName
   *           DOCUMENT ME!
   *
   * @return Unique host name {@code String}
   */
  public String makeHostName(String newDomainName) {
    long tm = System.currentTimeMillis();

    return new String(String.valueOf(tm + rd.nextInt(10)).substring(10) + "." + newDomainName);
  }

  /**
   * Makes a unique contact name using the current time.
   *
   * @return Unique contact name {@code String}
   */
  public String makeContactName() {
    long tm = System.currentTimeMillis();

    return new String("Con" + String.valueOf(tm + rd.nextInt(5)).substring(7));
  }

  /**
   * Print the start of a test with the {@code Thread} name if the current
   * thread is a {@code TestThread}.
   *
   * @param aTest
   *           name for the test
   */
  private void printStart(String aTest) {
    if (Thread.currentThread() instanceof TestThread) {
      System.out.print(Thread.currentThread().getName() + ", iteration " + iteration + ": ");

      cat.info(Thread.currentThread().getName() + ", iteration " + iteration + ": " + aTest + " Start");
    }

    System.out.println("Start of " + aTest);

    System.out.println("****************************************************************\n");
  }

  /**
   * Print the end of a test with the {@code Thread} name if the current
   * thread is a {@code TestThread}.
   *
   * @param aTest
   *           name for the test
   */
  private void printEnd(String aTest) {
    System.out.println("****************************************************************");

    if (Thread.currentThread() instanceof TestThread) {
      System.out.print(Thread.currentThread().getName() + ", iteration " + iteration + ": ");

      cat.info(Thread.currentThread().getName() + ", iteration " + iteration + ": " + aTest + " End");
    }

    System.out.println("End of " + aTest);

    System.out.println("\n");
  }

  /**
   * Print message
   *
   * @param aMsg
   *           message to print
   */
  private void printMsg(String aMsg) {
    if (Thread.currentThread() instanceof TestThread) {
      System.out.println(Thread.currentThread().getName() + ", iteration " + iteration + ": " + aMsg);

      cat.info(Thread.currentThread().getName() + ", iteration " + iteration + ": " + aMsg);
    }

    else {
      System.out.println(aMsg);

      cat.info(aMsg);
    }
  }

  // End EPPLowBalanceDomainTst.printMsg(String)

  /**
   * Print error message
   *
   * @param aMsg
   *           errpr message to print
   */
  private void printError(String aMsg) {
    if (Thread.currentThread() instanceof TestThread) {
      System.err.println(Thread.currentThread().getName() + ", iteration " + iteration + ": " + aMsg);

      cat.error(Thread.currentThread().getName() + ", iteration " + iteration + ": " + aMsg);
    }

    else {
      System.err.println(aMsg);

      cat.error(aMsg);
    }
  }

}