23 lines
749 B
Python
23 lines
749 B
Python
|
import pytest
|
||
|
|
|
||
|
|
from tests.testapp import factories
|
||
|
|
from wagtail_draftail_snippet.richtext import SnippetLinkHandler
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.django_db
|
||
|
|
def test_snippet_link_handler():
|
||
|
|
advert = factories.AdvertFactory(text='advert', url='https://www.example.com')
|
||
|
|
advert.save()
|
||
|
|
|
||
|
|
assert advert.text == 'advert'
|
||
|
|
assert advert.url == 'https://www.example.com'
|
||
|
|
|
||
|
|
# Empty link created in case of exception
|
||
|
|
result = SnippetLinkHandler.expand_db_attributes({'id': 0})
|
||
|
|
assert result == "<a>"
|
||
|
|
|
||
|
|
# Test snippet template render correctly
|
||
|
|
attrs = {'id': 1, 'data-app-name': 'testapp', 'data-model-name': 'Advert'}
|
||
|
|
result = SnippetLinkHandler.expand_db_attributes(attrs)
|
||
|
|
assert result == f'<a href="{ advert.url }/{ advert.id }">'
|