From 2c584f1a15c32c33759827b3d7d6a8be9fa9ac6f Mon Sep 17 00:00:00 2001
From: "WQY\\qiong" <qiongyu.wang@outlook.com>
Date: Sat, 4 Feb 2023 14:46:24 +0800
Subject: [PATCH] Test tank cascade deletion

---
 test_tjnetwork.py | 38 ++++++++++++++++++++++++++++++++++++++
 1 file changed, 38 insertions(+)

diff --git a/test_tjnetwork.py b/test_tjnetwork.py
index 344fb5a..1bd5cc5 100644
--- a/test_tjnetwork.py
+++ b/test_tjnetwork.py
@@ -855,6 +855,40 @@ class TestApi:
         self.leave(p)
 
 
+    def test_tank_del(self):
+        p = 'test_tank_del'
+        self.enter(p)
+
+        add_junction(p, ChangeSet({'id': 'j0', 'x': 0.0, 'y': 10.0, 'elevation': 20.0}))
+        add_tank(p, ChangeSet({'id': 't0', 'x': 0.0, 'y': 10.0, 'elevation': 20.0, 'init_level': 1.0, 'min_level': 0.0, 'max_level': 2.0, 'diameter': 10.0, 'min_vol': 100.0, 'vol_curve': None, 'overflow': OVERFLOW_NO}))
+        add_pipe(p, ChangeSet({'id': 'p0', 'node1': 'j0', 'node2': 't0', 'length': 100.0, 'diameter': 10.0, 'roughness': 0.1, 'minor_loss': 0.5, 'status': PIPE_STATUS_OPEN }))
+        set_tank_reaction(p, ChangeSet({'tank': 't0', 'value': 10.0}))
+        assert is_junction(p, 'j0')
+        assert is_tank(p, 't0')
+        assert is_pipe(p, 'p0')
+        assert get_tank_reaction(p, 't0')['value'] == 10.0
+
+        delete_tank(p, ChangeSet({'id': 't0'}))
+        assert is_junction(p, 'j0')
+        assert is_tank(p, 't0') == False
+        assert is_pipe(p, 'p0') == False
+        assert get_tank_reaction(p, 't0')['value'] == None
+
+        execute_undo(p)
+        assert is_junction(p, 'j0')
+        assert is_tank(p, 't0')
+        assert is_pipe(p, 'p0')
+        assert get_tank_reaction(p, 't0')['value'] == 10.0
+
+        execute_redo(p)
+        assert is_junction(p, 'j0')
+        assert is_tank(p, 't0') == False
+        assert is_pipe(p, 'p0') == False
+        assert get_tank_reaction(p, 't0')['value'] == None
+
+        self.leave(p)
+
+
     # 5 pipe
 
 
@@ -2453,6 +2487,8 @@ class TestApi:
         add_pattern(p, ChangeSet({'id' : 'p0', 'factors': [1.0, 2.0, 3.0]}))
         add_pattern(p, ChangeSet({'id' : 'p1', 'factors': [1.0, 2.0, 3.0]}))
 
+        assert get_source(p, 'j0') == {}
+
         add_source(p, ChangeSet({'node': 'j0', 's_type': SOURCE_TYPE_CONCEN, 'strength': 10.0, 'pattern': 'p0'}))
         s = get_source(p, 'j0')
         assert s['node'] == 'j0'
@@ -2483,6 +2519,8 @@ class TestApi:
 
         delete_source(p, ChangeSet({'node': 'j0'}))
 
+        assert get_source(p, 'j0') == {}
+
         self.leave(p)