def test_get_arrangements(self, get_nodes_mock):
get_nodes_mock.return_value = {
'nodes': [{
'id':
1,
'labels': ['Arrangement'],
'properties': [{
'key': 'arrangement_type',
'value': 'test'
}]
}, {
'id':
2,
'labels': ['Arrangement'],
'properties': [{
'key': 'arrangement_type',
'value': 'test2'
}]
}],
'errors':
None
}
arrangement_one = BasicArrangementOut(arrangement_type="test", id=1)
arrangement_two = BasicArrangementOut(arrangement_type="test2", id=2)
arrangements = ArrangementsOut(
arrangements=[arrangement_one, arrangement_two])
arrangement_service = ArrangementService()
result = arrangement_service.get_arrangements()
self.assertEqual(result, arrangements)
get_nodes_mock.assert_called_once_with("Arrangement")
def test_get_arrangements_empty(self, get_nodes_mock):
get_nodes_mock.return_value = {'nodes': [], 'errors': None}
arrangements = ArrangementsOut(arrangements=[])
arrangement_service = ArrangementService()
result = arrangement_service.get_arrangements()
self.assertEqual(result, arrangements)
get_nodes_mock.assert_called_once_with("Arrangement")
def set_arrangements(self):
"""
Initialize values of arrangement distances
"""
arrangement_service = ArrangementService()
created_arrangements = \
[arrangement.arrangement_distance
for arrangement in
arrangement_service.get_arrangements().arrangements]
[
arrangement_service.save_arrangement(
ArrangementIn(arrangement_type=arrangement.value[0],
arrangement_distance=arrangement.value[1]))
for arrangement in Arrangement
if arrangement.value[1] not in created_arrangements
]