added testcase derived from issue #2

staffanm · Jan 21, 2016 · fa12188 · fa12188
1 parent d61816b
commit fa12188
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 1 deletion.
diff --git a/tests/test_layeredconfig.py b/tests/test_layeredconfig.py
@@ -1244,6 +1244,24 @@ def test_layered_subsections(self):
         self.assertEqual(['force', 'home', 'loglevel'], list(cfg.mymodule))
 
 
+    def test_layered_yaml(self):
+        # see https://github.com/staffanm/layeredconfig/issues/2
+        with open("1.yaml", "w") as fp:
+            fp.write("""a:
+  b: b
+""")
+        with open("2.yaml", "w") as fp:
+            fp.write("somevar: value")
+
+        try:
+            yamls = [YAMLFile('1.yaml'), YAMLFile('2.yaml')]
+            cfg = LayeredConfig(*yamls)
+            self.assertEqual(cfg.somevar, 'value')
+            self.assertEqual(cfg.a.b, 'b')
+        finally:
+            os.unlink("1.yaml")
+            os.unlink("2.yaml")
+
 
 class TestSubsections(unittest.TestCase):
     def test_list(self):

diff --git a/tests/test_withFuture.py b/tests/test_withFuture.py
@@ -14,7 +14,7 @@
 else: 
     import unittest
 
-from layeredconfig import LayeredConfig, Defaults, Environment
+from layeredconfig import LayeredConfig, Defaults, Environment, INIFile
 
 
 class TestFuture(unittest.TestCase):