def test_get_db_attributes(self):
soup = BeautifulSoup('<a data-id="test-id">foo</a>', 'html5lib')
tag = soup.a
result = EditorHtmlDocumentLinkHandler.get_db_attributes(tag)
self.assertEqual(
result,
{'id': 'test-id'}
)
def test_expand_db_attributes_for_editor_with_missing_id(self):
result = EditorHtmlDocumentLinkHandler.expand_db_attributes({})
self.assertEqual(result, '<a data-linktype="document">')
def test_expand_db_attributes_for_editor_preserves_id_of_nonexistent_document(
self):
result = EditorHtmlDocumentLinkHandler.expand_db_attributes({'id': 0})
self.assertEqual(result, '<a data-linktype="document" data-id="0">')
def test_expand_db_attributes_for_editor(self):
result = EditorHtmlDocumentLinkHandler.expand_db_attributes({'id': 1})
self.assertEqual(
result,
'<a data-linktype="document" data-id="1" href="/documents/1/test.pdf">'
)
def test_get_db_attributes(self):
soup = BeautifulSoup('<a data-id="test-id">foo</a>', 'html5lib')
tag = soup.a
result = EditorHtmlDocumentLinkHandler.get_db_attributes(tag)
self.assertEqual(result, {'id': 'test-id'})
def test_expand_db_attributes_for_editor_with_missing_id(self):
result = EditorHtmlDocumentLinkHandler.expand_db_attributes({})
self.assertEqual(
result,
'<a data-linktype="document">'
)
def test_expand_db_attributes_for_editor_preserves_id_of_nonexistent_document(self):
result = EditorHtmlDocumentLinkHandler.expand_db_attributes({'id': 0})
self.assertEqual(
result,
'<a data-linktype="document" data-id="0">'
)
def test_expand_db_attributes_for_editor(self):
result = EditorHtmlDocumentLinkHandler.expand_db_attributes({'id': 1})
self.assertEqual(
result,
'<a data-linktype="document" data-id="1" href="/documents/1/test.pdf">'
)