91 lines
2.3 KiB
C++
91 lines
2.3 KiB
C++
//
|
|
// 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.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_ProjectHandle 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() |