juick-server-jdbc/src/test/java/com/juick/tests/util/MockUtils.java


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34

package com.juick.tests.util;

import com.juick.Message;
import com.juick.User;
import org.apache.commons.text.RandomStringGenerator;

import java.util.Date;

/**
 * Created by vitalyster on 12.01.2017.
 */
public class MockUtils {
    final static RandomStringGenerator generator = new RandomStringGenerator.Builder().withinRange('a', 'z').build();
    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 ? generator.generate(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;
    }
}