Tests for SnippetEmbedHandler

tests/test_snippet_handlers.py

@@ -1,5 +1,5 @@
 import pytest
-from wagtail_draftail_snippet.richtext import SnippetLinkHandler
+from wagtail_draftail_snippet.richtext import SnippetLinkHandler, SnippetEmbedHandler
 
 
 class TestSnippetLinkHandler:
@@ -26,3 +26,29 @@ class TestSnippetLinkHandler:
         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"
+
+    @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 == ""
+
+    @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>'

tests/testapp/templates/testapp/advert_snippet_embed.html

@@ -0,0 +1 @@
+<div>{{ object.text }}</div>