Move classes for tests to test package
[Sone.git] / src / test / java / net / pterodactylus / sone / test / TestPostReplyBuilder.java
diff --git a/src/test/java/net/pterodactylus/sone/test/TestPostReplyBuilder.java b/src/test/java/net/pterodactylus/sone/test/TestPostReplyBuilder.java
new file mode 100644 (file)
index 0000000..e09e120
--- /dev/null
@@ -0,0 +1,70 @@
+package net.pterodactylus.sone.test;
+
+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.PostReply;
+import net.pterodactylus.sone.data.Sone;
+import net.pterodactylus.sone.database.PostReplyBuilder;
+
+/**
+ * {@link PostReplyBuilder} that returns a mocked {@link PostReply}.
+ *
+ * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
+ */
+public class TestPostReplyBuilder implements PostReplyBuilder {
+
+       private final PostReply postReply = mock(PostReply.class);
+
+       @Override
+       public PostReplyBuilder to(String postId) {
+               when(postReply.getPostId()).thenReturn(postId);
+               return this;
+       }
+
+       @Override
+       public PostReply build() throws IllegalStateException {
+               return postReply;
+       }
+
+       @Override
+       public PostReplyBuilder randomId() {
+               when(postReply.getId()).thenReturn(randomUUID().toString());
+               return this;
+       }
+
+       @Override
+       public PostReplyBuilder withId(String id) {
+               when(postReply.getId()).thenReturn(id);
+               return this;
+       }
+
+       @Override
+       public PostReplyBuilder from(String senderId) {
+               Sone sone = mock(Sone.class);
+               when(sone.getId()).thenReturn(senderId);
+               when(postReply.getSone()).thenReturn(sone);
+               return this;
+       }
+
+       @Override
+       public PostReplyBuilder currentTime() {
+               when(postReply.getTime()).thenReturn(currentTimeMillis());
+               return this;
+       }
+
+       @Override
+       public PostReplyBuilder withTime(long time) {
+               when(postReply.getTime()).thenReturn(time);
+               return this;
+       }
+
+       @Override
+       public PostReplyBuilder withText(String text) {
+               when(postReply.getText()).thenReturn(text);
+               return this;
+       }
+
+}