def test_make_links_node(self): r = ioc_et.make_links_node(self.links) self.assertEqual(r.tag, 'links') for i, link in enumerate(r.getchildren()): self.assertEqual(link.tag, 'link') rel, href, value = self.links[i] self.assertEqual(link.attrib['rel'], rel) self.assertEqual(link.attrib.get('href'), href) self.assertEqual(link.text, value)
def add_link(self, rel, value, href=None): """ Add a Link metadata element to the IOC. :param rel: Type of the link. :param value: Value of the link text. :param href: A href value assigned to the link. :return: True """ links_node = self.metadata.find('links') if links_node is None: links_node = ioc_et.make_links_node() self.metadata.append(links_node) link_node = ioc_et.make_link_node(rel, value, href) links_node.append(link_node) return True