Project

General

Profile

« Previous | Next » 

Revision 60357

Added by Michele Artini about 3 years ago

Removed MessageType from Message

View differences:

DnetMessageManagerTest.java
6 6 
import org.junit.Before;
7 7 
import org.junit.Test;
8 8 

  		




  9
  
  
    
import eu.dnetlib.message.Message;


  		




  10
  
  
    
import eu.dnetlib.message.MessageType;


  		




  
  9
  
    
import eu.dnetlib.dhp.message.Message;


  		




  11
  10
  
    
import eu.dnetlib.msro.rmi.MSROException;


  		




  12
  11
  
    

  		




  13
  12
  
    
public class DnetMessageManagerTest {


  		




  14
  13
  
    

  		




  15
  14
  
    
	private DnetMessageManager manager;


  		




  16
  15
  
    

  		




  
  16
  
    
	private static final int MAX_N_MESSAGES = 10;


  		




  
  17
  
    

  		




  17
  18
  
    
	@Before


  		




  18
  19
  
    
	public void setUp() throws Exception {


  		




  19
  20
  
    
		manager = new DnetMessageManager();


  		




  20
  
  
    
		manager.setMaxNumberOfOnGoingMessages(10);


  		




  21
  
  
    
		manager.setMaxNumberOfReports(100);


  		




  
  21
  
    
		manager.setMaxNumberOfMessages(MAX_N_MESSAGES);


  		




  22
  22
  
    
	}


  		




  23
  23
  
    

  		




  24
  24
  
    
	@Test


  		




  25
  
  
    
	public void testRegisterMessage_ongoing_multi_wfs() throws MSROException {


  		




  26
  
  
    
		for (long i = 0; i < manager.getMaxNumberOfOnGoingMessages(); i++) {


  		




  27
  
  
    
			manager.registerMessage(prepareMessage("wf-" + i, MessageType.ONGOING));


  		




  28
  
  
    
			assertTrue(manager.getOnGonginMessages().size() <= manager.getMaxNumberOfOnGoingMessages());


  		




  
  25
  
    
	public void testRegisterMessage_multi_wfs() throws MSROException {


  		




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


  		




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


  		




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


  		




  29
  29
  
    
		}


  		




  30
  30
  
    

  		




  31
  
  
    
		for (long i = manager.getMaxNumberOfOnGoingMessages(); i < 200; i++) {


  		




  32
  
  
    
			manager.registerMessage(prepareMessage("wf-" + i, MessageType.ONGOING));


  		




  33
  
  
    
			assertEquals(manager.getMaxNumberOfOnGoingMessages(), manager.getOnGonginMessages().size());


  		




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


  		




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


  		




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


  		




  34
  34
  
    
		}


  		




  35
  35
  
    
	}


  		




  36
  36
  
    

  		




  37
  37
  
    
	@Test


  		




  38
  
  
    
	public void testRegisterMessage_ongoing_single_wf() throws MSROException {


  		




  
  38
  
    
	public void testRegisterMessage_single_wf() throws MSROException {


  		




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


  		




  40
  
  
    
			manager.registerMessage(prepareMessage("wf-xxxx", MessageType.ONGOING));


  		




  41
  
  
    
			assertEquals(1, manager.getOnGonginMessages().size());


  		




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


  		




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


  		




  42
  42
  
    
		}


  		




  43
  43
  
    
	}


  		




  44
  44
  
    

  		




  45
  
  
    
	@Test


  		




  46
  
  
    
	public void testRegisterMessage_reports_multi_wf() throws MSROException {


  		




  47
  
  
    
		for (long i = 0; i < manager.getMaxNumberOfReports(); i++) {


  		




  48
  
  
    
			manager.registerMessage(prepareMessage("wf-" + i, MessageType.REPORT));


  		




  49
  
  
    
			assertTrue(manager.getReports().size() <= manager.getMaxNumberOfReports());


  		




  50
  
  
    
		}


  		




  51
  
  
    

  		




  52
  
  
    
		for (long i = manager.getMaxNumberOfReports(); i < 200; i++) {


  		




  53
  
  
    
			manager.registerMessage(prepareMessage("wf-" + i, MessageType.REPORT));


  		




  54
  
  
    
			assertEquals(manager.getMaxNumberOfReports(), manager.getReports().size());


  		




  55
  
  
    
		}


  		




  56
  
  
    
	}


  		




  57
  
  
    

  		




  58
  
  
    
	@Test


  		




  59
  
  
    
	public void testRegisterMessage_reports_single_wf() throws MSROException {


  		




  60
  
  
    
		for (long i = 1; i < 100; i++) {


  		




  61
  
  
    
			manager.registerMessage(prepareMessage("wf-xxxx", MessageType.REPORT));


  		




  62
  
  
    
			assertEquals(1, manager.getReports().size());


  		




  63
  
  
    
			assertEquals(i, manager.getReports("wf-xxxx", false).size());


  		




  64
  
  
    
		}


  		




  65
  
  
    
	}


  		




  66
  
  
    

  		




  67
  
  
    
	@Test


  		




  68
  
  
    
	public void testRegisterMessage_reports_single_wf_delete() throws MSROException {


  		




  69
  
  
    
		for (long i = 1; i < 100; i++) {


  		




  70
  
  
    
			manager.registerMessage(prepareMessage("wf-xxxx", MessageType.REPORT));


  		




  71
  
  
    
			assertEquals(1, manager.getReports().size());


  		




  72
  
  
    
			assertEquals(1, manager.getReports("wf-xxxx", true).size());


  		




  73
  
  
    
			assertEquals(0, manager.getReports().size());


  		




  74
  
  
    
		}


  		




  75
  
  
    
	}


  		




  76
  
  
    

  		




  77
  
  
    
	private Message prepareMessage(final String wfId, final MessageType type) {


  		




  
  45
  
    
	private Message prepareMessage(final String wfId) {


  		




  78
  46
  
    
		final Message msg = new Message();


  		




  79
  47
  
    
		msg.setWorkflowId(wfId);


  		




  80
  
  
    
		msg.setType(type);


  		




  81
  48
  
    
		return msg;


  		




  82
  49
  
    
	}


  		




  83
  50
  
    

  		












Also available in:  Unified diff