/*
    Copyright (C) 2012 Carl Hetherington <cth@carlh.net>

    This program is free software; you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation; either version 2 of the License, or
    (at your option) any later version.

    This program 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 General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with this program; if not, write to the Free Software
    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.

*/

#include <iostream>
#include <cmath>
#include <boost/filesystem.hpp>
#include <libxml++/libxml++.h>
#include "cxml.h"

#define BOOST_TEST_DYN_LINK
#define BOOST_TEST_MODULE libdcp_test
#include <boost/test/unit_test.hpp>

using std::string;
using std::cout;
using std::vector;
using std::list;
using boost::shared_ptr;

BOOST_AUTO_TEST_CASE (read_test)
{
	cxml::NodePtr document = cxml::read_file ("test/ref/a.xml");

	BOOST_CHECK_EQUAL (document->string_child("B"), "42");
	BOOST_CHECK_EQUAL (document->number_child<int>("B"), 42);
	BOOST_CHECK_EQUAL (document->number_child<float>("B"), 42);
	BOOST_CHECK_EQUAL (document->string_child("C"), "fred");
	BOOST_CHECK_EQUAL (document->number_child<double>("D"), 42.9);
	BOOST_CHECK_EQUAL (document->string_child("E"), "yes");
	BOOST_CHECK_EQUAL (document->bool_child("E"), true);
	BOOST_CHECK_THROW (document->bool_child("F"), cxml::Error);

	BOOST_CHECK (document->optional_string_child("B"));
	BOOST_CHECK_EQUAL (document->optional_string_child("B").get(), "42");
	BOOST_CHECK (document->optional_number_child<int>("B"));
	BOOST_CHECK_EQUAL (document->optional_number_child<int>("B").get(), 42);
	BOOST_CHECK (document->optional_number_child<float>("B"));
	BOOST_CHECK_EQUAL (document->optional_number_child<float>("B").get(), 42);
	BOOST_CHECK (document->optional_string_child("C"));
	BOOST_CHECK_EQUAL (document->optional_string_child("C").get(), "fred");
	BOOST_CHECK (document->optional_number_child<double>("D"));
	BOOST_CHECK_EQUAL (document->optional_number_child<double>("D").get(), 42.9);
	BOOST_CHECK_THROW (document->child("D")->string_attribute ("foo"), cxml::Error);
	BOOST_CHECK (document->optional_string_child("E"));
	BOOST_CHECK_EQUAL (document->optional_string_child("E").get(), "yes");
	BOOST_CHECK_EQUAL (document->child("E")->string_attribute ("foo"), "bar");
	BOOST_CHECK (document->optional_bool_child("E"));
	BOOST_CHECK_EQUAL (document->optional_bool_child("E").get(), true);
	BOOST_CHECK_THROW (document->optional_bool_child("F"), cxml::Error);

	BOOST_CHECK_EQUAL (document->children("F").size(), 2);
	BOOST_CHECK_EQUAL (document->children("F").front()->content(), "1");
	BOOST_CHECK_EQUAL (document->children("F").front()->number_attribute<int>("fred"), 4);
	BOOST_CHECK_EQUAL (document->children("F").back()->content(), "2");
	BOOST_CHECK_EQUAL (document->children("F").back()->bool_attribute("jim"), true);

	BOOST_CHECK (!document->optional_bool_child("G"));

	list<shared_ptr<cxml::Node> > h = document->children ("H");
	BOOST_CHECK_EQUAL (h.size(), 1);
	BOOST_CHECK_EQUAL (h.front()->children("I").size(), 2);
	BOOST_CHECK_EQUAL (h.front()->children("I").front()->content(), "testing");
	BOOST_CHECK_EQUAL (h.front()->children("I").back()->content(), "more testing");

	BOOST_CHECK_EQUAL (document->children("J").size(), 1);
	BOOST_CHECK_EQUAL (document->children("J").front()->children("K").size(), 1);
	BOOST_CHECK_EQUAL (document->children("J").front()->children("K").front()->content(), "jim");
}

BOOST_AUTO_TEST_CASE (write_test)
{
	cxml::NodePtr document (new cxml::Node);
	document->set_name ("A");

	document->add_child("B")->set_content ("42");
	document->add_child("C")->set_content ("fred");
	document->add_child("D")->set_content ("42.9");
	cxml::NodePtr E = document->add_child("E");
	E->set_content ("1");
	E->set_attribute ("foo", "bar");
	cxml::NodePtr F1 = document->add_child("F");
	F1->set_content ("1");
	F1->set_attribute ("fred", "4");
	cxml::NodePtr F2 = document->add_child("F");
	F2->set_content ("2");
	F2->set_attribute ("jim", "1");
	cxml::NodePtr h = document->add_child ("H");
	h->add_child("I")->set_content ("testing");
	h->add_child("I")->set_content ("more testing");
	document->add_child("J")->add_child("K")->set_content ("jim");

	write_to_file_formatted (document, "build/test/b.xml");

	int r = system ("diff -u test/ref/b.xml build/test/b.xml");
	BOOST_CHECK_EQUAL (WEXITSTATUS (r), 0);
}

BOOST_AUTO_TEST_CASE (write2_test)
{
	cxml::NodePtr document (new cxml::Node);
	document->set_name ("A");
	document->add_child ("B");
	document->set_namespace_declaration ("http://my.namespace");
	document->set_namespace_declaration ("http://my.other.namespace", "foo");
	write_to_file_formatted (document, "build/test/c.xml");

	int r = system ("diff -u test/ref/c.xml build/test/c.xml");
	BOOST_CHECK_EQUAL (WEXITSTATUS (r), 0);
}