X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Ftest%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Ffcp%2FVerifiers.java;h=ed1a0f3e693658e04e5823b4cb5c8f492ec53f34;hb=a05ce7308e3cc981f97756a374e1a5bb0f34e51d;hp=82f28d9e6f510f539ee84e76b64f30c86be7c2d1;hpb=7662db855611501efe295eece8ab1bca3773722d;p=Sone.git

diff --git a/src/test/java/net/pterodactylus/sone/fcp/Verifiers.java b/src/test/java/net/pterodactylus/sone/fcp/Verifiers.java
index 82f28d9..ed1a0f3 100644
--- a/src/test/java/net/pterodactylus/sone/fcp/Verifiers.java
+++ b/src/test/java/net/pterodactylus/sone/fcp/Verifiers.java
@@ -20,13 +20,18 @@ package net.pterodactylus.sone.fcp;
 import static com.google.common.collect.FluentIterable.from;
 import static java.lang.String.format;
 import static net.pterodactylus.sone.data.Reply.FUTURE_REPLY_FILTER;
+import static net.pterodactylus.sone.template.SoneAccessor.getNiceName;
 import static org.hamcrest.MatcherAssert.assertThat;
 import static org.hamcrest.Matchers.is;
+import static org.hamcrest.Matchers.notNullValue;
 
-import java.util.Collection;
+import java.util.List;
 
 import net.pterodactylus.sone.data.Post;
 import net.pterodactylus.sone.data.PostReply;
+import net.pterodactylus.sone.data.Profile.Field;
+import net.pterodactylus.sone.data.Sone;
+import net.pterodactylus.sone.freenet.fcp.Command.Response;
 
 import freenet.node.FSParseException;
 import freenet.support.SimpleFieldSet;
@@ -40,6 +45,12 @@ import org.hamcrest.CoreMatchers;
  */
 public class Verifiers {
 
+	static void verifyAnswer(Response response, String messageName) {
+		assertThat(response, notNullValue());
+		assertThat(response.getReplyParameters(), notNullValue());
+		assertThat(response.getReplyParameters().get("Message"), is(messageName));
+	}
+
 	static void verifyPost(SimpleFieldSet replyParameters, String prefix, Post post) throws FSParseException {
 		assertThat(replyParameters.get(format("%sID", prefix)), is(post.getId()));
 		assertThat(replyParameters.get(format("%sSone", prefix)), is(post.getSone().getId()));
@@ -48,7 +59,7 @@ public class Verifiers {
 		assertThat(replyParameters.get(format("%sText", prefix)), is(post.getText()));
 	}
 
-	static void verifyPosts(SimpleFieldSet postFieldSet, String prefix, Collection<Post> posts) throws FSParseException {
+	static void verifyPosts(SimpleFieldSet postFieldSet, String prefix, List<Post> posts) throws FSParseException {
 		assertThat(postFieldSet.getInt(prefix + "Count"), CoreMatchers.is(posts.size()));
 		int postIndex = 0;
 		for (Post post : posts) {
@@ -64,7 +75,7 @@ public class Verifiers {
 		assertThat(replyParameters.get(format("%sText", prefix)), is(postReply.getText()));
 	}
 
-	static void verifyPostReplies(SimpleFieldSet postFieldSet, String prefix, Collection<PostReply> postReplies) throws FSParseException {
+	static void verifyPostReplies(SimpleFieldSet postFieldSet, String prefix, List<PostReply> postReplies) throws FSParseException {
 		assertThat(postFieldSet.getInt(prefix + "Count"), CoreMatchers.is(from(postReplies).filter(FUTURE_REPLY_FILTER).size()));
 		int postReplyIndex = 0;
 		for (PostReply postReply : from(postReplies).filter(FUTURE_REPLY_FILTER)) {
@@ -73,7 +84,7 @@ public class Verifiers {
 		}
 	}
 
-	static void verifyPostsWithReplies(SimpleFieldSet postFieldSet, String prefix, Collection<Post> posts) throws FSParseException {
+	static void verifyPostsWithReplies(SimpleFieldSet postFieldSet, String prefix, List<Post> posts) throws FSParseException {
 		assertThat(postFieldSet.getInt(prefix + "Count"), CoreMatchers.is(posts.size()));
 		int postIndex = 0;
 		for (Post post : posts) {
@@ -88,4 +99,34 @@ public class Verifiers {
 		verifyPostReplies(postFieldSet, prefix + "Replies.", post.getReplies());
 	}
 
+	static void verifyFollowedSone(SimpleFieldSet simpleFieldSet, String prefix, Sone sone) throws FSParseException {
+		verifyNotFollowedSone(simpleFieldSet, prefix, sone);
+		assertThat(simpleFieldSet.getBoolean(prefix + "Followed"), is(true));
+	}
+
+	static void verifyNotFollowedSone(SimpleFieldSet simpleFieldSet, String prefix, Sone sone) throws FSParseException {
+		assertThat(simpleFieldSet.get(prefix + "Name"), is(sone.getName()));
+		assertThat(simpleFieldSet.get(prefix + "NiceName"), is(getNiceName(sone)));
+		assertThat(simpleFieldSet.getLong(prefix + "LastUpdated"), is(sone.getTime()));
+		assertThat(simpleFieldSet.getInt(prefix + "Field.Count"), is(sone.getProfile().getFields().size()));
+		int fieldIndex = 0;
+		for (Field field : sone.getProfile().getFields()) {
+			assertThat(simpleFieldSet.get(prefix + "Field." + fieldIndex + ".Name"), is(field.getName()));
+			assertThat(simpleFieldSet.get(prefix + "Field." + fieldIndex + ".Value"), is(field.getValue()));
+			fieldIndex++;
+		}
+	}
+
+	static void verifySones(SimpleFieldSet simpleFieldSet, String prefix, List<Sone> sones) throws FSParseException {
+		assertThat(simpleFieldSet.getInt(prefix + "Count"), is(sones.size()));
+		int soneIndex = 0;
+		for (Sone sone : sones) {
+			assertThat(simpleFieldSet.get(prefix + soneIndex + ".ID"), is(sone.getId()));
+			assertThat(simpleFieldSet.get(prefix + soneIndex + ".Name"), is(sone.getName()));
+			assertThat(simpleFieldSet.get(prefix + soneIndex + ".NiceName"), is(getNiceName(sone)));
+			assertThat(simpleFieldSet.getLong(prefix + soneIndex + ".Time"), is(sone.getTime()));
+			soneIndex++;
+		}
+	}
+
 }