add origin unit tests

Author: reuvenharrison

decode_test.go

@@ -21,13 +21,12 @@ import (
 	"fmt"
 	"io"
 	"math"
-	"os"
 	"reflect"
 	"strings"
 	"time"
 
+	yaml "github.com/oasdiff/yaml3"
 	. "gopkg.in/check.v1"
-	"github.com/oasdiff/yaml3"
 )
 
 var unmarshalIntTest = 123
@@ -1742,34 +1741,6 @@ func (s *S) TestFuzzCrashers(c *C) {
 	}
 }
 
-func (s *S) TestLocation(c *C) {
-	input := `paths:
-  /testpath:
-    get:
-      responses:
-        "200":
-          $ref: "#/components/responses/testpath_200_response"
-
-components:
-  responses:
-    testpath_200_response:
-      description: a custom response
-      content:
-        application/json:
-          schema:
-            type: string
-`
-	dec := yaml.NewDecoder(bytes.NewBufferString(input))
-	dec.Origin(false)
-	var out any
-	err := dec.Decode(&out)
-	c.Assert(err, IsNil)
-	result, err := yaml.Marshal(out)
-	c.Assert(err, IsNil)
-	fo, _ := os.Create("output.yaml")
-	fo.Write(result)
-}
-
 //var data []byte
 //func init() {
 //	var err error

origin_test.go

@@ -0,0 +1,63 @@
+package yaml_test
+
+import (
+	"bytes"
+
+	yaml "github.com/oasdiff/yaml3"
+	. "gopkg.in/check.v1"
+)
+
+func (s *S) TestOrigin_Disabled(c *C) {
+	input := `
+root:
+    hello: world
+`
+
+	dec := yaml.NewDecoder(bytes.NewBufferString(input[1:]))
+	dec.Origin(false)
+	var out any
+	err := dec.Decode(&out)
+	c.Assert(err, IsNil)
+	result, err := yaml.Marshal(out)
+	c.Assert(err, IsNil)
+
+	buf := new(bytes.Buffer)
+	buf.Write(result)
+
+	c.Assert(buf.String(), Equals, input[1:])
+}
+
+func (s *S) TestOrigin_Enabled(c *C) {
+	input := `
+root:
+    hello: world
+`
+
+	dec := yaml.NewDecoder(bytes.NewBufferString(input[1:]))
+	dec.Origin(true)
+	var out any
+	err := dec.Decode(&out)
+	c.Assert(err, IsNil)
+	result, err := yaml.Marshal(out)
+	c.Assert(err, IsNil)
+
+	buf := new(bytes.Buffer)
+	buf.Write(result)
+
+	output := `
+root:
+    hello: world
+    origin:
+        fields:
+            hello:
+                column: 5
+                line: 2
+                name: hello
+        key:
+            column: 1
+            line: 1
+            name: root
+`
+
+	c.Assert(buf.String(), Equals, output[1:])
+}