1 files changed, 44 insertions, 0 deletions
diff --git a/src/mongo/gotools/src/github.com/mongodb/mongo-tools/common/bsonutil/undefined_test.go b/src/mongo/gotools/src/github.com/mongodb/mongo-tools/common/bsonutil/undefined_test.go
new file mode 100644
index 00000000000..8aee1a61488
--- /dev/null
+++ b/src/mongo/gotools/src/github.com/mongodb/mongo-tools/common/bsonutil/undefined_test.go
@@ -0,0 +1,44 @@
+// Copyright (C) MongoDB, Inc. 2014-present.
+//
+// Licensed under the Apache License, Version 2.0 (the "License"); you may
+// not use this file except in compliance with the License. You may obtain
+// a copy of the License at http://www.apache.org/licenses/LICENSE-2.0
+
+package bsonutil
+
+import (
+	"github.com/mongodb/mongo-tools/common/json"
+	. "github.com/smartystreets/goconvey/convey"
+	"gopkg.in/mgo.v2/bson"
+	"testing"
+)
+
+func TestUndefinedValue(t *testing.T) {
+
+	Convey("When converting JSON with undefined values", t, func() {
+
+		Convey("works for undefined literal", func() {
+			key := "key"
+			jsonMap := map[string]interface{}{
+				key: json.Undefined{},
+			}
+
+			err := ConvertJSONDocumentToBSON(jsonMap)
+			So(err, ShouldBeNil)
+			So(jsonMap[key], ShouldResemble, bson.Undefined)
+		})
+
+		Convey(`works for undefined document ('{ "$undefined": true }')`, func() {
+			key := "key"
+			jsonMap := map[string]interface{}{
+				key: map[string]interface{}{
+					"$undefined": true,
+				},
+			}
+
+			err := ConvertJSONDocumentToBSON(jsonMap)
+			So(err, ShouldBeNil)
+			So(jsonMap[key], ShouldResemble, bson.Undefined)
+		})
+	})
+}