tests refactoring

1 files changed, 33 insertions, 0 deletions

diff --git a/juick-server/src/test/java/com/juick/tests/util/MockUtils.java b/juick-server/src/test/java/com/juick/tests/util/MockUtils.java
new file mode 100644
index 00000000..c67b26ef
--- /dev/null
+++ b/juick-server/src/test/java/com/juick/tests/util/MockUtils.java
@@ -0,0 +1,33 @@
+package com.juick.tests.util;
+
+import com.juick.Message;
+import com.juick.User;
+import org.apache.commons.lang3.RandomStringUtils;
+
+import java.util.Date;
+
+/**
+ * Created by vitalyster on 12.01.2017.
+ */
+public class MockUtils {
+    public static Message mockMessage(Integer mid, final User user, final String messageText) {
+        Message msg = new Message();
+
+        msg.setMid(mid);
+        msg.setUser(user);
+        msg.setText(messageText == null ? RandomStringUtils.randomAlphanumeric(24) : messageText);
+        msg.setDate(new Date());
+        return msg;
+    }
+
+    public static User mockUser(final int uid, final String name, final String password) {
+        User user = new User();
+
+        user.setName(name);
+        user.setUid(uid);
+        user.setCredentials(password);
+        user.setBanned(false);
+
+        return user;
+    }
+}