def test_tag_key_exists(self):
key = 'key1'
value = 'val1'
tag_map = tags.TagMap(tags=[tags.Tag(key, value)])
self.assertTrue(tag_map.tag_key_exists(key))
self.assertFalse(tag_map.tag_key_exists('nokey'))
def test_constructor_explicit(self):
tags_list = [
tags.Tag(tags.TagKey('key1'), tags.TagValue('value1')),
tags.Tag(tags.TagKey('key2'), tags.TagValue('value2'))
]
tag_map = tags.TagMap(tags=tags_list)
self.assertEqual(tag_map.map, dict(tags_list))
def test_value(self):
key = 'key1'
value = 'value1'
tag_map = tags.TagMap(tags=[tags.Tag(key, value)])
test_val = tag_map.get_value(key)
self.assertEqual(test_val, value)
with self.assertRaises(KeyError):
tag_map.get_value(key='not_in_map')
def test_delete(self):
key = 'key1'
tag1 = tags.Tag('key1', 'value1')
tag2 = tags.Tag('key2', 'value2')
tt = [tag1, tag2]
tag_map = tags.TagMap(tags=tt)
tag_map.delete(key=key)
self.assertEqual(tag_map.map, {tag2.key: tag2.value})
def test_insert(self):
test_key = tags.TagKey('key1')
test_value = tags.TagValue('value1')
tag_map = tags.TagMap()
tag_map.insert(key=test_key, value=test_value)
self.assertEqual({test_key: test_value}, tag_map.map)
tag_map.insert(key=test_key, value=test_value)
self.assertEqual({test_key: test_value}, tag_map.map)
self.assertRaises(
ValueError, tag_map.insert, key='Æ!01kr', value=test_value)
def test_update(self):
key_1 = 'key1'
val1 = 'value1'
tag = tags.Tag(key_1, val1)
tag_map = tags.TagMap([tag])
tag_map.update(key=key_1, value=val1)
self.assertEqual({'key1': 'value1'}, tag_map.map)
key_2 = 'key2'
tag_map.update(key=key_2, value=val1)
self.assertEqual({'key1': 'value1'}, tag_map.map)
val_2 = 'value2'
tag_map.update(key=key_1, value=val_2)
self.assertEqual({'key1': 'value2'}, tag_map.map)
def configure():
"""Globally enables metrics collection."""
global _METRICS_ENABLED
if _METRICS_ENABLED:
return
_METRICS_ENABLED = True
STATS.view_manager.register_exporter(_new_exporter())
latency_view = view.View(
"fireci/latency", "Latency of fireci execution stages", _TAGS, _m_latency,
aggregation.LastValueAggregation())
success_view = view.View(
"fireci/success", "Success indication of fireci execution stages", _TAGS,
_m_success, aggregation.LastValueAggregation())
STATS.view_manager.register_view(latency_view)
STATS.view_manager.register_view(success_view)
context = tags.TagMap()
for tag in _TAGS:
if tag.upper() in os.environ:
context.insert(tag, tags.TagValue(os.environ[tag.upper()]))
execution_context.set_current_tag_map(context)
def test_constructor_defaults(self):
tag_map = tags.TagMap()
self.assertEqual(tag_map.map, {})