//
// test_demand_categories.cpp
//

/*
This is a test for the demand categories names get\set APIs
A demand category name is set, the network is saved, reopened and the new demand category name is checked.
*/

#define BOOST_TEST_MODULE "toolkit"
#include <boost/test/included/unit_test.hpp>

#include <string>
#include "epanet2_2.h"

// NOTE: Project Home needs to be updated to run unit test
#define DATA_PATH_INP "./net1.inp"
#define DATA_PATH_RPT "./test.rpt"
#define DATA_PATH_OUT "./test.out"

using namespace std;

boost::test_tools::predicate_result check_string(std::string test, std::string ref)
{
	if (ref.compare(test) == 0)
		return true;
	else
		return false;
}

BOOST_AUTO_TEST_SUITE (test_toolkit)

BOOST_AUTO_TEST_CASE(test_demand_categories)
{
	string path_inp(DATA_PATH_INP);
	string inp_save("net1_dem_cat.inp");
	string path_rpt(DATA_PATH_RPT);
	string path_out(DATA_PATH_OUT);

	char node_id[] = "12";
	char demand_category[] = "Demand category name";
	char demname[80];


	int error = 0;
    int Nindex, ndem;

    EN_Project ph = NULL;

    error = EN_createproject(&ph);
    error = EN_open(ph, path_inp.c_str(), path_rpt.c_str(), path_out.c_str());

    error = EN_getnodeindex(ph, node_id, &Nindex);
    BOOST_REQUIRE(error == 0);
    error = EN_getnumdemands(ph, Nindex, &ndem);
    BOOST_REQUIRE(error == 0);
    BOOST_CHECK(ndem == 1);

	error = EN_setdemandname(ph, Nindex, ndem, demand_category);
    BOOST_REQUIRE(error == 0);
    error = EN_saveinpfile(ph, inp_save.c_str());
    BOOST_REQUIRE(error == 0);

    error = EN_close(ph);
	BOOST_REQUIRE(error == 0);
	error = EN_deleteproject(&ph);
	BOOST_REQUIRE(error == 0);

	BOOST_TEST_CHECKPOINT("Reopening saved input file");
    error = EN_createproject(&ph);
	BOOST_REQUIRE(error == 0);
	error = EN_open(ph, inp_save.c_str(), path_rpt.c_str(), path_out.c_str());
	BOOST_REQUIRE(error == 0);

    error = EN_getnodeindex(ph, node_id, &Nindex);
    BOOST_REQUIRE(error == 0);
    error = EN_getnumdemands(ph, Nindex, &ndem);
    BOOST_REQUIRE(error == 0);
    BOOST_CHECK(ndem == 1);

	error = EN_getdemandname(ph, Nindex, ndem, demname);
    BOOST_REQUIRE(error == 0);
    BOOST_CHECK(check_string(demname, demand_category));

	error = EN_close(ph);
	BOOST_REQUIRE(error == 0);
	error = EN_deleteproject(&ph);
	BOOST_REQUIRE(error == 0);
}

BOOST_AUTO_TEST_SUITE_END()