@SuppressWarnings("unchecked") protected Edge mockEdge(String id, Set<String> labels) { Edge e = mock(Edge.class); View v = mock(View.class); Object d = mock(Object.class); when(e.getContent()).thenReturn(v); when(v.getDefinition()).thenReturn(d); when(definitionAdapter.getId(eq(d))).thenReturn(DefinitionId.build(id)); when(definitionAdapter.getLabels(eq(d))).thenReturn(labels); when(e.getLabels()).thenReturn(labels); return e; }
@Test public void testBuild() { final Edge<Definition<Object>, Node> edge = tested.build(UUID, definition); assertNotNull(edge); assertEquals(UUID, edge.getUUID()); assertEquals(3, edge.getLabels().size()); assertTrue(edge.getLabels().contains(ID)); assertTrue(edge.getLabels().contains("label1")); assertTrue(edge.getLabels().contains("label2")); } }
@Before @SuppressWarnings("unchecked") public void setup() throws Exception { super.setup(); final Set<String> edgeLabels = Collections.singleton(EDGE_ID); when(edge.getContent()).thenReturn(edgeContent); when(edge.getLabels()).thenReturn(edgeLabels); when(edgeContent.getDefinition()).thenReturn(edgeDefinition); when(definitionAdapter.getId(eq(edgeDefinition))).thenReturn(DefinitionId.build(EDGE_ID)); when(context.getConnector()).thenReturn(edge); when(context.getSource()).thenReturn(Optional.of(parent)); when(context.getTarget()).thenReturn(Optional.of(candidate)); tested = new GraphConnectionEvaluationHandler(definitionManager, HANDLER); }
@Before @SuppressWarnings("unchecked") public void setup() throws Exception { super.setup(); final Set<String> edgeLabels = Collections.singleton(EDGE_ID); evalUtils = spy(new GraphEvaluationHandlerUtils(definitionManager)); when(context.getCandidate()).thenReturn(candidate); when(context.getEdge()).thenReturn(edge); when(edge.getContent()).thenReturn(edgeContent); when(edge.getLabels()).thenReturn(edgeLabels); when(edgeContent.getDefinition()).thenReturn(edgeDefinition); when(definitionAdapter.getId(eq(edgeDefinition))).thenReturn(DefinitionId.build(EDGE_ID)); tested = new ConnectorCardinalityEvaluationHandler(evalUtils, new EdgeCardinalityEvaluationHandler()); }