From: Zack Cerza <zack@redhat.com>
Date: Wed, 2 Dec 2015 16:58:45 +0000 (-0700)
Subject: Implement YamlConfig.get()
X-Git-Tag: 1.1.0~737^2~3
X-Git-Url: http://git-server-git.apps.pok.os.sepia.ceph.com/?a=commitdiff_plain;h=2452d6f61dafdfb87a93002575adbd0fc89fdbb0;p=teuthology.git

Implement YamlConfig.get()

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

diff --git a/teuthology/config.py b/teuthology/config.py
index 8f3e2ea0..07e4b3db 100644
--- 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 30c32491..2b84f809 100644
--- 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):