Update pump api and test

This commit is contained in:
wqy
2022-09-25 01:05:08 +08:00
parent 3b20811605
commit dce54171fb
5 changed files with 132 additions and 147 deletions

View File

@@ -454,7 +454,7 @@ class TestApi:
self.leave(p)
'''
def test_pump(self):
p = "test_pump"
self.enter(p)
@@ -488,30 +488,30 @@ class TestApi:
pumps = get_links(p)
assert len(pumps) == 1
assert get_pump_properties(p, 'p1')['node1'] == 'j1'
assert get_pump_properties(p, 'p1')['node2'] == 'j2'
assert get_pump(p, 'p1')['node1'] == 'j1'
assert get_pump(p, 'p1')['node2'] == 'j2'
change_set = set_pump_node1(p, 'p1', 'j2')
change_set = set_pump(p, 'p1', {'node1': 'j2'})
assert len(change_set.operations) == 0
change_set = set_pump_node2(p, 'p1', 'j1')
change_set = set_pump(p, 'p1', {'node2': 'j1'})
assert len(change_set.operations) == 0
change_set = set_pump_node1(p, 'p1', 'j3')
change_set = set_pump(p, 'p1', {'node1': 'j3'})
assert len(change_set.operations) == 1
assert change_set.operations[0]['operation'] == 'update'
assert change_set.operations[0]['type'] == PUMP
assert change_set.operations[0]['id'] == 'p1'
assert change_set.operations[0]['properties'] == 'node1'
assert change_set.operations[0]['properties'] == ['node1']
change_set = set_pump_node2(p, 'p1', 'j4')
change_set = set_pump(p, 'p1', {'node2': 'j4'})
assert len(change_set.operations) == 1
assert change_set.operations[0]['operation'] == 'update'
assert change_set.operations[0]['type'] == PUMP
assert change_set.operations[0]['id'] == 'p1'
assert change_set.operations[0]['properties'] == 'node2'
assert change_set.operations[0]['properties'] == ['node2']
assert get_pump_properties(p, 'p1')['node1'] == 'j3'
assert get_pump_properties(p, 'p1')['node2'] == 'j4'
assert get_pump(p, 'p1')['node1'] == 'j3'
assert get_pump(p, 'p1')['node2'] == 'j4'
change_set = add_pump(p, 'p2', 'j1', 'j2')
assert len(change_set.operations) == 1
@@ -544,7 +544,7 @@ class TestApi:
self.leave(p)
'''
def test_valve(self):
p = "test_valve"
self.enter(p)