From 685c8a3a338576cb305c605f6d48a7a91e40717c Mon Sep 17 00:00:00 2001 From: "WQY\\qiong" Date: Sat, 4 Feb 2023 15:08:08 +0800 Subject: [PATCH] Test pump cascade deletion --- test_tjnetwork.py | 45 +++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 45 insertions(+) diff --git a/test_tjnetwork.py b/test_tjnetwork.py index 181590f..1456dbb 100644 --- a/test_tjnetwork.py +++ b/test_tjnetwork.py @@ -1296,6 +1296,51 @@ class TestApi: self.leave(p) + def test_pipe_del(self): + p = 'test_pipe_del' + self.enter(p) + + add_junction(p, ChangeSet({'id': 'j0', 'x': 0.0, 'y': 10.0, 'elevation': 20.0})) + add_junction(p, ChangeSet({'id': 'j1', 'x': 0.0, 'y': 10.0, 'elevation': 20.0})) + add_pump(p, ChangeSet({'id': 'p0', 'node1': 'j0', 'node2': 'j1', 'power': 0.0})) + set_tag(p, ChangeSet({'t_type': TAG_TYPE_LINK, 'id': 'p0', 'tag': 'p0t' })) + set_status(p, ChangeSet({'link': 'p0', 'status': LINK_STATUS_OPEN, 'setting': 10.0})) + set_pump_energy(p, ChangeSet({'pump' : 'p0', 'price': 1.0})) + set_vertex(p, ChangeSet({'link' : 'p0', 'coords': [{'x': 1.0, 'y': 2.0}, {'x': 2.0, 'y': 1.0}]})) + assert is_pump(p, 'p0') + assert get_tag(p, TAG_TYPE_LINK, 'p0')['tag'] == 'p0t' + assert get_status(p, 'p0')['status'] == LINK_STATUS_OPEN + assert get_status(p, 'p0')['setting'] == 10.0 + assert get_pump_energy(p, 'p0')['price'] == 1.0 + assert get_vertex(p, 'p0')['coords'] == [{'x': 1.0, 'y': 2.0}, {'x': 2.0, 'y': 1.0}] + + delete_pump(p, ChangeSet({'id': 'p0'})) + assert is_pump(p, 'p0') == False + assert get_tag(p, TAG_TYPE_LINK, 'p0')['tag'] == None + assert get_status(p, 'p0')['status'] == None + assert get_status(p, 'p0')['setting'] == None + assert get_pump_energy(p, 'p0')['price'] == None + assert get_vertex(p, 'p0')['coords'] == [] + + execute_undo(p) + assert is_pump(p, 'p0') + assert get_tag(p, TAG_TYPE_LINK, 'p0')['tag'] == 'p0t' + assert get_status(p, 'p0')['status'] == LINK_STATUS_OPEN + assert get_status(p, 'p0')['setting'] == 10.0 + assert get_pump_energy(p, 'p0')['price'] == 1.0 + assert get_vertex(p, 'p0')['coords'] == [{'x': 1.0, 'y': 2.0}, {'x': 2.0, 'y': 1.0}] + + execute_redo(p) + assert is_pump(p, 'p0') == False + assert get_tag(p, TAG_TYPE_LINK, 'p0')['tag'] == None + assert get_status(p, 'p0')['status'] == None + assert get_status(p, 'p0')['setting'] == None + assert get_pump_energy(p, 'p0')['price'] == None + assert get_vertex(p, 'p0')['coords'] == [] + + self.leave(p) + + # 7 valve