From edd731904e584a1c28addd460da3c90ee767a7c3 Mon Sep 17 00:00:00 2001
From: wqy <wqyuyu@outlook.com>
Date: Sat, 17 Sep 2022 22:02:53 +0800
Subject: [PATCH] Add tank test case

---
 test_tjnetwork.py | 100 +++++++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 99 insertions(+), 1 deletion(-)

diff --git a/test_tjnetwork.py b/test_tjnetwork.py
index 7d47a1a..c910181 100644
--- a/test_tjnetwork.py
+++ b/test_tjnetwork.py
@@ -113,6 +113,8 @@ class TestApi:
         assert change_set.updated[0]['property_value'] == str({'x': 100.0, 'y': 200.0})
         assert get_reservoir_coord(p, 'r0') == { 'x': 100.0, 'y': 200.0 }
 
+        # TODO: pattern
+
         change_set = delete_reservoir(p, 'r0')
         assert len(change_set.deleted) == 1
         assert change_set.deleted[0]['type'] == RESERVOIR
@@ -121,7 +123,103 @@ class TestApi:
         self.leave(p)
 
     def test_tank(self):
-        pass
+        p = "test_tank"
+
+        self.enter(p)
+
+        change_set = add_tank(p, "t0", 10.0, 10.0, 10.0, 10.0, 10.0, 10.0, 10.0, 10.0)
+        assert len(change_set.added) == 1
+        assert change_set.added[0]['type'] == TANK
+        assert change_set.added[0]['id'] == "t0"
+
+        change_set = add_tank(p, "t1", 0.0, 10.0, 20.0)
+        assert len(change_set.added) == 1
+        assert change_set.added[0]['type'] == TANK
+        assert change_set.added[0]['id'] == "t1"
+
+        tanks = get_nodes(p)
+        assert len(tanks) == 2
+        assert tanks[0] == 't0'
+        assert tanks[1] == 't1'
+
+        assert get_tank_elevation(p, 't0') == 10.0
+        change_set = set_tank_elevation(p, 't0', 100.0)
+        assert len(change_set.updated) == 1
+        assert change_set.updated[0]['type'] == TANK
+        assert change_set.updated[0]['id'] == "t0"
+        assert change_set.updated[0]['property'] == 'elevation'
+        assert get_tank_elevation(p, 't0') == 100.0
+
+        assert get_tank_init_level(p, 't0') == 10.0
+        change_set = set_tank_init_level(p, 't0', 100.0)
+        assert len(change_set.updated) == 1
+        assert change_set.updated[0]['type'] == TANK
+        assert change_set.updated[0]['id'] == "t0"
+        assert change_set.updated[0]['property'] == 'init_level'
+        assert get_tank_init_level(p, 't0') == 100.0
+
+        assert get_tank_min_level(p, 't0') == 10.0
+        change_set = set_tank_min_level(p, 't0', 100.0)
+        assert len(change_set.updated) == 1
+        assert change_set.updated[0]['type'] == TANK
+        assert change_set.updated[0]['id'] == "t0"
+        assert change_set.updated[0]['property'] == 'min_level'
+        assert get_tank_min_level(p, 't0') == 100.0
+
+        assert get_tank_max_level(p, 't0') == 10.0
+        change_set = set_tank_max_level(p, 't0', 100.0)
+        assert len(change_set.updated) == 1
+        assert change_set.updated[0]['type'] == TANK
+        assert change_set.updated[0]['id'] == "t0"
+        assert change_set.updated[0]['property'] == 'max_level'
+        assert get_tank_max_level(p, 't0') == 100.0
+
+        assert get_tank_diameter(p, 't0') == 10.0
+        change_set = set_tank_diameter(p, 't0', 100.0)
+        assert len(change_set.updated) == 1
+        assert change_set.updated[0]['type'] == TANK
+        assert change_set.updated[0]['id'] == "t0"
+        assert change_set.updated[0]['property'] == 'diameter'
+        assert get_tank_diameter(p, 't0') == 100.0
+
+        assert get_tank_min_vol(p, 't0') == 10.0
+        change_set = set_tank_min_vol(p, 't0', 100.0)
+        assert len(change_set.updated) == 1
+        assert change_set.updated[0]['type'] == TANK
+        assert change_set.updated[0]['id'] == "t0"
+        assert change_set.updated[0]['property'] == 'min_vol'
+        assert get_tank_min_vol(p, 't0') == 100.0
+
+        # TODO: vol_curve
+
+        assert get_tank_overflow(p, 't') == None
+        assert get_tank_overflow(p, 't0') == 'NULL'
+        change_set = set_tank_overflow(p, 't0', "XXX")
+        assert len(change_set.updated) == 0
+        change_set = set_tank_overflow(p, 't0', OVERFLOW_YES)
+        assert len(change_set.updated) == 1
+        assert change_set.updated[0]['type'] == TANK
+        assert change_set.updated[0]['id'] == "t0"
+        assert change_set.updated[0]['property'] == 'overflow'
+        assert get_tank_overflow(p, 't0') == OVERFLOW_YES
+
+        change_set = delete_tank(p, "t0")
+        assert len(change_set.deleted) == 1
+        assert change_set.deleted[0]['type'] == TANK
+        assert change_set.deleted[0]['id'] == "t0"
+
+        tanks = get_nodes(p)
+        assert len(tanks) == 1
+
+        change_set = delete_tank(p, "t1")
+        assert len(change_set.deleted) == 1
+        assert change_set.deleted[0]['type'] == TANK
+        assert change_set.deleted[0]['id'] == "t1"
+
+        tanks = get_nodes(p)
+        assert len(tanks) == 0
+
+        self.leave(p)
 
     def test_pipe(self):
         pass