add function to output JSON schema

miracl · Jan 23, 2018 · 81779af · 81779af
1 parent 3968ff1
commit 81779af
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 0 deletions.
diff --git a/conflate.go b/conflate.go
@@ -174,3 +174,8 @@ func (c *Conflate) MarshalYAML() ([]byte, error) {
 func (c *Conflate) MarshalTOML() ([]byte, error) {
 	return tomlMarshal(c.data)
 }
+
+// MarshalSchema exports the schema as JSON
+func (c *Conflate) MarshalSchema() ([]byte, error) {
+	return jsonMarshal(c.schema)
+}
diff --git a/conflate_test.go b/conflate_test.go
@@ -310,6 +310,24 @@ func TestConflate_MarshalTOML(t *testing.T) {
 	assert.Equal(t, testMarshalTOML, data)
 }
 
+func TestConflate_MarshalSchema(t *testing.T) {
+	c, err := FromFiles()
+	assert.Nil(t, err)
+	err = c.SetSchemaData([]byte("{}"))
+	assert.Nil(t, err)
+	data, err := c.MarshalSchema()
+	assert.Nil(t, err)
+	assert.Equal(t, "{}\n", string(data))
+}
+
+func TestConflate_MarshalSchemaNil(t *testing.T) {
+	c, err := FromFiles()
+	assert.Nil(t, err)
+	data, err := c.MarshalSchema()
+	assert.Nil(t, err)
+	assert.Equal(t, string(data), "null\n")
+}
+
 func TestConflate_MergeDataError(t *testing.T) {
 	c := New()
 	err := c.MergeData([]byte("{not valid"))