#include "rpc_helper.h"

#include <cppunit/extensions/HelperMacros.h>

#include "RpcRequest.h"
#include "RecoverableException.h"
#ifdef ENABLE_XML_RPC
# include "XmlRpcRequestParserStateMachine.h"
#endif // ENABLE_XML_RPC

namespace aria2 {

namespace rpc {

class RpcHelperTest:public CppUnit::TestFixture {

  CPPUNIT_TEST_SUITE(RpcHelperTest);
#ifdef ENABLE_XML_RPC
  CPPUNIT_TEST(testParseMemory);
  CPPUNIT_TEST(testParseMemory_shouldFail);
  CPPUNIT_TEST(testParseMemory_withoutStringTag);
#endif // ENABLE_XML_RPC
  CPPUNIT_TEST_SUITE_END();
public:
  void setUp() {}

  void tearDown() {}

#ifdef ENABLE_XML_RPC
  void testParseMemory();
  void testParseMemory_shouldFail();
  void testParseMemory_withoutParams();
  void testParseMemory_withoutStringTag();
#endif // ENABLE_XML_RPC
};

CPPUNIT_TEST_SUITE_REGISTRATION(RpcHelperTest);

#ifdef ENABLE_XML_RPC

void RpcHelperTest::testParseMemory()
{
  std::string s =
    "<?xml version=\"1.0\"?>"
    "<methodCall>"
    "  <methodName>aria2.addURI</methodName>"
    "    <params>"
    "      <param>"
    "        <value><i4>100</i4></value>"
    "      </param>"
    "      <param>"
    "       <value>"
    "         <struct>"
    "           <member>"
    "             <name>max-count</name>"
    "             <value><i4>65535</i4></value>"
    "           </member>"
    "           <member>"
    "             <name>seed-ratio</name>"
    "             <value><double>0.99</double></value>"
    "           </member>"
    "         </struct>"
    "       </value>"
    "     </param>"
    "     <param>"
    "       <value>"
    "         <array>"
    "           <data>"
    "             <value><string>pudding</string></value>"
    "             <value><base64>aGVsbG8gd29ybGQ=</base64></value>"
    "           </data>"
    "         </array>"
    "       </value>"
    "     </param>"
    "   </params>"
    "</methodCall>";
  RpcRequest req = xmlParseMemory(s.c_str(), s.size());

  CPPUNIT_ASSERT_EQUAL(std::string("aria2.addURI"), req.methodName);
  CPPUNIT_ASSERT_EQUAL((size_t)3, req.params->size());
  CPPUNIT_ASSERT_EQUAL((Integer::ValueType)100,
                       downcast<Integer>(req.params->get(0))->i());
  const Dict* dict = downcast<Dict>(req.params->get(1));
  CPPUNIT_ASSERT_EQUAL((Integer::ValueType)65535,
                       downcast<Integer>(dict->get("max-count"))->i());
  // Current implementation handles double as string.
  CPPUNIT_ASSERT_EQUAL(std::string("0.99"),
                       downcast<String>(dict->get("seed-ratio"))->s());
  const List* list = downcast<List>(req.params->get(2));
  CPPUNIT_ASSERT_EQUAL(std::string("pudding"), downcast<String>(list->get(0))->s());
  CPPUNIT_ASSERT_EQUAL(std::string("hello world"), downcast<String>(list->get(1))->s());
}

void RpcHelperTest::testParseMemory_shouldFail()
{
  try {
    std::string s =
      "<methodCall>"
      "  <methodName>aria2.addURI</methodName>"
      "    <params>"
      "      <param>"
      "        <value><i4>100</i4></value>"
      "      </param>";
    xmlParseMemory(s.c_str(), s.size());
    CPPUNIT_FAIL("exception must be thrown.");
  } catch(RecoverableException& e) {
    // success
  }
}

void RpcHelperTest::testParseMemory_withoutParams()
{
  {
    std::string s =
      "<methodCall>"
      "  <methodName>aria2.addURI</methodName>"
      "    <params>"
      "    </params>"
      "</methodCall>";
    RpcRequest req = xmlParseMemory(s.c_str(), s.size());
    CPPUNIT_ASSERT(req.params);
  }
  {
    std::string s =
      "<methodCall>"
      "  <methodName>aria2.addURI</methodName>"
      "</methodCall>";
    RpcRequest req = xmlParseMemory(s.c_str(), s.size());
    CPPUNIT_ASSERT(req.params->size());
  }
}

void RpcHelperTest::testParseMemory_withoutStringTag()
{
  std::string s =
    "<?xml version=\"1.0\"?>"
    "<methodCall>"
    "  <methodName>aria2.addUri</methodName>"
    "    <params>"
    "      <param>"
    "        <value>http://aria2.sourceforge.net</value>"
    "      </param>"
    "      <param>"
    "        <value>http://aria2.<foo/>sourceforge.net</value>"
    "      </param>"
    "      <param>"
    "       <value>"
    "         <struct>"
    "           <member>"
    "             <name>hello</name>"
    "             <value>world</value>"
    "           </member>"
    "         </struct>"
    "       </value>"
    "     </param>"
    "     <param>"
    "       <value>"
    "         <array>"
    "           <data>"
    "             <value>apple</value>"
    "             <value>banana</value>"
    "             <value><string>lemon</string>peanuts</value>"
    "           </data>"
    "         </array>"
    "       </value>"
    "     </param>"
    "   </params>"
    "</methodCall>";
  RpcRequest req =
    xmlParseMemory(s.c_str(), s.size());

  CPPUNIT_ASSERT_EQUAL((size_t)4, req.params->size());
  CPPUNIT_ASSERT_EQUAL(std::string("http://aria2.sourceforge.net"),
                       downcast<String>(req.params->get(0))->s());
  CPPUNIT_ASSERT_EQUAL(std::string("http://aria2.sourceforge.net"),
                       downcast<String>(req.params->get(1))->s());
  const Dict* dict = downcast<Dict>(req.params->get(2));
  CPPUNIT_ASSERT_EQUAL(std::string("world"),
                       downcast<String>(dict->get("hello"))->s());
  const List* list = downcast<List>(req.params->get(3));
  CPPUNIT_ASSERT_EQUAL(std::string("apple"), downcast<String>(list->get(0))->s());
  CPPUNIT_ASSERT_EQUAL(std::string("banana"), downcast<String>(list->get(1))->s());
  CPPUNIT_ASSERT_EQUAL(std::string("lemon"), downcast<String>(list->get(2))->s());
}

#endif // ENABLE_XML_RPC

} // namespace rpc

} // namespace aria2
