tests/org.eclipse.emfforms.core.services.segments.mapping.test/src/org/eclipse/emfforms/internal/core/services/segments/mapping/EMFMappingValueProperty_Test.java

/*******************************************************************************
 * Copyright (c) 2011-2018 EclipseSource Muenchen GmbH and others.
 *
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License 2.0
 * which accompanies this distribution, and is available at
 * https://www.eclipse.org/legal/epl-2.0/
 *
 * SPDX-License-Identifier: EPL-2.0
 *
 * Contributors:
 * Lucas Koehler - initial API and implementation
 ******************************************************************************/
package org.eclipse.emfforms.internal.core.services.segments.mapping;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNull;

import org.eclipse.emf.common.command.BasicCommandStack;
import org.eclipse.emf.ecore.EClass;
import org.eclipse.emf.edit.domain.AdapterFactoryEditingDomain;
import org.eclipse.emf.edit.provider.ComposedAdapterFactory;
import org.eclipse.emfforms.core.services.databinding.testmodel.test.model.A;
import org.eclipse.emfforms.core.services.databinding.testmodel.test.model.C;
import org.eclipse.emfforms.core.services.databinding.testmodel.test.model.TestFactory;
import org.eclipse.emfforms.core.services.databinding.testmodel.test.model.TestPackage;
import org.junit.Test;

/**
 * JUnit test cases for {@link EMFMappingValueProperty}.
 *
 * @author Lucas Koehler
 *
 */
public class EMFMappingValueProperty_Test {

	@Test
	public void testDoGetValue() {
		final C c = TestFactory.eINSTANCE.createC();
		final EClass eClass1 = TestPackage.eINSTANCE.getB();
		final A a1 = TestFactory.eINSTANCE.createA();
		c.getEClassToA().put(eClass1, a1);

		final EMFMappingValueProperty mappingValueProperty = new EMFMappingValueProperty(null, eClass1,
			TestPackage.eINSTANCE.getC_EClassToA());
		assertEquals(a1, mappingValueProperty.doGetValue(c));
	}

	@Test
	public void testDoGetValueNoMapEntry() {
		final C c = TestFactory.eINSTANCE.createC();
		final EClass eClass1 = TestPackage.eINSTANCE.getB();

		final EMFMappingValueProperty mappingValueProperty = new EMFMappingValueProperty(null, eClass1,
			TestPackage.eINSTANCE.getC_EClassToA());
		assertNull(mappingValueProperty.doGetValue(c));
	}

	@Test
	public void testDoSetValueObjectReplaceEntry() {
		final C c = TestFactory.eINSTANCE.createC();
		final EClass eClass1 = TestPackage.eINSTANCE.getB();
		final A a1 = TestFactory.eINSTANCE.createA();
		final A a2 = TestFactory.eINSTANCE.createA();
		a2.setB(TestFactory.eINSTANCE.createB());
		c.getEClassToA().put(eClass1, a1);

		final AdapterFactoryEditingDomain domain = new AdapterFactoryEditingDomain(
			new ComposedAdapterFactory(ComposedAdapterFactory.Descriptor.Registry.INSTANCE),
			new BasicCommandStack());

		final EMFMappingValueProperty mappingValueProperty = new EMFMappingValueProperty(domain, eClass1,
			TestPackage.eINSTANCE.getC_EClassToA());
		assertEquals(a1, mappingValueProperty.doGetValue(c));

		mappingValueProperty.doSetValue(c, a2);
		assertEquals(a2, mappingValueProperty.doGetValue(c));
	}

	@Test
	public void testDoSetValueObjectAddEntry() {
		final C c = TestFactory.eINSTANCE.createC();
		final EClass eClass1 = TestPackage.eINSTANCE.getA();
		final A a1 = TestFactory.eINSTANCE.createA();
		final AdapterFactoryEditingDomain domain = new AdapterFactoryEditingDomain(
			new ComposedAdapterFactory(ComposedAdapterFactory.Descriptor.Registry.INSTANCE),
			new BasicCommandStack());

		final EMFMappingValueProperty mappingValueProperty = new EMFMappingValueProperty(domain, eClass1,
			TestPackage.eINSTANCE.getC_EClassToA());
		mappingValueProperty.doSetValue(c, a1);

		assertEquals(a1, mappingValueProperty.doGetValue(c));
	}
}