From b64dbd900e6a671ec1a2749f3b7e2b9f0b250657 Mon Sep 17 00:00:00 2001
From: =?utf8?q?David=20=E2=80=98Bombe=E2=80=99=20Roden?=
 <bombe@pterodactylus.net>
Date: Tue, 18 Mar 2014 06:46:16 +0100
Subject: [PATCH] Add unit test for comic site filter.

---
 .../rhynodge/filters/ComicSiteFilterTest.java      | 102 +++++++++++++++++++++
 1 file changed, 102 insertions(+)
 create mode 100644 src/test/java/net/pterodactylus/rhynodge/filters/ComicSiteFilterTest.java

diff --git a/src/test/java/net/pterodactylus/rhynodge/filters/ComicSiteFilterTest.java b/src/test/java/net/pterodactylus/rhynodge/filters/ComicSiteFilterTest.java
new file mode 100644
index 0000000..a449708
--- /dev/null
+++ b/src/test/java/net/pterodactylus/rhynodge/filters/ComicSiteFilterTest.java
@@ -0,0 +1,102 @@
+package net.pterodactylus.rhynodge.filters;
+
+import static com.google.common.base.Optional.fromNullable;
+import static java.util.Arrays.asList;
+import static org.hamcrest.MatcherAssert.assertThat;
+import static org.hamcrest.Matchers.contains;
+import static org.hamcrest.Matchers.hasSize;
+import static org.hamcrest.Matchers.is;
+
+import java.util.Collections;
+import java.util.List;
+
+import net.pterodactylus.rhynodge.State;
+import net.pterodactylus.rhynodge.states.ComicState;
+import net.pterodactylus.rhynodge.states.ComicState.Comic;
+import net.pterodactylus.rhynodge.states.ComicState.Strip;
+import net.pterodactylus.rhynodge.states.FailedState;
+import net.pterodactylus.rhynodge.states.HtmlState;
+
+import com.google.common.base.Optional;
+import org.jsoup.nodes.Document;
+import org.junit.Test;
+
+/**
+ * Unit test for {@link ComicSiteFilter}.
+ *
+ * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
+ */
+public class ComicSiteFilterTest {
+
+	@Test(expected = IllegalArgumentException.class)
+	public void comicSiteFilterRequiresHtmlState() {
+		State state = new FailedState();
+		ComicSiteFilter comicSiteFilter = new TestComicSiteFilter(null, null, null);
+		comicSiteFilter.filter(state);
+	}
+
+	@Test
+	public void htmlWithoutTitleIsNotRecognizedAsComic() {
+		State state = new HtmlState("http://foo/", null);
+		ComicSiteFilter comicSiteFilter = new TestComicSiteFilter(null, null, null);
+		State newState = comicSiteFilter.filter(state);
+		assertThat(newState instanceof ComicState, is(true));
+	}
+
+	@Test
+	public void htmlWithTitleButWithoutImagesIsNotRecognizedAsComic() {
+		State state = new HtmlState("http://foo/", null);
+		ComicSiteFilter comicSiteFilter = new TestComicSiteFilter("Title", Collections.<String>emptyList(), null);
+		State newState = comicSiteFilter.filter(state);
+		assertThat(newState instanceof ComicState, is(true));
+	}
+
+	@Test
+	public void everyUrlGetAStrip() {
+		State state = new HtmlState("http://foo/", null);
+		ComicSiteFilter comicSiteFilter = new TestComicSiteFilter("Title", asList("url1.gif", "url2.gif", "url3.gif"), asList("Comment 1", "Comment 2"));
+		State newState = comicSiteFilter.filter(state);
+		assertThat(newState instanceof ComicState, is(true));
+		ComicState comicState = (ComicState) newState;
+		assertThat(comicState.comics(), hasSize(1));
+		Comic comic = comicState.comics().get(0);
+		assertThat(comic.strips(), contains(new Strip("http://foo/url1.gif", "Comment 1"), new Strip("http://foo/url2.gif", "Comment 2"), new Strip("http://foo/url3.gif", "")));
+	}
+
+	@Test(expected = IllegalStateException.class)
+	public void illegalUrlThrowsException() {
+		State state = new HtmlState("http://foo/^", null);
+		ComicSiteFilter comicSiteFilter = new TestComicSiteFilter("Title", asList("url1.gif"), Collections.<String>emptyList());
+		comicSiteFilter.filter(state);
+	}
+
+	private static class TestComicSiteFilter extends ComicSiteFilter {
+
+		private final String title;
+		private final List<String> urls;
+		private final List<String> comments;
+
+		public TestComicSiteFilter(String title, List<String> urls, List<String> comments) {
+			this.title = title;
+			this.urls = urls;
+			this.comments = comments;
+		}
+
+		@Override
+		protected Optional<String> extractTitle(Document document) {
+			return fromNullable(title);
+		}
+
+		@Override
+		protected List<String> extractImageUrls(Document document) {
+			return urls;
+		}
+
+		@Override
+		protected List<String> extractImageComments(Document document) {
+			return comments;
+		}
+
+	}
+
+}
-- 
2.7.4