private void verifyEntityProperties( Vertex entityVertex, String entityName, RelationshipType entityType) { Assert.assertEquals(entityName, entityVertex.getProperty(RelationshipProperty.NAME.getName())); Assert.assertEquals( entityType.getName(), entityVertex.getProperty(RelationshipProperty.TYPE.getName())); Assert.assertNotNull(entityVertex.getProperty(RelationshipProperty.TIMESTAMP.getName())); }
private void verifyEntityGraph(RelationshipType feedType, String classification) { // feeds owned by a user List<String> feedNamesOwnedByUser = getFeedsOwnedByAUser(feedType.getName()); Assert.assertEquals( feedNamesOwnedByUser, Arrays.asList("impression-feed", "clicks-feed", "imp-click-join1", "imp-click-join2")); // feeds classified as secure verifyFeedsClassifiedAsSecure( feedType.getName(), Arrays.asList("impression-feed", "clicks-feed", "imp-click-join2")); // feeds owned by a user and classified as secure verifyFeedsOwnedByUserAndClassification( feedType.getName(), classification, Arrays.asList("impression-feed", "clicks-feed", "imp-click-join2")); }
private Vertex getEntityVertex(String entityName, RelationshipType entityType) { GraphQuery entityQuery = getQuery() .has(RelationshipProperty.NAME.getName(), entityName) .has(RelationshipProperty.TYPE.getName(), entityType.getName()); Iterator<Vertex> iterator = entityQuery.vertices().iterator(); Assert.assertTrue(iterator.hasNext()); Vertex entityVertex = iterator.next(); Assert.assertNotNull(entityVertex); return entityVertex; }