D-Net

package eu.dnetlib.msro.openaireplus.workflows.nodes.dhp.message;

import static org.junit.Assert.assertEquals;

import static org.junit.Assert.assertTrue;

import org.junit.Before;

import org.junit.Test;

import eu.dnetlib.dhp.message.Message;

import eu.dnetlib.msro.rmi.MSROException;

public class DnetMessageManagerTest {

	private DnetMessageManager manager;

	private static final int MAX_N_MESSAGES = 10;

	@Before

	public void setUp() throws Exception {

		manager = new DnetMessageManager();

		manager.setMaxNumberOfMessages(MAX_N_MESSAGES);

	}

	@Test

	public void testRegisterMessage_multi_wfs() throws MSROException {

		for (long i = 0; i < MAX_N_MESSAGES; i++) {

			manager.registerMessage(prepareMessage("wf-" + i));

			assertTrue(manager.getMessages().size() <= MAX_N_MESSAGES);

		}

		for (long i = MAX_N_MESSAGES; i < 200; i++) {

			manager.registerMessage(prepareMessage("wf-" + i));

			assertEquals(MAX_N_MESSAGES, manager.getMessages().size());

		}

	}

	@Test

	public void testRegisterMessage_single_wf() throws MSROException {

		for (long i = 0; i < 100; i++) {

			manager.registerMessage(prepareMessage("wf-xxxx"));

			assertEquals(1, manager.getMessages().size());

		}

	}

	private Message prepareMessage(final String wfId) {

		final Message msg = new Message();

		msg.setWorkflowId(wfId);

		return msg;

	}

}