def test_entities(self):
entity = {
'key1': 'val1',
'key2': 'val2',
'entities': [
{
'k1': 'v1'
}
],
'handle': 'handle'
}
parent = 'parent'
expected = [
{
'class': 'child',
'parentHandle': 'parent',
'handle': 'handle'
},
{
'class': 'child',
'parentHandle': 'handle'
}
]
actual = rdap._parse_entity(entity, parent)
self.assertEqual(expected, actual)
def test_no_desired_fields(self):
entity = {
'key1': 'val1',
'key2': 'val2'
}
parent = 'parent'
expected = [
{
'class': 'child',
'parentHandle': 'parent'
}
]
actual = rdap._parse_entity(entity, parent)
self.assertEqual(expected, actual)
def test_roles(self):
entity = {
'key1': 'val1',
'key2': 'val2',
'roles': ['cool_person', 'dope_dude']
}
parent = 'parent'
expected = [
{
'class': 'child',
'parentHandle': 'parent',
'roles': 'cool_person,dope_dude'
}
]
actual = rdap._parse_entity(entity, parent)
self.assertEqual(expected, actual)
def test_events(self, mock_parse_events):
mock_parse_events.return_value = {'derp': 'herp'}
entity = {
'key1': 'val1',
'key2': 'val2',
'events': 'events'
}
parent = 'parent'
expected = [
{
'class': 'child',
'parentHandle': 'parent',
'derp': 'herp'
}
]
actual = rdap._parse_entity(entity, parent)
self.assertEqual(expected, actual)
mock_parse_events.assert_called_once_with('events')
def test_vcard_array(self, mock_parse_vcard):
mock_parse_vcard.return_value = {'derp': 'herp'}
entity = {
'key1': 'val1',
'key2': 'val2',
'vcardArray': ['vcard', 'stuff']
}
parent = 'parent'
expected = [
{
'class': 'child',
'parentHandle': 'parent',
'derp': 'herp'
}
]
actual = rdap._parse_entity(entity, parent)
self.assertEqual(expected, actual)
mock_parse_vcard.assert_called_once_with('stuff')