Boolean values should be output as bool in JSON. (#663)

SFDO-Tooling · May 12, 2022 · 4a51f8f · 4a51f8f
1 parent fff7c66
commit 4a51f8f
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/snowfakery/output_streams.py b/snowfakery/output_streams.py
@@ -253,6 +253,7 @@ class JSONOutputStream(FileOutputStream):
         **OutputStream.encoders,
         datetime.date: str,
         datetime.datetime: str,
+        bool: bool,
     }
     is_text = True
 

diff --git a/tests/test_output_streams.py b/tests/test_output_streams.py
@@ -259,6 +259,15 @@ def test_error_generates_empty_string(self):
             output_stream.close()
             assert s.getvalue() == ""
 
+    def test_bool(self):
+        yaml = """
+          - object: foo
+            fields:
+                is_true: True
+            """
+        values = self.do_output(yaml)["foo"][0]
+        assert values["is_true"] is True
+
 
 class TestCSVOutputStream(OutputCommonTests):
     def do_output(self, yaml):