X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Ftest%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2FMatchers.java;h=41eb0dddbc925631f1773b9e8d99b5c44783fab9;hb=ec06ae64c86f0b06bb0cf9f8b289e7907e81dffa;hp=61cee46c1240676dcba8314f5d992363fb2e5c6e;hpb=72f305f1e5986d0badf3991d57693823258d1599;p=Sone.git

diff --git a/src/test/java/net/pterodactylus/sone/Matchers.java b/src/test/java/net/pterodactylus/sone/Matchers.java
index 61cee46..41eb0dd 100644
--- a/src/test/java/net/pterodactylus/sone/Matchers.java
+++ b/src/test/java/net/pterodactylus/sone/Matchers.java
@@ -17,23 +17,18 @@
 
 package net.pterodactylus.sone;
 
-import static com.google.common.base.Objects.equal;
-import static com.google.common.collect.Iterators.size;
-import static java.util.Arrays.asList;
 import static java.util.regex.Pattern.compile;
 
 import java.io.IOException;
 import java.io.InputStream;
-import java.util.Collection;
-import java.util.Iterator;
-import java.util.List;
 
-import freenet.support.SimpleFieldSet;
+import net.pterodactylus.sone.data.Post;
+import net.pterodactylus.sone.data.PostReply;
 
-import com.google.common.base.Objects;
-import com.google.common.collect.Lists;
+import com.google.common.base.Optional;
 import org.hamcrest.Description;
 import org.hamcrest.Matcher;
+import org.hamcrest.TypeSafeDiagnosingMatcher;
 import org.hamcrest.TypeSafeMatcher;
 
 /**
@@ -57,4 +52,172 @@ public class Matchers {
 		};
 	}
 
+	public static Matcher<InputStream> delivers(final byte[] data) {
+		return new TypeSafeMatcher<InputStream>() {
+			byte[] readData = new byte[data.length];
+
+			@Override
+			protected boolean matchesSafely(InputStream inputStream) {
+				int offset = 0;
+				try {
+					while (true) {
+						int r = inputStream.read();
+						if (r == -1) {
+							return offset == data.length;
+						}
+						if (offset == data.length) {
+							return false;
+						}
+						if (data[offset] != (readData[offset] = (byte) r)) {
+							return false;
+						}
+						offset++;
+					}
+				} catch (IOException ioe1) {
+					return false;
+				}
+			}
+
+			@Override
+			public void describeTo(Description description) {
+				description.appendValue(data);
+			}
+
+			@Override
+			protected void describeMismatchSafely(InputStream item,
+					Description mismatchDescription) {
+				mismatchDescription.appendValue(readData);
+			}
+		};
+	}
+
+	public static Matcher<Post> isPost(String postId, long time,
+			String text, Optional<String> recipient) {
+		return new PostMatcher(postId, time, text, recipient);
+	}
+
+	public static Matcher<PostReply> isPostReply(String postReplyId,
+			String postId, long time, String text) {
+		return new PostReplyMatcher(postReplyId, postId, time, text);
+	}
+
+	private static class PostMatcher extends TypeSafeDiagnosingMatcher<Post> {
+
+		private final String postId;
+		private final long time;
+		private final String text;
+		private final Optional<String> recipient;
+
+		private PostMatcher(String postId, long time, String text,
+				Optional<String> recipient) {
+			this.postId = postId;
+			this.time = time;
+			this.text = text;
+			this.recipient = recipient;
+		}
+
+		@Override
+		protected boolean matchesSafely(Post post,
+				Description mismatchDescription) {
+			if (!post.getId().equals(postId)) {
+				mismatchDescription.appendText("ID is not ")
+						.appendValue(postId);
+				return false;
+			}
+			if (post.getTime() != time) {
+				mismatchDescription.appendText("Time is not @")
+						.appendValue(time);
+				return false;
+			}
+			if (!post.getText().equals(text)) {
+				mismatchDescription.appendText("Text is not ")
+						.appendValue(text);
+				return false;
+			}
+			if (recipient.isPresent()) {
+				if (!post.getRecipientId().isPresent()) {
+					mismatchDescription.appendText(
+							"Recipient not present");
+					return false;
+				}
+				if (!post.getRecipientId().get().equals(recipient.get())) {
+					mismatchDescription.appendText("Recipient is not ")
+							.appendValue(recipient.get());
+					return false;
+				}
+			} else {
+				if (post.getRecipientId().isPresent()) {
+					mismatchDescription.appendText("Recipient is present");
+					return false;
+				}
+			}
+			return true;
+		}
+
+		@Override
+		public void describeTo(Description description) {
+			description.appendText("is post with ID ")
+					.appendValue(postId);
+			description.appendText(", created at @").appendValue(time);
+			description.appendText(", text ").appendValue(text);
+			if (recipient.isPresent()) {
+				description.appendText(", directed at ")
+						.appendValue(recipient.get());
+			}
+		}
+
+	}
+
+	private static class PostReplyMatcher
+			extends TypeSafeDiagnosingMatcher<PostReply> {
+
+		private final String postReplyId;
+		private final String postId;
+		private final long time;
+		private final String text;
+
+		private PostReplyMatcher(String postReplyId, String postId, long time,
+				String text) {
+			this.postReplyId = postReplyId;
+			this.postId = postId;
+			this.time = time;
+			this.text = text;
+		}
+
+		@Override
+		protected boolean matchesSafely(PostReply postReply,
+				Description mismatchDescription) {
+			if (!postReply.getId().equals(postReplyId)) {
+				mismatchDescription.appendText("is post reply ")
+						.appendValue(postReply.getId());
+				return false;
+			}
+			if (!postReply.getPostId().equals(postId)) {
+				mismatchDescription.appendText("is reply to ")
+						.appendValue(postReply.getPostId());
+				return false;
+			}
+			if (postReply.getTime() != time) {
+				mismatchDescription.appendText("is created at @").appendValue(
+						postReply.getTime());
+				return false;
+			}
+			if (!postReply.getText().equals(text)) {
+				mismatchDescription.appendText("says ")
+						.appendValue(postReply.getText());
+				return false;
+			}
+			return true;
+		}
+
+		@Override
+		public void describeTo(Description description) {
+			description.appendText("is post reply ").appendValue(postReplyId);
+			description.appendText(", replies to post ").appendValue(postId);
+			description.appendText(", is created at @").appendValue(time);
+			description.appendText(", says ").appendValue(text);
+		}
+
+	}
+
 }