X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Ftest%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Ftext%2FSoneTextParserTest.java;h=25ec62a2c5f4ef8a0f8c30dcf6df3731304d448f;hb=52c5e4136f8251b504941d662dba7b2ffc33863d;hp=f0b0e1bd9464549fb23eb3d2f9530bd99c5b67e9;hpb=6b9c7c806fd1fc168e598d6e4eb6ee46ad2aa6fd;p=Sone.git
diff --git a/src/test/java/net/pterodactylus/sone/text/SoneTextParserTest.java b/src/test/java/net/pterodactylus/sone/text/SoneTextParserTest.java
index f0b0e1b..25ec62a 100644
--- a/src/test/java/net/pterodactylus/sone/text/SoneTextParserTest.java
+++ b/src/test/java/net/pterodactylus/sone/text/SoneTextParserTest.java
@@ -1,5 +1,5 @@
/*
- * Sone - SoneTextParserTest.java - Copyright © 2011 David Roden
+ * Sone - SoneTextParserTest.java - Copyright © 2011â2013 David Roden
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
@@ -17,17 +17,44 @@
package net.pterodactylus.sone.text;
+import static org.hamcrest.MatcherAssert.assertThat;
+import static org.hamcrest.Matchers.is;
+import static org.hamcrest.Matchers.notNullValue;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.when;
+
import java.io.IOException;
import java.io.StringReader;
+import java.util.Arrays;
+import java.util.Collection;
+import java.util.Collections;
+import java.util.HashMap;
+import java.util.HashSet;
+import java.util.Map;
+import java.util.Set;
+
+import net.pterodactylus.sone.data.Post;
+import net.pterodactylus.sone.data.Sone;
+import net.pterodactylus.sone.data.impl.IdOnlySone;
+import net.pterodactylus.sone.database.PostProvider;
+import net.pterodactylus.sone.database.SoneProvider;
-import junit.framework.TestCase;
+import com.google.common.base.Function;
+import com.google.common.base.Optional;
+import org.junit.Test;
+import org.mockito.Mockito;
+import org.mockito.stubbing.OngoingStubbing;
/**
* JUnit test case for {@link SoneTextParser}.
*
* @author David âBombeâ Roden
*/
-public class SoneTextParserTest extends TestCase {
+public class SoneTextParserTest {
+
+ private final SoneProvider soneProvider = new TestSoneProvider();
+ private final TestPostProvider postProvider = new TestPostProvider();
+ private final SoneTextParser soneTextParser = new SoneTextParser(soneProvider, postProvider);
/**
* Tests basic plain-text operation of the parser.
@@ -35,40 +62,233 @@ public class SoneTextParserTest extends TestCase {
* @throws IOException
* if an I/O error occurs
*/
+ @Test
public void testPlainText() throws IOException {
- SoneTextParser soneTextParser = new SoneTextParser(null, null);
Iterable parts;
- StringBuilder text = new StringBuilder();
/* check basic operation. */
- text.setLength(0);
parts = soneTextParser.parse(null, new StringReader("Test."));
- assertNotNull("Parts", parts);
- for (Part part : parts) {
- assertTrue("Part is PlainTextPart", part instanceof PlainTextPart);
- text.append(((PlainTextPart) part).getText());
- }
- assertEquals("Part Text", "Test.", text.toString());
+ assertThat(parts, notNullValue());
+ assertThat(convertText(parts, PlainTextPart.class), is("Test."));
/* check empty lines at start and end. */
- text.setLength(0);
parts = soneTextParser.parse(null, new StringReader("\nTest.\n\n"));
- assertNotNull("Parts", parts);
- for (Part part : parts) {
- assertTrue("Part is PlainTextPart", part instanceof PlainTextPart);
- text.append(((PlainTextPart) part).getText());
- }
- assertEquals("Part Text", "Test.", text.toString());
+ assertThat(parts, notNullValue());
+ assertThat(convertText(parts, PlainTextPart.class), is("Test."));
/* check duplicate empty lines in the text. */
- text.setLength(0);
parts = soneTextParser.parse(null, new StringReader("\nTest.\n\n\nTest."));
- assertNotNull("Parts", parts);
+ assertThat(parts, notNullValue());
+ assertThat(convertText(parts, PlainTextPart.class), is("Test.\n\nTest."));
+ }
+
+ /**
+ * Tests parsing of KSK links.
+ *
+ * @throws IOException
+ * if an I/O error occurs
+ */
+ @Test
+ public void testKSKLinks() throws IOException {
+ Iterable parts;
+
+ /* check basic links. */
+ parts = soneTextParser.parse(null, new StringReader("KSK@gpl.txt"));
+ assertThat(parts, notNullValue());
+ assertThat(convertText(parts, FreenetLinkPart.class), is("[KSK@gpl.txt|gpl.txt|gpl.txt]"));
+
+ /* check embedded links. */
+ parts = soneTextParser.parse(null, new StringReader("Link is KSK@gpl.txt\u200b."));
+ assertThat(parts, notNullValue());
+ assertThat(convertText(parts, PlainTextPart.class, FreenetLinkPart.class), is("Link is [KSK@gpl.txt|gpl.txt|gpl.txt]\u200b."));
+
+ /* check embedded links and line breaks. */
+ parts = soneTextParser.parse(null, new StringReader("Link is KSK@gpl.txt\nKSK@test.dat\n"));
+ assertThat(parts, notNullValue());
+ assertThat(convertText(parts, PlainTextPart.class, FreenetLinkPart.class),
+ is("Link is [KSK@gpl.txt|gpl.txt|gpl.txt]\n[KSK@test.dat|test.dat|test.dat]"));
+ }
+
+ /**
+ * Test case for a bug that was discovered in 0.6.7.
+ *
+ * @throws IOException
+ * if an I/O error occurs
+ */
+ @Test
+ public void testEmptyLinesAndSoneLinks() throws IOException {
+ Iterable parts;
+
+ /* check basic links. */
+ parts = soneTextParser.parse(null, new StringReader("Some text.\n\nLink to sone://DAxKQzS48mtaQc7sUVHIgx3fnWZPQBz0EueBreUVWrU and stuff."));
+ assertThat(parts, notNullValue());
+ assertThat(convertText(parts, PlainTextPart.class, SonePart.class),
+ is("Some text.\n\nLink to [Sone|DAxKQzS48mtaQc7sUVHIgx3fnWZPQBz0EueBreUVWrU] and stuff."));
+ }
+
+ /**
+ * Test for a bug discovered in Sone 0.8.4 where a plain âhttp://â would be
+ * parsed into a link.
+ *
+ * @throws IOException
+ * if an I/O error occurs
+ */
+ @Test
+ public void testEmpyHttpLinks() throws IOException {
+ Iterable parts;
+
+ /* check empty http links. */
+ parts = soneTextParser.parse(null, new StringReader("Some text. Empty link: http:// â nice!"));
+ assertThat(parts, notNullValue());
+ assertThat(convertText(parts, PlainTextPart.class), is("Some text. Empty link: http:// â nice!"));
+ }
+
+ @Test
+ public void linksToPostAreParsedCorrectly() throws IOException {
+ postProvider.addValidPostId("foo", "Post about foo...");
+ Iterable parts = soneTextParser.parse(null, new StringReader("This post://foo is awesome."));
+ assertThat(convertText(parts, PlainTextPart.class, PostPart.class), is("This [post|foo|Post about foo...] is awesome."));
+ }
+
+ //
+ // PRIVATE METHODS
+ //
+
+ /**
+ * Converts all given {@link Part}s into a string, validating that the
+ * partâs classes match only the expected classes.
+ *
+ * @param parts
+ * The parts to convert to text
+ * @param validClasses
+ * The valid classes; if no classes are given, all classes are
+ * valid
+ * @return The converted text
+ */
+ private static String convertText(Iterable parts, Class>... validClasses) {
+ StringBuilder text = new StringBuilder();
for (Part part : parts) {
- assertTrue("Part is PlainTextPart", part instanceof PlainTextPart);
- text.append(((PlainTextPart) part).getText());
+ assertThat(part, notNullValue());
+ boolean classValid = validClasses.length == 0;
+ for (Class> validClass : validClasses) {
+ if (validClass.isAssignableFrom(part.getClass())) {
+ classValid = true;
+ break;
+ }
+ }
+ assertThat("Partâs Class (" + part.getClass() + ") is not one of " + Arrays.toString(validClasses), classValid, is(true));
+ if (part instanceof PlainTextPart) {
+ text.append(part.getText());
+ } else if (part instanceof FreenetLinkPart) {
+ FreenetLinkPart freenetLinkPart = (FreenetLinkPart) part;
+ text.append('[')
+ .append(freenetLinkPart.getLink())
+ .append('|')
+ .append(freenetLinkPart.isTrusted() ? "trusted|" : "")
+ .append(freenetLinkPart.getTitle())
+ .append('|')
+ .append(freenetLinkPart.getText())
+ .append(']');
+ } else if (part instanceof LinkPart) {
+ LinkPart linkPart = (LinkPart) part;
+ text.append('[')
+ .append(linkPart.getLink())
+ .append('|')
+ .append(linkPart.getTitle())
+ .append('|')
+ .append(linkPart.getText())
+ .append(']');
+ } else if (part instanceof SonePart) {
+ SonePart sonePart = (SonePart) part;
+ text.append("[Sone|").append(sonePart.getSone().getId()).append(']');
+ } else if (part instanceof PostPart) {
+ PostPart postPart = (PostPart) part;
+ text.append("[post|").append(postPart.getPost().getId()).append('|').append(postPart.getPost().getText()).append(']');
+ }
}
- assertEquals("Part Text", "Test.\n\nTest.", text.toString());
+ return text.toString();
+ }
+
+ /**
+ * Mock Sone provider.
+ *
+ * @author David âBombeâ Roden
+ */
+ private static class TestSoneProvider implements SoneProvider {
+
+ @Override
+ public Function> soneLoader() {
+ return new Function>() {
+ @Override
+ public Optional apply(String soneId) {
+ return getSone(soneId);
+ }
+ };
+ }
+
+ /**
+ * {@inheritDoc}
+ */
+ @Override
+ public Optional getSone(final String soneId) {
+ return Optional.of(new IdOnlySone(soneId));
+ }
+
+ /**
+ * {@inheritDocs}
+ */
+ @Override
+ public Collection getSones() {
+ return null;
+ }
+
+ /**
+ * {@inheritDocs}
+ */
+ @Override
+ public Collection getLocalSones() {
+ return null;
+ }
+
+ /**
+ * {@inheritDocs}
+ */
+ @Override
+ public Collection getRemoteSones() {
+ return null;
+ }
+
+ }
+
+ private static class TestPostProvider implements PostProvider {
+
+ private final Map posts = new HashMap();
+
+ private void addValidPostId(String validPostId, String text) {
+ posts.put(validPostId, text);
+ }
+
+ @Override
+ public Collection getDirectedPosts(String recipientId) {
+ return Collections.emptyList();
+ }
+
+ @Override
+ public Collection getPosts(String soneId) {
+ return Collections.emptyList();
+ }
+
+ @Override
+ public Optional getPost(String postId) {
+ if (posts.containsKey(postId)) {
+ Post post = mock(Post.class);
+ when(post.getId()).thenReturn(postId);
+ when(post.getText()).thenReturn(posts.get(postId));
+ return Optional.of(post);
+ }
+ return Optional.absent();
+ }
+
}
}