def test_non_existing_element(self):
el = etree.fromstring('''
<root></root>
''')
with self.assertRaises(IndexError):
get_agent(el)
def test_multiple_agents(self):
el = etree.fromstring('''
<root>
<agent ROLE="foo">
<name>foo_name</name>
<note>foo_note</note>
</agent>
<agent ROLE="bar">
<name>bar_name</name>
<note>bar_note</note>
</agent>
</root>
''')
self.assertEqual(get_agent(el, ROLE='foo'), {'name': 'foo_name', 'notes': ['foo_note']})
self.assertEqual(get_agent(el, ROLE='bar'), {'name': 'bar_name', 'notes': ['bar_note']})
def test_existing_othertype(self):
el = etree.fromstring('''
<root>
<agent OTHERTYPE="foo">
<name>foo_name</name>
<note>foo_note</note>
</agent>
</root>
''')
self.assertEqual(get_agent(el, OTHERTYPE='foo'), {'name': 'foo_name', 'notes': ['foo_note']})
def test_non_existing_element(self):
el = etree.fromstring('''
<root></root>
''')
self.assertIsNone(get_agent(el))
