def test_expand_db_attributes_not_for_editor(self):
result = DocumentLinkHandler.expand_db_attributes(
{'id': 1},
False
)
self.assertEqual(result,
'<a href="/documents/1/">')
def test_expand_db_attributes_for_editor(self):
result = DocumentLinkHandler.expand_db_attributes(
{'id': 1},
True
)
self.assertEqual(result,
'<a data-linktype="document" data-id="1" href="/documents/1/">')
def test_get_db_attributes(self):
soup = BeautifulSoup(
'<a data-id="test-id">foo</a>'
)
tag = soup.a
result = DocumentLinkHandler.get_db_attributes(tag)
self.assertEqual(result,
{'id': 'test-id'})
def test_expand_db_attributes_not_for_editor(self):
result = DocumentLinkHandler.expand_db_attributes({'id': 1}, False)
self.assertEqual(result, '<a href="/documents/1/">')
def test_expand_db_attributes_for_editor(self):
result = DocumentLinkHandler.expand_db_attributes({'id': 1}, True)
self.assertEqual(
result,
'<a data-linktype="document" data-id="1" href="/documents/1/">')
def test_expand_db_attributes_document_does_not_exist(self):
result = DocumentLinkHandler.expand_db_attributes({'id': 0}, False)
self.assertEqual(result, '<a>')
def test_get_db_attributes(self):
soup = BeautifulSoup('<a data-id="test-id">foo</a>')
tag = soup.a
result = DocumentLinkHandler.get_db_attributes(tag)
self.assertEqual(result, {'id': 'test-id'})
def test_expand_db_attributes_document_does_not_exist(self):
result = DocumentLinkHandler.expand_db_attributes(
{'id': 0},
False
)
self.assertEqual(result, '<a>')
