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_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_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_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_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)