/
SingleMessageListener.java
35 lines (28 loc) · 1.04 KB
/
SingleMessageListener.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
35
package com.waltersson.sniper;
import static org.fest.assertions.Assertions.assertThat;
import static org.junit.Assert.assertThat;
import java.util.concurrent.ArrayBlockingQueue;
import java.util.concurrent.TimeUnit;
import org.hamcrest.Matcher;
import org.jivesoftware.smack.Chat;
import org.jivesoftware.smack.MessageListener;
import org.jivesoftware.smack.packet.Message;
public class SingleMessageListener implements MessageListener
{
private final ArrayBlockingQueue<Message> messages = new ArrayBlockingQueue<>(1);
@Override
public void processMessage(Chat chat, Message message)
{
messages.add(message);
}
public void receivesAMessage() throws InterruptedException
{
assertThat(messages.poll(5, TimeUnit.SECONDS)).isNotNull();
}
public void receivesAMessage(Matcher<? super String> messageMatcher) throws InterruptedException
{
final Message message = messages.poll(5, TimeUnit.SECONDS);
assertThat(message).isNotNull();
assertThat(message.getBody(), messageMatcher);
}
}