Revision 60423
Added by Claudio Atzori about 3 years ago
| DnetMessageManagerTest.java | ||
|---|---|---|
| 3 | 3 |
import static org.junit.Assert.assertEquals; |
| 4 | 4 |
import static org.junit.Assert.assertTrue; |
| 5 | 5 |
|
| 6 |
import eu.dnetlib.dhp.message.MessageType; |
|
| 6 | 7 |
import org.junit.Before; |
| 7 | 8 |
import org.junit.Test; |
| 8 | 9 |
|
| ... | ... | |
| 24 | 25 |
@Test |
| 25 | 26 |
public void testRegisterMessage_multi_wfs() throws MSROException {
|
| 26 | 27 |
for (long i = 0; i < MAX_N_MESSAGES; i++) {
|
| 27 |
manager.registerMessage(prepareMessage("wf-" + i));
|
|
| 28 |
assertTrue(manager.getMessages().size() <= MAX_N_MESSAGES); |
|
| 28 |
manager.registerMessage(new Message(MessageType.ONGOING, "wf-" + i));
|
|
| 29 |
assertTrue(manager.getOngoingMessages().size() <= MAX_N_MESSAGES);
|
|
| 29 | 30 |
} |
| 30 | 31 |
|
| 31 | 32 |
for (long i = MAX_N_MESSAGES; i < 200; i++) {
|
| 32 |
manager.registerMessage(prepareMessage("wf-" + i));
|
|
| 33 |
assertEquals(MAX_N_MESSAGES, manager.getMessages().size()); |
|
| 33 |
manager.registerMessage(new Message(MessageType.ONGOING, "wf-" + i));
|
|
| 34 |
assertEquals(MAX_N_MESSAGES, manager.getOngoingMessages().size());
|
|
| 34 | 35 |
} |
| 35 | 36 |
} |
| 36 | 37 |
|
| 37 | 38 |
@Test |
| 38 | 39 |
public void testRegisterMessage_single_wf() throws MSROException {
|
| 39 | 40 |
for (long i = 0; i < 100; i++) {
|
| 40 |
manager.registerMessage(prepareMessage("wf-xxxx"));
|
|
| 41 |
assertEquals(1, manager.getMessages().size()); |
|
| 41 |
manager.registerMessage(new Message(MessageType.ONGOING, "wf-xxxx"));
|
|
| 42 |
assertEquals(1, manager.getOngoingMessages().size());
|
|
| 42 | 43 |
} |
| 43 | 44 |
} |
| 44 | 45 |
|
| 45 |
private Message prepareMessage(final String wfId) {
|
|
| 46 |
final Message msg = new Message(); |
|
| 47 |
msg.setWorkflowId(wfId); |
|
| 48 |
return msg; |
|
| 49 |
} |
|
| 50 |
|
|
| 51 | 46 |
} |
Also available in: Unified diff
updated hadoop aggregation workflow reporting