tests/org.eclipse.epsilon.flock.engine.test.unit/src/org/eclipse/epsilon/flock/model/domain/MigrationStrategyTests.java

/*******************************************************************************
 * Copyright (c) 2011 The University of York.
 * This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License 2.0
 * which is available at https://www.eclipse.org/legal/epl-2.0/
 * 
 * Contributors:
 *     Louis Rose - initial API and implementation
 ******************************************************************************/
package org.eclipse.epsilon.flock.model.domain;

import static org.hamcrest.Matchers.containsInAnyOrder;
import static org.junit.Assert.assertThat;
import static org.mockito.Mockito.mock;

import java.util.Collection;

import org.eclipse.epsilon.common.module.ModuleElement;
import org.eclipse.epsilon.flock.model.domain.rules.MigrateRule;
import org.eclipse.epsilon.flock.model.domain.typemappings.Deletion;
import org.eclipse.epsilon.flock.model.domain.typemappings.Retyping;
import org.junit.Test;

public class MigrationStrategyTests {

	private final MigrateRule rule = mock(MigrateRule.class);
	private final Retyping retyping = mock(Retyping.class);
	private final Deletion deletion = mock(Deletion.class);
	
	
	@Test
	public void constructorAddsConstructs() {
		final MigrationStrategy strategy = new MigrationStrategy(retyping, rule, deletion);
		
		assertThat(strategy.getTypeMappingsAndRules(),
		           containsInAnyOrder((ModuleElement)retyping,
		                              (ModuleElement)rule,
		                              (ModuleElement)deletion));
	}
	
	@Test
	public void addRule() {
		final MigrationStrategy strategy = new MigrationStrategy();
		
		strategy.addRule(rule);
		
		
		final Collection<ModuleElement> contents = strategy.getTypeMappingsAndRules();
		
		assertThat(contents, containsInAnyOrder((ModuleElement)rule));
	}
	
	@Test
	public void addTypeMappingConstruct() {
		final MigrationStrategy strategy = new MigrationStrategy();
		
		strategy.addTypeMappingConstruct(retyping);
		strategy.addTypeMappingConstruct(deletion);
		
		
		final Collection<ModuleElement> contents = strategy.getTypeMappingsAndRules();
		
		assertThat(contents, containsInAnyOrder((ModuleElement)retyping,
		                                        (ModuleElement)deletion));
	}
}