Reorganizing unit tests

Unit tests for the toolkit are consolidated into one test module with a separate test suite for related API functions.
2019-03-21 13:11:10 -04:00
parent 80ce382a38
commit 5687c63548
17 changed files with 511 additions and 363 deletions
--- a/tests/test_control.cpp
+++ b/tests/test_control.cpp
@@ -0,0 +1,87 @@
+/*
+ ******************************************************************************
+ Project:      OWA EPANET
+ Version:      2.2
+ Module:       test_control.cpp
+ Description:  Tests EPANET toolkit api functions
+ Authors:      see AUTHORS
+ Copyright:    see AUTHORS
+ License:      see LICENSE
+ Last Updated: 03/21/2019
+ ******************************************************************************
+*/
+
+/*
+This is a test for the API functions that adds rules and deletes
+nodes and links from a project. Deletion can be conditional on
+node or link appearing in any simple or rule-based controls.
+*/
+
+
+#define BOOST_ALL_DYN_LINK
+#include <boost/test/unit_test.hpp>
+
+#include "test_toolkit.hpp"
+
+
+char R1[] = "RULE 1 \n IF NODE 2 LEVEL < 100 \n THEN LINK 9 STATUS = OPEN";
+char R2[] = "RULE 2\nIF SYSTEM TIME = 4\nTHEN LINK 9 STATUS = CLOSED\nAND LINK 31 STATUS = CLOSED";
+char R3[] = "RULE 3\nIF NODE 23 PRESSURE ABOVE 140\nAND NODE 2 LEVEL > 120\n"
+            "THEN LINK 113 STATUS = CLOSED\nELSE LINK 22 STATUS = CLOSED";
+
+
+BOOST_AUTO_TEST_SUITE (test_controls)
+
+BOOST_FIXTURE_TEST_CASE(test_add_get_rule,  FixtureOpenClose)
+{
+    int ruleCount, nP, nTA, nEA;
+    int link113, node23, link22, pump9_before, pump9_after;
+    double priority;
+
+
+    // Add the 3 rules to the project
+    error = EN_addrule(ph, R1);
+    BOOST_REQUIRE(error == 0);
+    error = EN_addrule(ph, R2);
+    BOOST_REQUIRE(error == 0);
+    error = EN_addrule(ph, R3);
+    BOOST_REQUIRE(error == 0);
+
+    // Check that rules were added
+    error = EN_getcount(ph, EN_RULECOUNT, &ruleCount);
+    BOOST_CHECK(ruleCount == 3);
+
+    // Check the number of clauses in rule 3
+    error = EN_getrule(ph, 3, &nP, &nTA, &nEA, &priority);
+    BOOST_CHECK(nP == 2);
+    BOOST_CHECK(nTA == 1);
+    BOOST_CHECK(nTA == 1);
+
+    // Try to delete link 113 conditionally which will fail
+    // because it's in rule 3
+    EN_getlinkindex(ph, (char *)"113", &link113);
+    error = EN_deletelink(ph, link113, EN_CONDITIONAL);
+    BOOST_REQUIRE(error == 261);
+
+    // Delete node 23 unconditionally which will result in the
+    // deletion of rule 3 as well as links 22 and 113
+    EN_getnodeindex(ph, (char *)"23", &node23);
+    EN_getlinkindex(ph, (char *)"22", &link22);
+    EN_getlinkindex(ph, (char *)"9", &pump9_before);
+    error = EN_deletenode(ph, node23, EN_UNCONDITIONAL);
+    BOOST_REQUIRE(error == 0);
+
+    // Check that there are now only 2 rules
+    error = EN_getcount(ph, EN_RULECOUNT, &ruleCount);
+    BOOST_CHECK(ruleCount == 2);
+
+    // Check that link 22 no longer exists
+    error = EN_getlinkindex(ph, (char *)"22", &link22);
+    BOOST_CHECK(error > 0);
+
+    // Check that the index of pump9 has been reduced by 2
+    error = EN_getlinkindex(ph, (char *)"9", &pump9_after);
+    BOOST_CHECK(pump9_before - pump9_after == 2);
+}
+
+BOOST_AUTO_TEST_SUITE_END()