Add test for not cancelling on HTML MIME type

[Sone.git] / src / test / kotlin / net / pterodactylus / sone / core / DefaultElementLoaderTest.kt

diff --git a/src/test/kotlin/net/pterodactylus/sone/core/DefaultElementLoaderTest.kt b/src/test/kotlin/net/pterodactylus/sone/core/DefaultElementLoaderTest.kt
index ed66dd9..d31900d 100644 (file)
--- a/src/test/kotlin/net/pterodactylus/sone/core/DefaultElementLoaderTest.kt
+++ b/src/test/kotlin/net/pterodactylus/sone/core/DefaultElementLoaderTest.kt
@@ -71,27 +71,34 @@ class DefaultElementLoaderTest {
        }
 
        @Test
-       fun `element loader does  cancel on audio mime type`() {
+       fun `element loader does cancel on audio mime type`() {
                elementLoader.loadElement(IMAGE_ID)
                verify(freenetInterface).startFetch(eq(freenetURI), callback.capture())
                assertThat(callback.value.shouldCancel(freenetURI, "audio/mpeg", sizeOkay), `is`(true))
        }
 
        @Test
-       fun `element loader does not cancel on video mime type`() {
+       fun `element loader does cancel on video mime type`() {
                elementLoader.loadElement(IMAGE_ID)
                verify(freenetInterface).startFetch(eq(freenetURI), callback.capture())
                assertThat(callback.value.shouldCancel(freenetURI, "video/mkv", sizeOkay), `is`(true))
        }
 
        @Test
-       fun `element loader does not cancel on text mime type`() {
+       fun `element loader does cancel on text mime type`() {
                elementLoader.loadElement(IMAGE_ID)
                verify(freenetInterface).startFetch(eq(freenetURI), callback.capture())
                assertThat(callback.value.shouldCancel(freenetURI, "text/plain", sizeOkay), `is`(true))
        }
 
        @Test
+       fun `element loader does not cancel on text html mime type`() {
+               elementLoader.loadElement(IMAGE_ID)
+               verify(freenetInterface).startFetch(eq(freenetURI), callback.capture())
+               assertThat(callback.value.shouldCancel(freenetURI, "text/html", sizeOkay), `is`(false))
+       }
+
+       @Test
        fun `image loader can load image`() {
                elementLoader.loadElement(decomposedKey)
                verify(freenetInterface).startFetch(eq(FreenetURI(decomposedKey)), callback.capture())