def test_adding_multiple_values_in_init(self):
paragraph = HTMLPDFField.Paragraph(HTMLPDFField.Text('test'))
image = HTMLPDFField.Image(None, attrs={'src': 'image.jpg'})
html_list = HTMLPDFField.ElementList(paragraph, image)
self.assertEqual(len(html_list), 2)
self.assertEqual(html_list[0], paragraph)
self.assertEqual(html_list[1], image)
def test_anchor_clean_with_relative_url_with_request_without_slash(self):
request = RequestFactory().get('/some-page/',
HTTP_HOST='example-pdf-host.com',
secure=True)
anchor = HTMLPDFField.Anchor(None, attrs={'href': 'data/'})
anchor.clean({'request': request})
self.assertEqual(anchor.attrs['href'],
'https://example-pdf-host.com/some-page/data/')
def test_anchor_clean_with_relative_url_with_request(self):
request = RequestFactory().get('/page/other/',
HTTP_HOST='example-pdf-host.com',
secure=False)
anchor = HTMLPDFField.Anchor(None, attrs={'href': '/data/'})
anchor.clean({'request': request})
self.assertEqual(anchor.attrs['href'],
'http://example-pdf-host.com/data/')
def test_image_without_src(self):
with self.assertRaises(PDFFieldCleaningError) as cm:
HTMLPDFField.Image(None)
exception_message = str(cm.exception)
self.assertIn('src', exception_message)
self.assertIn('HTMLPDFFieldImage', exception_message)
def setUp(self):
self.html_field = HTMLPDFField('some_name')
class TestHTMLPDFField(TestCase):
def setUp(self):
self.html_field = HTMLPDFField('some_name')
def test_field_name(self):
self.assertEqual(self.html_field.name, 'some_name')
def test_simple_paragraph(self):
cleaned_value = self.html_field.clean('<p>Lorem Ipsum</p>')
# Cleaning a field returns a list with only paragraph inside it
self.assertIsInstance(cleaned_value, HTMLPDFField.ElementList)
self.assertEqual(len(cleaned_value), 1)
paragraph = cleaned_value[0]
self.assertIsInstance(paragraph, HTMLPDFField.Paragraph)
# Paragaph should contain a list with a text object
paragraph_value = paragraph.value
self.assertIsInstance(paragraph_value, HTMLPDFField.ElementList)
self.assertEqual(len(paragraph_value), 1)
text = paragraph_value[0]
self.assertIsInstance(text, HTMLPDFField.Text)
# Text object should contain string
text_value = text.value
self.assertIsInstance(text_value, str)
self.assertEqual(text_value, 'Lorem Ipsum')
def test_list(self):
cleaned_value = self.html_field.clean('<ul><li>Item 1</li>'
'<li>Item 2</li></ul>')
# Cleaning a field returns a list with an unordered list
self.assertIsInstance(cleaned_value, HTMLPDFField.ElementList)
self.assertEqual(len(cleaned_value), 1)
unordered_list = cleaned_value[0]
self.assertIsInstance(unordered_list, HTMLPDFField.UnorderedList)
unordered_list_value = unordered_list.value
self.assertIsInstance(unordered_list_value, HTMLPDFField.ElementList)
self.assertEqual(len(unordered_list_value), 2)
for list_item in unordered_list_value:
self.assertIsInstance(list_item, HTMLPDFField.ListItem)
self.assertIsInstance(list_item.value, HTMLPDFField.ElementList)
self.assertEqual(len(list_item.value), 1)
self.assertIsInstance(list_item.value[0], HTMLPDFField.Text)
self.assertIsInstance(list_item.value[0].value, str)
self.assertIn('Item ', list_item.value[0].value)
def test_image(self):
cleaned_value = self.html_field.clean('<img src="image.jpg">')
self.assertEqual(len(cleaned_value), 1)
image = cleaned_value[0]
self.assertIsInstance(image, HTMLPDFField.Image)
self.assertEqual(image.attrs['src'], 'image.jpg')
def test_image_without_src(self):
cleaned_value = self.html_field.clean('<img>')
# If no argument is passed, src will be none
self.assertEqual(len(cleaned_value), 1)
self.assertIsNone(cleaned_value[0].attrs['src'])
def test_image_with_empty_src(self):
cleaned_value = self.html_field.clean('<img src="">')
# If no argument is passed, src will be an empty string
self.assertEqual(len(cleaned_value), 1)
self.assertIsInstance(cleaned_value[0].attrs['src'], str)
self.assertEqual(cleaned_value[0].attrs['src'], '')
def test_ignoring_embeds(self):
cleaned_value = self.html_field.clean('<embed>')
self.assertIsInstance(cleaned_value, HTMLPDFField.ElementList)
self.assertEqual(len(cleaned_value), 0)
def test_ignoring_iframes(self):
cleaned_value = self.html_field.clean('<iframe><p>Test</p></iframe>')
self.assertIsInstance(cleaned_value, HTMLPDFField.ElementList)
self.assertEqual(len(cleaned_value), 0)
def test_ignoring_divs_without_content(self):
cleaned_value = self.html_field.clean('<div></div>')
self.assertIsInstance(cleaned_value, HTMLPDFField.ElementList)
self.assertEqual(len(cleaned_value), 0)
def test_ignoring_divs_with_content(self):
cleaned_value = self.html_field.clean('<div><p>Test</p></div>')
self.assertIsInstance(cleaned_value, HTMLPDFField.ElementList)
self.assertEqual(len(cleaned_value), 1)
self.assertIsInstance(cleaned_value[0], HTMLPDFField.Paragraph)
self.assertIsInstance(cleaned_value[0].value, HTMLPDFField.ElementList)
self.assertIsInstance(cleaned_value[0].value[0], HTMLPDFField.Text)
self.assertEqual(cleaned_value[0].value[0].value, 'Test')
def test_anchor_clean_with_relative_url_without_request(self):
anchor = HTMLPDFField.Anchor(None, attrs={'href': '/data/'})
anchor.clean({})
self.assertEqual(anchor.attrs['href'], '')
def test_anchor_clean_with_absolute_url(self):
anchor = HTMLPDFField.Anchor(None, attrs={'href': 'http://dgg.gg'})
anchor.clean({})
self.assertEqual(anchor.attrs['href'], 'http://dgg.gg')
def test_anchor_with_href(self):
anchor = HTMLPDFField.Anchor(None, attrs={'href': 'http://dgg.gg'})
self.assertEqual(anchor.attrs['href'], 'http://dgg.gg')
def test_anchor_with_empty_href(self):
with self.assertRaises(PDFFieldCleaningError) as cm:
HTMLPDFField.Anchor(None)
exception_message = str(cm.exception)
self.assertIn('href', exception_message)
self.assertIn('HTMLPDFFieldAnchor', exception_message)
def test_append(self):
self.html_list.append(HTMLPDFField.Paragraph(
HTMLPDFField.Text('test')))
self.assertEqual(len(self.html_list), 1)
def test_adding_value_in_init(self):
paragraph = HTMLPDFField.Paragraph(HTMLPDFField.Text('test'))
html_list = HTMLPDFField.ElementList(paragraph)
self.assertEqual(len(html_list), 1)
self.assertEqual(html_list[0], paragraph)
def setUp(self):
self.html_list = HTMLPDFField.ElementList()
def test_image_with_src(self):
image = HTMLPDFField.Image(None,
attrs={'src': 'http://dgg.gg/image.jpg'})
self.assertEqual(image.attrs['src'], 'http://dgg.gg/image.jpg')