TezTaskID firstTask = tasks.keySet().iterator().next();
startVertex(v3);
Vertex v1 = vertices.get("vertex1");
EdgeManager mockEdgeManager = mock(EdgeManager.class);
Map<Vertex, EdgeManager> edgeManager = Collections.singletonMap(
v1, mockEdgeManager);
v3.setParallelism(1, edgeManager);
Assert.assertEquals(1, v3.getTotalTasks());
Assert.assertEquals(1, tasks.size());