ModelEventsTest.java

/* Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 
 *      http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.activiti.engine.test.api.event;

import org.activiti.engine.delegate.event.ActivitiEvent;
import org.activiti.engine.delegate.event.ActivitiEventType;
import org.activiti.engine.delegate.event.ActivitiEntityEvent;
import org.activiti.engine.impl.test.PluggableActivitiTestCase;
import org.activiti.engine.repository.Model;

/**
 * Test case for all {@link ActivitiEvent}s related to models.
 * 
 * @author Frederik Heremans
 */
public class ModelEventsTest extends PluggableActivitiTestCase {

	private TestActivitiEntityEventListener listener;
	
	/**
	 * Test create, update and delete events of model entities.
	 */
	public void testModelEvents() throws Exception {
		Model model = null;
		try {
			model = repositoryService.newModel();
			model.setName("My model");
			model.setKey("key");
			repositoryService.saveModel(model);
			
			// Check create event
			assertEquals(2, listener.getEventsReceived().size());
			assertEquals(ActivitiEventType.ENTITY_CREATED, listener.getEventsReceived().get(0).getType());
			assertEquals(model.getId(), ((Model) ((ActivitiEntityEvent) listener.getEventsReceived().get(0)).getEntity()).getId());
			
			assertEquals(ActivitiEventType.ENTITY_INITIALIZED, listener.getEventsReceived().get(1).getType());
			assertEquals(model.getId(), ((Model) ((ActivitiEntityEvent) listener.getEventsReceived().get(1)).getEntity()).getId());
			listener.clearEventsReceived();
			
			
			// Update model
			model = repositoryService.getModel(model.getId());
			model.setName("Updated");
			repositoryService.saveModel(model);
			assertEquals(1, listener.getEventsReceived().size());
			assertEquals(ActivitiEventType.ENTITY_UPDATED, listener.getEventsReceived().get(0).getType());
			assertEquals(model.getId(), ((Model) ((ActivitiEntityEvent) listener.getEventsReceived().get(0)).getEntity()).getId());
			listener.clearEventsReceived();
			
			// Test additional update-methods (source and extra-source)
			repositoryService.addModelEditorSource(model.getId(), "test".getBytes());
			repositoryService.addModelEditorSourceExtra(model.getId(), "test extra".getBytes());
			assertEquals(2, listener.getEventsReceived().size());
			assertEquals(ActivitiEventType.ENTITY_UPDATED, listener.getEventsReceived().get(0).getType());
			assertEquals(ActivitiEventType.ENTITY_UPDATED, listener.getEventsReceived().get(1).getType());
			listener.clearEventsReceived();
			
			// Delete model events
			repositoryService.deleteModel(model.getId());
			assertEquals(1, listener.getEventsReceived().size());
			assertEquals(ActivitiEventType.ENTITY_DELETED, listener.getEventsReceived().get(0).getType());
			assertEquals(model.getId(), ((Model) ((ActivitiEntityEvent) listener.getEventsReceived().get(0)).getEntity()).getId());
			listener.clearEventsReceived();
			
		} finally {
			if(model != null && repositoryService.getModel(model.getId()) != null) {
				repositoryService.deleteModel(model.getId());
			}
		}
	}
	
	@Override
	protected void setUp() throws Exception {
	  super.setUp();
	  listener = new TestActivitiEntityEventListener(Model.class);
	  processEngineConfiguration.getEventDispatcher().addEventListener(listener);
	}
	
	@Override
	protected void tearDown() throws Exception {
	  super.tearDown();
	  
	  if(listener != null) {
	  	processEngineConfiguration.getEventDispatcher().removeEventListener(listener);
	  }
	}
}