Improve coverage of linked element renderer filter

[Sone.git] / src / test / kotlin / net / pterodactylus / sone / template / LinkedElementRenderFilterTest.kt

diff --git a/src/test/kotlin/net/pterodactylus/sone/template/LinkedElementRenderFilterTest.kt b/src/test/kotlin/net/pterodactylus/sone/template/LinkedElementRenderFilterTest.kt
index 20ee99a..c0c83b3 100644 (file)
--- a/src/test/kotlin/net/pterodactylus/sone/template/LinkedElementRenderFilterTest.kt
+++ b/src/test/kotlin/net/pterodactylus/sone/template/LinkedElementRenderFilterTest.kt
@@ -1,10 +1,11 @@
 package net.pterodactylus.sone.template
 
-import net.pterodactylus.sone.core.LinkedImage
+import net.pterodactylus.sone.core.LinkedElement
 import net.pterodactylus.util.template.HtmlFilter
 import net.pterodactylus.util.template.TemplateContextFactory
 import org.hamcrest.MatcherAssert.assertThat
 import org.hamcrest.Matchers.`is`
+import org.hamcrest.Matchers.nullValue
 import org.jsoup.Jsoup
 import org.junit.Test
 
@@ -22,8 +23,23 @@ class LinkedElementRenderFilterTest {
        private val filter = LinkedElementRenderFilter(templateContextFactory)
 
        @Test
+       fun `filter returns null for objects that are not linked elements`() {
+               assertThat(filter.format(null, Any(), null), nullValue())
+       }
+
+       @Test
+       fun `filter renders loading animation for not loaded elements`() {
+               val html = filter.format(null, LinkedElement("KSK@gpl.png", loading = true), emptyMap<String, Any?>()) as String
+               val spanNode = Jsoup.parseBodyFragment(html).body().child(0)
+               assertThat(spanNode.nodeName(), `is`("span"))
+               assertThat(spanNode.attr("class"), `is`("linked-element"))
+               assertThat(spanNode.attr("title"), `is`("KSK@gpl.png"))
+               assertThat(spanNode.attr("style"), `is`("background-image: url('images/loading-animation.gif')"))
+       }
+
+       @Test
        fun `filter can render linked images`() {
-               val html = filter.format(null, LinkedImage("KSK@gpl.png"), emptyMap<String, Any?>()) as String
+               val html = filter.format(null, LinkedElement("KSK@gpl.png"), emptyMap<String, Any?>()) as String
                val linkNode = Jsoup.parseBodyFragment(html).body().child(0)
                assertThat(linkNode.nodeName(), `is`("a"))
                assertThat(linkNode.attr("href"), `is`("/KSK@gpl.png"))