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;
}
