X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Ftest%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Ftemplate%2FLinkedElementsFilterTest.kt;h=1f6f6819a1ecebd4fb41408911782ee18fc84ad6;hp=d519e41de00c9c1b69bd64d883c5e8746742cc54;hb=b91ce252d94a8876097b939e129dc33264cef2f5;hpb=1cd71a1c2a94f889d63d6ec3168d76da0a8b116b

diff --git a/src/test/kotlin/net/pterodactylus/sone/template/LinkedElementsFilterTest.kt b/src/test/kotlin/net/pterodactylus/sone/template/LinkedElementsFilterTest.kt
index d519e41..1f6f681 100644
--- a/src/test/kotlin/net/pterodactylus/sone/template/LinkedElementsFilterTest.kt
+++ b/src/test/kotlin/net/pterodactylus/sone/template/LinkedElementsFilterTest.kt
@@ -2,7 +2,6 @@ package net.pterodactylus.sone.template
 
 import net.pterodactylus.sone.core.ElementLoader
 import net.pterodactylus.sone.core.LinkedElement
-import net.pterodactylus.sone.core.LinkedImage
 import net.pterodactylus.sone.test.mock
 import net.pterodactylus.sone.text.FreenetLinkPart
 import net.pterodactylus.sone.text.LinkPart
@@ -26,14 +25,16 @@ class LinkedElementsFilterTest {
 				PlainTextPart("text"),
 				LinkPart("http://link", "link"),
 				FreenetLinkPart("KSK@link", "link", false),
+				FreenetLinkPart("KSK@loading.png", "link", false),
 				FreenetLinkPart("KSK@link.png", "link", false)
 		)
-		`when`(imageLoader.loadElement("KSK@link")).thenReturn(LinkedImage("KSK@link", true))
-		`when`(imageLoader.loadElement("KSK@link.png")).thenReturn(LinkedImage("KSK@link.png"))
+		`when`(imageLoader.loadElement("KSK@link")).thenReturn(LinkedElement("KSK@link", failed = true))
+		`when`(imageLoader.loadElement("KSK@loading.png")).thenReturn(LinkedElement("KSK@loading.png", loading = true))
+		`when`(imageLoader.loadElement("KSK@link.png")).thenReturn(LinkedElement("KSK@link.png"))
 		val loadedImages = filter.format(null, parts, null)
 		assertThat(loadedImages, contains<LinkedElement>(
-				LinkedImage("KSK@link", true),
-				LinkedImage("KSK@link.png")
+				LinkedElement("KSK@loading.png", failed = false, loading = true),
+				LinkedElement("KSK@link.png", failed = false, loading = false)
 		))
 	}