TezVertexID mockSrcVertexId1 = new TezVertexID(dagId, 1);
EdgeProperty eProp1 = new EdgeProperty(
EdgeProperty.DataMovementType.SCATTER_GATHER,
EdgeProperty.DataSourceType.PERSISTED,
SchedulingType.SEQUENTIAL,
new OutputDescriptor("out"),
new InputDescriptor("in"));
when(mockSrcVertex1.getVertexId()).thenReturn(mockSrcVertexId1);
Vertex mockSrcVertex2 = mock(Vertex.class);
TezVertexID mockSrcVertexId2 = new TezVertexID(dagId, 2);
EdgeProperty eProp2 = new EdgeProperty(
EdgeProperty.DataMovementType.SCATTER_GATHER,
EdgeProperty.DataSourceType.PERSISTED,
SchedulingType.SEQUENTIAL,
new OutputDescriptor("out"),
new InputDescriptor("in"));
when(mockSrcVertex2.getVertexId()).thenReturn(mockSrcVertexId2);
Vertex mockSrcVertex3 = mock(Vertex.class);
TezVertexID mockSrcVertexId3 = new TezVertexID(dagId, 3);
EdgeProperty eProp3 = new EdgeProperty(
EdgeProperty.DataMovementType.BROADCAST,
EdgeProperty.DataSourceType.PERSISTED,
SchedulingType.SEQUENTIAL,
new OutputDescriptor("out"),
new InputDescriptor("in"));
when(mockSrcVertex3.getVertexId()).thenReturn(mockSrcVertexId3);
Vertex mockManagedVertex = mock(Vertex.class);
TezVertexID mockManagedVertexId = new TezVertexID(dagId, 4);