src/test/kotlin/net/pterodactylus/rhynodge/filters/comics/QuestionableContentFilterTest.kt

   1 package net.pterodactylus.rhynodge.filters.comics
   2
   3 import net.pterodactylus.rhynodge.filters.ResourceLoader
   4 import net.pterodactylus.rhynodge.states.ComicState
   5 import net.pterodactylus.rhynodge.states.ComicState.Comic
   6 import net.pterodactylus.rhynodge.states.ComicState.Strip
   7 import net.pterodactylus.rhynodge.states.HtmlState
   8 import org.hamcrest.MatcherAssert.assertThat
   9 import org.hamcrest.Matchers.contains
  10 import org.junit.Test
  11
  12 class QuestionableContentFilterTest {
  13
  14         private val filter = QuestionableContentComicFilter()
  15         private val htmlState: HtmlState = ResourceLoader
  16                         .loadDocument(javaClass, "questionable-content.html", baseUrl)
  17                         .let { HtmlState(baseUrl, it) }
  18
  19         @Test
  20         fun `comic is extracted correctly`() {
  21                 val comicState = filter.filter(htmlState) as ComicState
  22                 assertThat(comicState.comics(), contains(
  23                                 Comic("").add(Strip("https://www.questionablecontent.net/comics/4355.png", ""))
  24                 ))
  25         }
  26
  27 }
  28
  29 private const val baseUrl = "https://www.questionablecontent.net/"