tests: test_dump_default_config()

stuarteberg · Jan 2, 2019 · 901cbdf · 901cbdf
1 parent a6ce10a
commit 901cbdf
Showing 1 changed file with 18 additions and 1 deletion.
diff --git a/confiddler/tests/test_core.py b/confiddler/tests/test_core.py
@@ -5,7 +5,7 @@
 import pytest
 from ruamel.yaml import YAML
 
-from confiddler import load_config, emit_defaults, validate, flow_style, ValidationError
+from confiddler import load_config, emit_defaults, validate, dump_default_config, flow_style, ValidationError
 
 yaml = YAML()
 yaml.default_flow_style = False
@@ -131,6 +131,23 @@ def test_emit_defaults():
     validate(defaults, schema)
 
 
+def test_dump_default_config():
+    schema = copy.deepcopy(TEST_SCHEMA)
+    defaults = emit_defaults(schema)
+
+    # To file object
+    f = StringIO()
+    dump_default_config(schema, f)
+    f.seek(0)
+
+    assert yaml.load(f) == defaults
+
+    # To string
+    s = dump_default_config(schema)
+    assert s == f.getvalue()
+
+
+
 def test_emit_incomplete_defaults():
     schema = copy.deepcopy(TEST_SCHEMA)