PojoSerializationTest.java

package org.gradoop.flink.model.impl.pojo;

import org.gradoop.common.GradoopTestUtils;
import org.gradoop.common.model.api.entities.EPGMGraphHead;
import org.gradoop.common.model.api.entities.EPGMVertex;
import org.gradoop.common.model.impl.properties.Properties;
import org.gradoop.flink.model.GradoopFlinkTestBase;
import org.gradoop.common.model.api.entities.EPGMEdge;
import org.gradoop.common.model.impl.id.GradoopId;
import org.gradoop.common.model.impl.id.GradoopIdList;
import org.gradoop.common.model.impl.pojo.EdgeFactory;
import org.gradoop.common.model.impl.pojo.GraphHeadFactory;
import org.gradoop.common.model.impl.pojo.VertexFactory;
import org.gradoop.flink.model.impl.GradoopFlinkTestUtils;
import org.junit.Assert;
import org.junit.Test;

public class PojoSerializationTest extends GradoopFlinkTestBase {

  @Test
  public void testVertexSerialization() throws Exception {
    EPGMVertex vertexIn = new VertexFactory().createVertex(
      "Person",
      Properties.createFromMap(GradoopTestUtils.SUPPORTED_PROPERTIES),
      GradoopIdList.fromExisting(GradoopId.get()));

    Assert.assertEquals("EPGMVertex POJOs were not equal",
      vertexIn, GradoopFlinkTestUtils.writeAndRead(vertexIn));
  }

  @Test
  public void testEdgeSerialization() throws Exception {
    EPGMEdge edgeIn = new EdgeFactory().createEdge(
      "knows",
      GradoopId.get(),
      GradoopId.get(),
      Properties.createFromMap(GradoopTestUtils.SUPPORTED_PROPERTIES),
      GradoopIdList.fromExisting(GradoopId.get(), GradoopId.get()));

    Assert.assertEquals("EPGMEdge POJOs were not equal",
      edgeIn, GradoopFlinkTestUtils.writeAndRead(edgeIn));
  }

  @Test
  public void testGraphHeadSerialization() throws Exception {
    EPGMGraphHead graphHeadIn = new GraphHeadFactory().createGraphHead(
      "Community",
      Properties.createFromMap(GradoopTestUtils.SUPPORTED_PROPERTIES)
    );

    Assert.assertEquals("EPGMGraphHead POJOs were not equal",
      graphHeadIn, GradoopFlinkTestUtils.writeAndRead(graphHeadIn));
  }


}