def test_update_entity_state(self):
# create instance event with host neighbor and check validity
prop = {'status': 'STARTING'}
(vertex, neighbors, processor) =\
self._create_and_check_entity(properties=prop)
# check added entity
vertex = processor.entity_graph.get_vertex(vertex.vertex_id)
self.assertEqual('STARTING', vertex.properties[VertexProperties.STATE])
# update instance event with state running
vertex.properties[VertexProperties.STATE] = 'RUNNING'
vertex.properties[VertexProperties.UPDATE_TIMESTAMP] = str(utcnow())
processor.update_entity(vertex, neighbors)
# check state
self._check_graph(processor, self.NUM_VERTICES_AFTER_CREATION,
self.NUM_EDGES_AFTER_CREATION)
vertex = processor.entity_graph.get_vertex(vertex.vertex_id)
self.assertEqual('RUNNING', vertex.properties[VertexProperties.STATE])
def mark_edge_as_deleted(self, edge):
"""Marks the edge as is deleted, and updates delete timestamp"""
# TODO(Alexey): change the update_edge so it will raise a trigger
edge[EProps.IS_DELETED] = True
edge[EProps.UPDATE_TIMESTAMP] = str(utcnow())
self.update_edge(edge)
def _add_sampling_time(entity):
entity['sample_date'] = str(utils.utcnow())
return entity
def mark_vertex_as_deleted(self, vertex):
"""Marks the vertex as is deleted, and updates deletion timestamp"""
# TODO(Alexey): change the update_vertex so it will raise a trigger
vertex[VProps.IS_DELETED] = True
vertex[VProps.UPDATE_TIMESTAMP] = str(utcnow())
self.update_vertex(vertex)
