def test_get_personalities_empty(self, get_nodes_mock):
get_nodes_mock.return_value = {'nodes': []}
personalities = PersonalitiesOut(personality=[])
personality_service = PersonalityService()
result = personality_service.get_personalities()
self.assertEqual(result, personalities)
get_nodes_mock.assert_called_once_with("Personality")
def test_get_personalities(self, get_nodes_mock):
get_nodes_mock.return_value = {'nodes': [{'id': 1, 'labels': ['Personality'],
'properties': [{'key': 'agreeableness', 'value': 2.5},
{'key': 'conscientiousness', 'value': 2.5},
{'key': 'extroversion', 'value': 2.5},
{'key': 'neuroticism', 'value': 2.5},
{'key': 'openess', 'value': 2.5}]},
{'id': 2, 'labels': ['Personality'],
'properties': [{'key': 'negative_affect', 'value': 0.5},
{'key': 'positive_affect', 'value': 0.5}]}]}
personality_big_five = BasicPersonalityBigFiveOut(agreeableness=2.5, conscientiousness=2.5, extroversion=2.5,
neuroticism=2.5, openess=2.5, id=1)
personality_panas = BasicPersonalityPanasOut(negative_affect=0.5, positive_affect=0.5, id=2)
personalities = PersonalitiesOut(personalities=[personality_big_five, personality_panas])
personality_service = PersonalityService()
result = personality_service.get_personalities()
self.assertEqual(result, personalities)
get_nodes_mock.assert_called_once_with("Personality")