2 * Sone - SoneTextParserTest.java - Copyright © 2011 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 java.io.IOException;
21 import java.io.StringReader;
23 import junit.framework.TestCase;
26 * JUnit test case for {@link SoneTextParser}.
28 * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
30 public class SoneTextParserTest extends TestCase {
33 * Tests basic plain-text operation of the parser.
36 * if an I/O error occurs
38 public void testPlainText() throws IOException {
39 SoneTextParser soneTextParser = new SoneTextParser(null, null);
42 /* check basic operation. */
43 parts = soneTextParser.parse(null, new StringReader("Test."));
44 assertNotNull("Parts", parts);
45 assertEquals("Part Text", "Test.", convertText(parts, PlainTextPart.class));
47 /* check empty lines at start and end. */
48 parts = soneTextParser.parse(null, new StringReader("\nTest.\n\n"));
49 assertNotNull("Parts", parts);
50 assertEquals("Part Text", "Test.", convertText(parts, PlainTextPart.class));
52 /* check duplicate empty lines in the text. */
53 parts = soneTextParser.parse(null, new StringReader("\nTest.\n\n\nTest."));
54 assertNotNull("Parts", parts);
55 assertEquals("Part Text", "Test.\n\nTest.", convertText(parts, PlainTextPart.class));
63 * Converts all given {@link Part}s into a string, validating that the
64 * part’s classes match only the expected classes.
67 * The parts to convert to text
69 * The valid classes; if no classes are given, all classes are
71 * @return The converted text
73 private String convertText(Iterable<Part> parts, Class<?>... validClasses) {
74 StringBuilder text = new StringBuilder();
75 for (Part part : parts) {
76 assertNotNull("Part", part);
77 boolean classValid = validClasses.length == 0;
78 for (Class<?> validClass : validClasses) {
79 if (validClass.isAssignableFrom(part.getClass())) {
85 assertEquals("Part’s Class", null, part.getClass());
87 if (part instanceof PlainTextPart) {
88 text.append(((PlainTextPart) part).getText());
89 } else if (part instanceof FreenetLinkPart) {
90 FreenetLinkPart freenetLinkPart = (FreenetLinkPart) part;
91 text.append('[').append(freenetLinkPart.getLink()).append('|').append(freenetLinkPart.isTrusted() ? "trusted|" : "").append(freenetLinkPart.getText()).append(']');
92 } else if (part instanceof LinkPart) {
93 LinkPart linkPart = (LinkPart) part;
94 text.append('[').append(linkPart.getLink()).append('|').append(linkPart.getText()).append(']');
97 return text.toString();