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)
