2 * Sone - SoneTextParserTest.java - Copyright © 2011–2016 David Roden
4 * This program is free software: you can redistribute it and/or modify
5 * it under the terms of the GNU General Public License as published by
6 * the Free Software Foundation, either version 3 of the License, or
7 * (at your option) any later version.
9 * This program is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 * GNU General Public License for more details.
14 * You should have received a copy of the GNU General Public License
15 * along with this program. If not, see <http://www.gnu.org/licenses/>.
18 package net.pterodactylus.sone.text;
20 import static org.hamcrest.MatcherAssert.assertThat;
21 import static org.hamcrest.Matchers.is;
22 import static org.hamcrest.Matchers.isIn;
23 import static org.hamcrest.Matchers.notNullValue;
25 import java.io.IOException;
26 import java.util.Collection;
28 import net.pterodactylus.sone.data.Sone;
29 import net.pterodactylus.sone.data.impl.IdOnlySone;
30 import net.pterodactylus.sone.database.SoneProvider;
32 import com.google.common.base.Function;
33 import com.google.common.base.Optional;
34 import org.junit.Test;
37 * JUnit test case for {@link SoneTextParser}.
39 * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
41 public class SoneTextParserTest {
43 @SuppressWarnings("static-method")
45 public void testPlainText() throws IOException {
46 SoneTextParser soneTextParser = new SoneTextParser(null, null);
49 /* check basic operation. */
50 parts = soneTextParser.parse("Test.", null);
51 assertThat("Parts", parts, notNullValue());
52 assertThat("Part Text", "Test.", is(convertText(parts, PlainTextPart.class)));
54 /* check empty lines at start and end. */
55 parts = soneTextParser.parse("\nTest.\n\n", null);
56 assertThat("Parts", parts, notNullValue());
57 assertThat("Part Text", "Test.", is(convertText(parts, PlainTextPart.class)));
59 /* check duplicate empty lines in the text. */
60 parts = soneTextParser.parse("\nTest.\n\n\nTest.", null);
61 assertThat("Parts", parts, notNullValue());
62 assertThat("Part Text", "Test.\n\nTest.", is(convertText(parts, PlainTextPart.class)));
65 @SuppressWarnings("static-method")
67 public void testKSKLinks() throws IOException {
68 SoneTextParser soneTextParser = new SoneTextParser(null, null);
71 /* check basic links. */
72 parts = soneTextParser.parse("KSK@gpl.txt", null);
73 assertThat("Parts", parts, notNullValue());
74 assertThat("Part Text", "[KSK@gpl.txt|gpl.txt|gpl.txt]", is(convertText(parts, FreenetLinkPart.class)));
76 /* check embedded links. */
77 parts = soneTextParser.parse("Link is KSK@gpl.txt\u200b.", null);
78 assertThat("Parts", parts, notNullValue());
79 assertThat("Part Text", "Link is [KSK@gpl.txt|gpl.txt|gpl.txt]\u200b.", is(convertText(parts, PlainTextPart.class, FreenetLinkPart.class)));
81 /* check embedded links and line breaks. */
82 parts = soneTextParser.parse("Link is KSK@gpl.txt\nKSK@test.dat\n", null);
83 assertThat("Parts", parts, notNullValue());
84 assertThat("Part Text", "Link is [KSK@gpl.txt|gpl.txt|gpl.txt]\n[KSK@test.dat|test.dat|test.dat]", is(convertText(parts, PlainTextPart.class, FreenetLinkPart.class)));
87 @SuppressWarnings({ "synthetic-access", "static-method" })
89 public void testEmptyLinesAndSoneLinks() throws IOException {
90 SoneTextParser soneTextParser = new SoneTextParser(new TestSoneProvider(), null);
93 /* check basic links. */
94 parts = soneTextParser.parse("Some text.\n\nLink to sone://DAxKQzS48mtaQc7sUVHIgx3fnWZPQBz0EueBreUVWrU and stuff.", null);
95 assertThat("Parts", parts, notNullValue());
96 assertThat("Part Text", "Some text.\n\nLink to [Sone|DAxKQzS48mtaQc7sUVHIgx3fnWZPQBz0EueBreUVWrU] and stuff.", is(convertText(parts, PlainTextPart.class, SonePart.class)));
99 @SuppressWarnings({ "synthetic-access", "static-method" })
101 public void testEmpyHttpLinks() throws IOException {
102 SoneTextParser soneTextParser = new SoneTextParser(new TestSoneProvider(), null);
103 Iterable<Part> parts;
105 /* check empty http links. */
106 parts = soneTextParser.parse("Some text. Empty link: http:// – nice!", null);
107 assertThat("Parts", parts, notNullValue());
108 assertThat("Part Text", "Some text. Empty link: http:// – nice!", is(convertText(parts, PlainTextPart.class)));
112 public void httpLinkWithoutParensEndsAtNextClosingParen() {
113 SoneTextParser soneTextParser = new SoneTextParser(null, null);
114 Iterable<Part> parts = soneTextParser.parse("Some text (and a link: http://example.sone/abc) – nice!", null);
115 assertThat("Parts", parts, notNullValue());
116 assertThat("Part Text", "Some text (and a link: [http://example.sone/abc|example.sone/abc|example.sone/abc]) – nice!", is(convertText(parts, PlainTextPart.class, LinkPart.class)));
120 public void httpLinkWithOpenedAndClosedParensEndsAtNextClosingParen() {
121 SoneTextParser soneTextParser = new SoneTextParser(null, null);
122 Iterable<Part> parts = soneTextParser.parse("Some text (and a link: http://example.sone/abc_(def)) – nice!", null);
123 assertThat("Parts", parts, notNullValue());
124 assertThat("Part Text", "Some text (and a link: [http://example.sone/abc_(def)|example.sone/abc_(def)|example.sone/abc_(def)]) – nice!", is(convertText(parts, PlainTextPart.class, LinkPart.class)));
128 public void punctuationIsIgnoredAtEndOfLinkBeforeWhitespace() {
129 SoneTextParser soneTextParser = new SoneTextParser(null, null);
130 Iterable<Part> parts = soneTextParser.parse("Some text and a link: http://example.sone/abc. Nice!", null);
131 assertThat("Parts", parts, notNullValue());
132 assertThat("Part Text", "Some text and a link: [http://example.sone/abc|example.sone/abc|example.sone/abc]. Nice!", is(convertText(parts, PlainTextPart.class, LinkPart.class)));
136 public void multiplePunctuationCharactersAreIgnoredAtEndOfLinkBeforeWhitespace() {
137 SoneTextParser soneTextParser = new SoneTextParser(null, null);
138 Iterable<Part> parts = soneTextParser.parse("Some text and a link: http://example.sone/abc... Nice!", null);
139 assertThat("Parts", parts, notNullValue());
140 assertThat("Part Text", "Some text and a link: [http://example.sone/abc|example.sone/abc|example.sone/abc]... Nice!", is(convertText(parts, PlainTextPart.class, LinkPart.class)));
144 * Converts all given {@link Part}s into a string, validating that the
145 * part’s classes match only the expected classes.
148 * The parts to convert to text
149 * @param validClasses
150 * The valid classes; if no classes are given, all classes are
152 * @return The converted text
154 private static String convertText(Iterable<Part> parts, Class<?>... validClasses) {
155 StringBuilder text = new StringBuilder();
156 for (Part part : parts) {
157 assertThat("Part", part, notNullValue());
158 if (validClasses.length != 0) {
159 assertThat("Part’s class", part.getClass(), isIn(validClasses));
161 if (part instanceof PlainTextPart) {
162 text.append(((PlainTextPart) part).getText());
163 } else if (part instanceof FreenetLinkPart) {
164 FreenetLinkPart freenetLinkPart = (FreenetLinkPart) part;
165 text.append('[').append(freenetLinkPart.getLink()).append('|').append(freenetLinkPart.isTrusted() ? "trusted|" : "").append(freenetLinkPart.getTitle()).append('|').append(freenetLinkPart.getText()).append(']');
166 } else if (part instanceof LinkPart) {
167 LinkPart linkPart = (LinkPart) part;
168 text.append('[').append(linkPart.getLink()).append('|').append(linkPart.getTitle()).append('|').append(linkPart.getText()).append(']');
169 } else if (part instanceof SonePart) {
170 SonePart sonePart = (SonePart) part;
171 text.append("[Sone|").append(sonePart.getSone().getId()).append(']');
174 return text.toString();
178 * Mock Sone provider.
180 * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
182 private static class TestSoneProvider implements SoneProvider {
185 public Function<String, Optional<Sone>> soneLoader() {
186 return new Function<String, Optional<Sone>>() {
188 public Optional<Sone> apply(String soneId) {
189 return getSone(soneId);
198 public Optional<Sone> getSone(final String soneId) {
199 return Optional.<Sone>of(new IdOnlySone(soneId));
206 public Collection<Sone> getSones() {
214 public Collection<Sone> getLocalSones() {
222 public Collection<Sone> getRemoteSones() {