def _validate_node_neighbor(self, node_neighbor, zone_vertex_id):
expected_node_neighbor = tbase.create_node_placeholder_vertex()
self.assertEqual(expected_node_neighbor, node_neighbor.vertex)
# Validate neighbor edge
edge = node_neighbor.edge
self.assertEqual(edge.source_id, node_neighbor.vertex.vertex_id)
self.assertEqual(edge.target_id, zone_vertex_id)
self.assertEqual(edge.label, EdgeLabels.CONTAINS)
def _create_node_neighbor(self, zone_vertex_id):
node_vertex = base.create_node_placeholder_vertex()
relation_edge = graph_utils.create_edge(
source_id=node_vertex.vertex_id,
target_id=zone_vertex_id,
relationship_type=EdgeLabels.CONTAINS
)
return base.Neighbor(node_vertex, relation_edge)
