/* 
 * Copyright (C) 2006 Registro.br. All rights reserved.
 * 
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are
 * met:
 * 1. Redistribution of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 * 
 * THIS SOFTWARE IS PROVIDED BY REGISTRO.BR ``AS IS AND ANY EXPRESS OR
 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
 * WARRANTIE OF FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO
 * EVENT SHALL REGISTRO.BR BE LIABLE FOR ANY DIRECT, INDIRECT,
 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
 * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS
 * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR
 * TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE
 * USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH
 * DAMAGE.
 */
/* $Id: ContactCreateTest.cpp 720 2006-05-06 21:40:46Z koji $ */
#include <memory>
#include <iostream>

#include "libepp_nicbr.H"

#include "ContactCreateTest.H"
#include "ContactCreate.H"
#include "FileUtil.H"
#include "IoException.H"
#include "XmlException.H"

using std::auto_ptr;

LIBEPP_NICBR_NS_USE

CPPUNIT_TEST_SUITE_REGISTRATION(ContactCreateTest);

ContactCreateTest::ContactCreateTest() {}

ContactCreateTest::~ContactCreateTest() {}

void ContactCreateTest::setUp() {}

void ContactCreateTest::tearDown() {}

void ContactCreateTest::set_xml_template_test()
{
  string to_be_parsed = "<command>"
       "<create>"
        "<contact:create"
          "xmlns:contact='urn:ietf:params:xml:ns:contact-1.0'"
          "xsi:schemaLocation='urn:ietf:params:xml:ns:contact-1.0"
          "contact-1.0.xsd'>"
           "<contact:id>$(id)$</contact:id>"
           "$(postal_info)$"
          "$(voice)$"
          "$(fax)$"
          "<contact:email>$(email)$</contact:email>"
          "$(auth_info)$"
          "$(disclose)$"
          "</contact:create>"
        "</create>"
    "$(clTRID)$"
       "</command>";

  ContactCreate contact_create;
  ContactCreateCmd *cmd = contact_create.get_command();
  CommonData common;

  common.set_id("cme254");
  PostalInfo postal_info;
  postal_info.set_type("int");
  postal_info.set_name("John Doe");
  postal_info.set_org("Example Inc.");
  postal_info.set_str1("123 Example Dr.");
  postal_info.set_str2("Suite 100");
  postal_info.set_str3("");
  postal_info.set_city( "Dulles");
  postal_info.set_sp("VA");
  postal_info.set_pc("20166-6503");
  postal_info.set_cc("US");
  common.insert_postal_info(postal_info);
  postal_info.set_type("loc");
  postal_info.set_name("John Doe");
  postal_info.set_org("Other Inc.");
  postal_info.set_str1("123 Street");
  postal_info.set_str2("7th floor");
  postal_info.set_str3("Suite 123");
  postal_info.set_city( "Miami");
  postal_info.set_sp("");
  postal_info.set_pc("");
  postal_info.set_cc("US");
  common.insert_postal_info(postal_info);
  
  CommonData::Phone voice;
  voice.ext = "1234";
  voice.number =  "+1.7035555555";
  CommonData::Phone fax;
  fax.ext = "3456";
  fax.number = "+1.7035555556";
  common.set_voice(voice);
  common.set_fax(fax);
  common.set_email("jdoe@example.com");
  AuthInfo authInfo;
  authInfo.set_roid("SH8013-REP");
  authInfo.set_pw("2fooBAR");
  cmd->set_authInfo(authInfo);
  
  CommonData::Disclose disclose;
  disclose.flag = false;
  disclose.name_int = true;
  disclose.voice = true;
  disclose.email = true;
  common.set_disclose(disclose);
  
  cmd->set_common_data(common);

  contact_create.get_command()->set_clTRID("ABC-12345");
  contact_create.set_xml_template(to_be_parsed);

  string expected = "<command>"
       "<create>"
        "<contact:create"
          "xmlns:contact='urn:ietf:params:xml:ns:contact-1.0'"
          "xsi:schemaLocation='urn:ietf:params:xml:ns:contact-1.0"
          "contact-1.0.xsd'>"
           "<contact:id>cme254</contact:id>"
           "<contact:postalInfo type='int'>"
              "<contact:name>John Doe</contact:name>"
              "<contact:org>Example Inc.</contact:org>"
              "<contact:addr>"
                "<contact:street>123 Example Dr.</contact:street>"
                "<contact:street>Suite 100</contact:street>"
                "<contact:city>Dulles</contact:city>"
                "<contact:sp>VA</contact:sp>"
                "<contact:pc>20166-6503</contact:pc>"
                "<contact:cc>US</contact:cc>"
              "</contact:addr>"
            "</contact:postalInfo>"
            "<contact:postalInfo type='loc'>"
               "<contact:name>John Doe</contact:name>"
               "<contact:org>Other Inc.</contact:org>"
               "<contact:addr>"
                 "<contact:street>123 Street</contact:street>"
                 "<contact:street>7th floor</contact:street>"
                 "<contact:street>Suite 123</contact:street>"
                 "<contact:city>Miami</contact:city>"
                 "<contact:cc>US</contact:cc>"
               "</contact:addr>"
             "</contact:postalInfo>"
             "<contact:voice x='1234'>+1.7035555555</contact:voice>"
             "<contact:fax x='3456'>+1.7035555556</contact:fax>"
             "<contact:email>jdoe@example.com</contact:email>"
            "<contact:authInfo>"
              "<contact:pw roid='SH8013-REP'>2fooBAR</contact:pw>"
            "</contact:authInfo>"
            "<contact:disclose flag='0'>"
            "<contact:name type =\"int\"/>"
            "<contact:voice/>"
            "<contact:email/>"
             "</contact:disclose>"
          "</contact:create>"
        "</create>"
        "<clTRID>ABC-12345</clTRID>"
     "</command>";

  CPPUNIT_ASSERT(expected == contact_create.get_xml());
  
  // tests reset method
  cmd->reset();
  CPPUNIT_ASSERT_EQUAL((string) "", cmd->get_common_data().get_id());
  CPPUNIT_ASSERT(cmd->get_common_data().get_postal_info().empty());
  CPPUNIT_ASSERT_EQUAL((string) "", cmd->get_common_data().get_voice().ext);
  CPPUNIT_ASSERT_EQUAL((string) "", cmd->get_common_data().get_voice().number);
  CPPUNIT_ASSERT_EQUAL((string) "", cmd->get_common_data().get_fax().ext);
  CPPUNIT_ASSERT_EQUAL((string) "", cmd->get_common_data().get_fax().number);
  CPPUNIT_ASSERT_EQUAL((string) "", cmd->get_common_data().get_email());
  CPPUNIT_ASSERT_EQUAL((string) "", cmd->get_authInfo().get_roid());
  CPPUNIT_ASSERT_EQUAL((string) "", cmd->get_authInfo().get_pw());
  CPPUNIT_ASSERT(cmd->get_common_data().get_disclose().flag == -1);
  CPPUNIT_ASSERT(cmd->get_common_data().get_disclose().name_int == false);
  CPPUNIT_ASSERT(cmd->get_common_data().get_disclose().name_loc == false);
  CPPUNIT_ASSERT(cmd->get_common_data().get_disclose().org_int == false);
  CPPUNIT_ASSERT(cmd->get_common_data().get_disclose().org_loc == false);
  CPPUNIT_ASSERT(cmd->get_common_data().get_disclose().addr_int == false);
  CPPUNIT_ASSERT(cmd->get_common_data().get_disclose().addr_loc == false);
  CPPUNIT_ASSERT(cmd->get_common_data().get_disclose().email == false);
  CPPUNIT_ASSERT(cmd->get_common_data().get_disclose().voice == false);
  CPPUNIT_ASSERT(cmd->get_common_data().get_disclose().fax == false);

  // parent attribute
  CPPUNIT_ASSERT(cmd->get_clTRID() == "");
}

void ContactCreateTest::command_test()
{
  ContactCreate contact_create;
  ContactCreateCmd *cmd = contact_create.get_command();
  CommonData common;

  common.set_id("cme254");
  PostalInfo postal_info;
  postal_info.set_type("int");
  postal_info.set_name("John Doe");
  postal_info.set_org("Example Inc.");
  postal_info.set_str1("123 Example Dr.");
  postal_info.set_str2("Suite 100");
  postal_info.set_str3("");
  postal_info.set_city("Dulles");
  postal_info.set_sp("VA");
  postal_info.set_pc("20166-6503");
  postal_info.set_cc("US");
  common.insert_postal_info(postal_info);
  postal_info.set_type("loc");
  postal_info.set_name("John Doe");
  postal_info.set_org("Other Inc.");
  postal_info.set_str1("123 Street");
  postal_info.set_str2("7th floor");
  postal_info.set_str3("Suite 123");
  postal_info.set_city("Miami");
  postal_info.set_sp("");
  postal_info.set_pc("");
  postal_info.set_cc("US");
  common.insert_postal_info(postal_info);
  
  CommonData::Phone voice;
  voice.ext = "1234";
  voice.number =  "+1.7035555555";
  CommonData::Phone fax;
  fax.ext = "3456";
  fax.number = "+1.7035555556";
  common.set_voice(voice);
  common.set_fax(fax);
  common.set_email("jdoe@example.com");
  AuthInfo authInfo;
  authInfo.set_roid("SH8013-REP");
  authInfo.set_pw("2fooBAR");
  cmd->set_authInfo(authInfo);
  
  CommonData::Disclose disclose;
  disclose.flag = false;
  disclose.name_int = true;
  disclose.voice = true;
  disclose.email = true;
  common.set_disclose(disclose);

  cmd->set_common_data(common);

  bool exception_caught = false;
  try {
    string xml_template = 
      FileUtil::read_file("../docs/templates/contact_create.xml");

    contact_create.get_command()->set_clTRID("ABC-12345");
    contact_create.set_xml_template(xml_template);

    DomParser parser;
    parser.enable_validation("../docs/schemas");
    parser.parse_command(contact_create.get_xml());
  } catch (const IoException &e) {
    exception_caught = true;
    printf("\nIO Exception: code [%d] message [%s]",
	   e.get_code(), e.get_msg().c_str());
  } catch (const XmlException &e) {
    exception_caught = true;
    printf("\nXml Exception: code [%d] message [%s] low level message [%s]\n",
	   e.get_code(), e.get_msg().c_str(), e.get_low_level_msg().c_str());
  }

  CPPUNIT_ASSERT(!exception_caught);
}

void ContactCreateTest::response_test()
{
  string expected = 
    "<?xml version=\"1.0\" encoding=\"UTF-8\" standalone=\"no\"?>"
    "<epp xmlns=\"urn:ietf:params:xml:ns:epp-1.0\" "
    "xmlns:xsi=\"http://www.w3.org/2001/XMLSchema-instance\" "
    "xsi:schemaLocation=\"urn:ietf:params:xml:ns:epp-1.0 "
    "epp-1.0.xsd\">"
    "<response>"
    "<result code=\"1000\">"
    "<msg>Command completed successfully</msg>"
    "</result>"
    "<resData>"
    "<contact:creData "
    "xmlns:contact=\"urn:ietf:params:xml:ns:contact-1.0\" "
    "xsi:schemaLocation=\"urn:ietf:params:xml:ns:contact-1.0 "
    "contact-1.0.xsd\">"
    "<contact:id>cem456</contact:id>"
    "<contact:crDate>2006-01-30T22:00:00.0Z</contact:crDate>"
    "</contact:creData>"
    "</resData>"
    "<trID>"
    "<clTRID>ABC-12345</clTRID>"
    "<svTRID>DEF-54321</svTRID>"
    "</trID>"
    "</response>"
    "</epp>";
  
  DomParser parser;
  parser.enable_validation("../docs/schemas");
  
  ContactCreate contact_create;
  CPPUNIT_ASSERT_NO_THROW(contact_create.set_response(expected, &parser));
  ContactCreateRsp *contact_create_rsp = contact_create.get_response();

  map <Response::ResultCode, Response::ResultInfo> results;
  map <Response::ResultCode, Response::ResultInfo>::const_iterator r_it;
  results = contact_create_rsp->get_result_list();
  r_it = results.begin();

  CPPUNIT_ASSERT(r_it != results.end());
  CPPUNIT_ASSERT_EQUAL(Response::OK, r_it->first);
 
  string result_code = "1000";
   
  string response = 
    "<?xml version=\"1.0\" encoding=\"UTF-8\" standalone=\"no\"?>"
    "<epp xmlns=\"urn:ietf:params:xml:ns:epp-1.0\" "
    "xmlns:xsi=\"http://www.w3.org/2001/XMLSchema-instance\" "
    "xsi:schemaLocation=\"urn:ietf:params:xml:ns:epp-1.0 "
    "epp-1.0.xsd\">"
    "<response>"
    "<result code=\"" + result_code + "\">"
    "<msg";
  
  string result_lang = contact_create_rsp->get_result_lang();
  if (result_lang != "en") {
    response += " lang='" + result_lang + "'";
  }
  
  response +=
    ">" + r_it->second.msg + "</msg>"
    "</result>"
    "<resData>"
    "<contact:creData "
    "xmlns:contact=\"urn:ietf:params:xml:ns:contact-1.0\" "
    "xsi:schemaLocation=\"urn:ietf:params:xml:ns:contact-1.0 "
    "contact-1.0.xsd\">"
    "<contact:id>" + contact_create_rsp->get_id() + "</contact:id>"
    "<contact:crDate>" + contact_create_rsp->get_crDate() + "</contact:crDate>"
    "</contact:creData>"
    "</resData>"
    "<trID>"
    "<clTRID>" + contact_create_rsp->get_clTRID() +"</clTRID>"
    "<svTRID>" + contact_create_rsp->get_svTRID() + "</svTRID>"
    "</trID>"
    "</response>"
    "</epp>";

  CPPUNIT_ASSERT(response == expected);
  
  /* Reset Test */
  CPPUNIT_ASSERT(contact_create_rsp->get_id() != "");
  CPPUNIT_ASSERT(contact_create_rsp->get_crDate() != "");
  
  contact_create_rsp->reset();
  
  CPPUNIT_ASSERT(contact_create_rsp->get_id() == "");
  CPPUNIT_ASSERT(contact_create_rsp->get_crDate() == "");
}