mirror/wagtail-draftail-snippet
Archived
1
Fork 0
Code Activity

Fix tests by using lower case model name

Browse source
This commit is contained in:
Parbhat Puri 2020-04-08 12:12:17 +05:30
parent 74ad4a2511
commit adcbf8bd2e
2 changed files with 22 additions and 23 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

42
tests/test_snippet_handlers.py
View file

@ -23,32 +23,32 @@ class TestSnippetLinkHandler:
Test snippet template renders correctly
"""
attrs = {"id": 1, "data-app-name": "testapp", "data-model-name": "Advert"}
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}">'
# class TestSnippetEmbedHandler:
# @pytest.mark.django_db
# def test_advert_setup_correctly(self, advert):
# assert advert.text == "advert"
# assert advert.url == "https://www.example.com"
class TestSnippetEmbedHandler:
@pytest.mark.django_db
def test_advert_setup_correctly(self, advert):
assert advert.text == "advert"
assert advert.url == "https://www.example.com"
# @pytest.mark.django_db
# def test_snippet_create_empty_block_on_error(self, advert):
# """
# Empty block created in case of exception
# """
@pytest.mark.django_db
def test_snippet_create_empty_block_on_error(self, advert):
"""
Empty block created in case of exception
"""
# result = SnippetEmbedHandler.expand_db_attributes({"id": 0})
# assert result == ""
result = SnippetEmbedHandler.expand_db_attributes({"id": 0})
assert result == ""
# @pytest.mark.django_db
# def test_snippet_embed_handler_renders(self, advert):
# """
# Test snippet template renders correctly
# """
@pytest.mark.django_db
def test_snippet_embed_handler_renders(self, advert):
"""
Test snippet template renders correctly
"""
# attrs = {"id": 1, "data-app-name": "testapp", "data-model-name": "Advert"}
# result = SnippetEmbedHandler.expand_db_attributes(attrs)
# assert result == f'<div>{advert.text}</div>'
attrs = {"id": 1, "data-app-name": "testapp", "data-model-name": "advert"}
result = SnippetEmbedHandler.expand_db_attributes(attrs)
assert result == f'<div>{advert.text}</div>'

3
wagtail_draftail_snippet/richtext.py
View file

@ -32,8 +32,7 @@ class SnippetLinkHandler(LinkHandler):
snippet_obj = cls.get_instance(attrs)
template = cls.get_template(attrs)
return render_to_string(template, {"object": snippet_obj})
except Exception as e:
raise e
except Exception:
return "<a>"