Fix SATW parser, add test

[rhynodge.git] / src / main / java / net / pterodactylus / rhynodge / filters / comics / ScandinaviaAndTheWorldComicFilter.java

diff --git a/src/main/java/net/pterodactylus/rhynodge/filters/comics/ScandinaviaAndTheWorldComicFilter.java b/src/main/java/net/pterodactylus/rhynodge/filters/comics/ScandinaviaAndTheWorldComicFilter.java
index 23590cb..2a32f6a 100644 (file)
--- a/src/main/java/net/pterodactylus/rhynodge/filters/comics/ScandinaviaAndTheWorldComicFilter.java
+++ b/src/main/java/net/pterodactylus/rhynodge/filters/comics/ScandinaviaAndTheWorldComicFilter.java
@@ -22,6 +22,7 @@ import com.google.common.base.Optional;
 import net.pterodactylus.rhynodge.filters.ComicSiteFilter;
 import org.jsoup.nodes.Document;
 import org.jsoup.nodes.Element;
+import org.jsoup.select.Elements;
 
 import java.util.List;
 
@@ -39,19 +40,18 @@ public class ScandinaviaAndTheWorldComicFilter extends ComicSiteFilter {
        // COMICSITEFILTER METHODS
        //
 
+       private Elements findImageElements(Document document) {
+               return document.select("img[itemprop=image]");
+       }
+
        @Override
        protected Optional<String> extractTitle(Document document) {
-               return Optional.of(document.select(".comicmid img").attr("title"));
+               return Optional.of(findImageElements(document).attr("title"));
        }
 
        @Override
        protected List<String> extractImageUrls(Document document) {
-               return from(document.select(".comicmid img")).transform(new Function<Element, String>() {
-                       @Override
-                       public String apply(Element element) {
-                               return element.attr("src");
-                       }
-               }).toList();
+               return from(findImageElements(document)).transform(element -> element.attr("src")).toList();
        }
 
        @Override