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)));
113 * Converts all given {@link Part}s into a string, validating that the
114 * part’s classes match only the expected classes.
117 * The parts to convert to text
118 * @param validClasses
119 * The valid classes; if no classes are given, all classes are
121 * @return The converted text
123 private static String convertText(Iterable<Part> parts, Class<?>... validClasses) {
124 StringBuilder text = new StringBuilder();
125 for (Part part : parts) {
126 assertThat("Part", part, notNullValue());
127 if (validClasses.length != 0) {
128 assertThat("Part’s class", part.getClass(), isIn(validClasses));
130 if (part instanceof PlainTextPart) {
131 text.append(((PlainTextPart) part).getText());
132 } else if (part instanceof FreenetLinkPart) {
133 FreenetLinkPart freenetLinkPart = (FreenetLinkPart) part;
134 text.append('[').append(freenetLinkPart.getLink()).append('|').append(freenetLinkPart.isTrusted() ? "trusted|" : "").append(freenetLinkPart.getTitle()).append('|').append(freenetLinkPart.getText()).append(']');
135 } else if (part instanceof LinkPart) {
136 LinkPart linkPart = (LinkPart) part;
137 text.append('[').append(linkPart.getLink()).append('|').append(linkPart.getTitle()).append('|').append(linkPart.getText()).append(']');
138 } else if (part instanceof SonePart) {
139 SonePart sonePart = (SonePart) part;
140 text.append("[Sone|").append(sonePart.getSone().getId()).append(']');
143 return text.toString();
147 * Mock Sone provider.
149 * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
151 private static class TestSoneProvider implements SoneProvider {
154 public Function<String, Optional<Sone>> soneLoader() {
155 return new Function<String, Optional<Sone>>() {
157 public Optional<Sone> apply(String soneId) {
158 return getSone(soneId);
167 public Optional<Sone> getSone(final String soneId) {
168 return Optional.<Sone>of(new IdOnlySone(soneId));
175 public Collection<Sone> getSones() {
183 public Collection<Sone> getLocalSones() {
191 public Collection<Sone> getRemoteSones() {