def test_save_arrangement_with_node_error(self, create_node_mock):
id_node = 1
create_node_mock.return_value = {'id': id_node, 'properties': None, "errors": ['error'], 'links': None}
arrangement = ArrangementIn(arrangement_type='personal two persons', arrangement_distance='intimate zone')
arrangement_service = ArrangementService()
result = arrangement_service.save_arrangement(arrangement)
self.assertEqual(result, ArrangementOut(arrangement_type='personal two persons', arrangement_distance='intimate zone', errors=['error']))
create_node_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
]
def test_save_arrangement_without_error(self, create_properties_mock, create_node_mock):
id_node = 1
create_node_mock.return_value = {'id': id_node, 'properties': None, "errors": None, 'links': None}
create_properties_mock.return_value = {'id': id_node, 'properties': [{'key': 'arrangement_type',
'value': 'personal two persons'},
{'key': 'arrangement_distance',
'value': 'intimate zone'}],
"errors": None, 'links': None}
arrangement = ArrangementIn(arrangement_type='personal two persons', arrangement_distance='intimate zone')
arrangement_service = ArrangementService()
result = arrangement_service.save_arrangement(arrangement)
self.assertEqual(result, ArrangementOut(id=id_node, arrangement_type='personal two persons', arrangement_distance='intimate zone'))
create_node_mock.assert_called_once_with('Arrangement')
create_properties_mock.assert_called_once_with(id_node, arrangement)