def test_get_most_specific_category(self):
self.mapper.get_commonscat_list_from_links.return_value = [
"",
u"Category:Cultural heritage ensembles in Bavaria"
]
text = u"{{Denkmalliste Sachsen Tabellenzeile|ID=1}}"
category = pageinfo.get_most_specific_category(self.mapper, text)
self.assertEqual(category, u"Category:Cultural heritage ensembles in Bavaria")
def test_get_most_specific_category_from_template(self):
self.mapper.get_commonscat.return_value = u"Category:Cultural heritage monuments in Munich"
text = u"{{Denkmalliste Sachsen Tabellenzeile|ID=1|Commonscat=Cultural heritage monuments in Munich}}"
category = pageinfo.get_most_specific_category(self.mapper, text, Mock())
self.assertEqual(category, u"Category:Cultural heritage monuments in Munich")
