DynamicContainmentTest.java

package org.emfjson.jackson.tests.dynamic;

import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.ObjectMapper;
import org.eclipse.emf.common.util.URI;
import org.eclipse.emf.ecore.EClass;
import org.eclipse.emf.ecore.EObject;
import org.eclipse.emf.ecore.EReference;
import org.eclipse.emf.ecore.resource.ResourceSet;
import org.eclipse.emf.ecore.util.EcoreUtil;
import org.emfjson.jackson.support.DynamicFixture;
import org.emfjson.jackson.support.StandardFixture;
import org.emfjson.jackson.utils.EObjects;
import org.junit.Assert;
import org.junit.ClassRule;
import org.junit.Test;

import static org.emfjson.jackson.databind.EMFContext.Attributes.RESOURCE_SET;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertSame;

public class DynamicContainmentTest {

	@ClassRule
	public static StandardFixture fixture = new DynamicFixture();

	private ObjectMapper mapper = fixture.mapper();
	private ResourceSet resourceSet = fixture.getResourceSet();

	@Test
	public void testSaveContainmentWithOpposite() {
		JsonNode expected = mapper.createObjectNode()
				.put("eClass", "http://emfjson/dynamic/model#//A")
				.set("containB", mapper.createObjectNode());

		EClass classA = (EClass) resourceSet.getEObject(URI.createURI("http://emfjson/dynamic/model#//A"), true);
		EClass classB = (EClass) resourceSet.getEObject(URI.createURI("http://emfjson/dynamic/model#//B"), true);

		EObject a1 = EcoreUtil.create(classA);
		EObject b1 = EcoreUtil.create(classB);
		EObjects.setOrAdd(b1, (EReference) classA.getEStructuralFeature("parent"), a1);

		Assert.assertEquals(expected, mapper.valueToTree(a1));
	}

	@Test
	public void testLoadContainmentWithOpposite() throws JsonProcessingException {
		JsonNode data = mapper.createObjectNode()
				.put("eClass", "http://emfjson/dynamic/model#//A")
				.put("someKind", "e1")
				.set("containB", mapper.createObjectNode()
						.put("eClass", "http://emfjson/dynamic/model#//B")
						.put("someKind", "e1"));

		EObject a1 = mapper
				.reader()
				.withAttribute(RESOURCE_SET, resourceSet)
				.treeToValue(data, EObject.class);

		EObject b1 = (EObject) a1.eGet(a1.eClass().getEStructuralFeature("containB"));

		assertNotNull(b1);
		assertSame(a1, b1.eGet(b1.eClass().getEStructuralFeature("parent")));
	}

}