Test reservoir cascade deletion

test_tjnetwork.py

@@ -612,6 +612,35 @@ class TestApi:
         self.leave(p)
 
 
+    def test_reservoir_del(self):
+        p = 'test_reservoir_del'
+        self.enter(p)
+
+        add_junction(p, ChangeSet({'id': 'j0', 'x': 0.0, 'y': 10.0, 'elevation': 20.0}))
+        add_reservoir(p, ChangeSet({'id': 'r0', 'x': 0.0, 'y': 10.0, 'head': 20.0}))
+        add_pipe(p, ChangeSet({'id': 'p0', 'node1': 'j0', 'node2': 'r0', 'length': 100.0, 'diameter': 10.0, 'roughness': 0.1, 'minor_loss': 0.5, 'status': PIPE_STATUS_OPEN }))
+        assert is_junction(p, 'j0')
+        assert is_reservoir(p, 'r0')
+        assert is_pipe(p, 'p0')
+
+        delete_reservoir(p, ChangeSet({'id': 'r0'}))
+        assert is_junction(p, 'j0')
+        assert is_reservoir(p, 'r0') == False
+        assert is_pipe(p, 'p0') == False
+
+        execute_undo(p)
+        assert is_junction(p, 'j0')
+        assert is_reservoir(p, 'r0')
+        assert is_pipe(p, 'p0')
+
+        execute_redo(p)
+        assert is_junction(p, 'j0')
+        assert is_reservoir(p, 'r0') == False
+        assert is_pipe(p, 'p0') == False
+
+        self.leave(p)
+
+
     # 4 tank