Implement YamlConfig.get()

Signed-off-by: Zack Cerza <zack@redhat.com>

diff --git a/teuthology/config.py b/teuthology/config.py
index 8f3e2ea00cc167e0140b003c50501cb88d178b60..07e4b3db1fa41df3a158bcb877030740d65e8e25 100644 (file)
--- a/teuthology/config.py
+++ b/teuthology/config.py
@@ -80,6 +80,9 @@ class YamlConfig(collections.MutableMapping):
         """
         return str(self)
 
+    def get(self, key, default=None):
+        return self._conf.get(key, default)
+
     def __str__(self):
         return yaml.safe_dump(self._conf, default_flow_style=False).strip()
 

diff --git a/teuthology/test/test_config.py b/teuthology/test/test_config.py
index 30c32491906bc31f249e38d66f073728d324b4d6..2b84f80992253e588148c9f4d76448c1132a3b44 100644 (file)
--- a/teuthology/test/test_config.py
+++ b/teuthology/test/test_config.py
@@ -70,6 +70,13 @@ class TestYamlConfig(object):
         d.update(conf_obj)
         assert d["foo"] == "bar"
 
+    def test_get(self):
+        conf_obj = self.test_class()
+        assert conf_obj.get('foo') is None
+        assert conf_obj.get('foo', 'bar') == 'bar'
+        conf_obj.foo = 'baz'
+        assert conf_obj.get('foo') == 'baz'
+
 
 class TestTeuthologyConfig(TestYamlConfig):
     def setup(self):