Move test builders to their own top-level classes.

[Sone.git] / src / test / java / net / pterodactylus / sone / TestPostBuilder.java

diff --git a/src/test/java/net/pterodactylus/sone/TestPostBuilder.java b/src/test/java/net/pterodactylus/sone/TestPostBuilder.java
new file mode 100644 (file)
index 0000000..29c4c40
--- /dev/null
+++ b/src/test/java/net/pterodactylus/sone/TestPostBuilder.java
@@ -0,0 +1,78 @@
+package net.pterodactylus.sone;
+
+import static com.google.common.base.Optional.fromNullable;
+import static java.lang.System.currentTimeMillis;
+import static java.util.UUID.randomUUID;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.when;
+
+import net.pterodactylus.sone.data.Post;
+import net.pterodactylus.sone.data.Sone;
+import net.pterodactylus.sone.database.PostBuilder;
+
+/**
+ * {@link PostBuilder} implementation that returns a mocked {@link Post}.
+ *
+ * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
+ */
+public class TestPostBuilder implements PostBuilder {
+
+       private final Post post = mock(Post.class);
+       private String recipientId = null;
+
+       @Override
+       public PostBuilder copyPost(Post post) throws NullPointerException {
+               return this;
+       }
+
+       @Override
+       public PostBuilder from(String senderId) {
+               final Sone sone = mock(Sone.class);
+               when(sone.getId()).thenReturn(senderId);
+               when(post.getSone()).thenReturn(sone);
+               return this;
+       }
+
+       @Override
+       public PostBuilder randomId() {
+               when(post.getId()).thenReturn(randomUUID().toString());
+               return this;
+       }
+
+       @Override
+       public PostBuilder withId(String id) {
+               when(post.getId()).thenReturn(id);
+               return this;
+       }
+
+       @Override
+       public PostBuilder currentTime() {
+               when(post.getTime()).thenReturn(currentTimeMillis());
+               return this;
+       }
+
+       @Override
+       public PostBuilder withTime(long time) {
+               when(post.getTime()).thenReturn(time);
+               return this;
+       }
+
+       @Override
+       public PostBuilder withText(String text) {
+               when(post.getText()).thenReturn(text);
+               return this;
+       }
+
+       @Override
+       public PostBuilder to(String recipientId) {
+               this.recipientId = recipientId;
+               return this;
+       }
+
+       @Override
+       public Post build() throws IllegalStateException {
+               when(post.getRecipientId()).thenReturn(fromNullable(recipientId));
+               return post;
+       }
+
+}