def test_key_values(self): LOG.debug('Zone transformer test: get key values') # Test setup zone_name = 'zone123' zone_transformer = ZoneTransformer(self.transformers) # Test action observed_key_fields = zone_transformer._key_values([zone_name]) # Test assertions self.assertEqual(EntityTypes.RESOURCE, observed_key_fields[0]) self.assertEqual( ZoneTransformer(self.transformers).ZONE_TYPE, observed_key_fields[1] ) self.assertEqual(zone_name, observed_key_fields[2])
def test_create_placeholder_vertex(self): LOG.debug('Zone transformer test: create placeholder vertex') # Test setup zone_name = 'zone123' timestamp = datetime.datetime.utcnow() zone_transformer = ZoneTransformer(self.transformers) # Test action placeholder = zone_transformer.create_placeholder_vertex( zone_name, timestamp ) # Test assertions observed_id_values = placeholder.vertex_id.split( TransformerBase.KEY_SEPARATOR) expected_id_values = ZoneTransformer(self.transformers)._key_values( [zone_name] ) self.assertEqual(observed_id_values, expected_id_values) observed_time = placeholder.get(VertexProperties.UPDATE_TIMESTAMP) self.assertEqual(observed_time, timestamp) observed_subtype = placeholder.get(VertexProperties.TYPE) self.assertEqual(observed_subtype, zone_transformer.ZONE_TYPE) observed_entity_id = placeholder.get(VertexProperties.ID) self.assertEqual(observed_entity_id, zone_name) observed_category = placeholder.get(VertexProperties.CATEGORY) self.assertEqual(observed_category, EntityTypes.RESOURCE) is_placeholder = placeholder.get(VertexProperties.IS_PLACEHOLDER) self.assertEqual(is_placeholder, True)