class CheckManagerTest(TestCase):
"""
Base class for all check manager tests.
"""
def setUp(self):
"""
initialize a manager.
"""
self.manager = Entitylink()
self.name = 'testentitylink'
self.id = str(uuid4())
self.ids = [self.id]
self.document_content = {
'computed_links': [{
'label': 'link1',
'url': 'http://www.canopsis.org'
}]
}
def clean(self):
self.manager.remove(self.ids)
def get_linklist(self):
return self.manager.find(ids=self.ids)
def linklist_count_equals(self, count):
result = list(self.get_linklist())
if DEBUG:
print result
result = len(result)
self.assertEqual(result, count)
