fix: added additional time support and tests

common/coral/time.go

@@ -2,6 +2,7 @@ package coral
 
 import (
 	"encoding/json"
+	"reflect"
 	"time"
 
 	"github.com/pkg/errors"
@@ -33,14 +34,24 @@ func (t *Time) UnmarshalJSON(data []byte) error {
 		date, ok := v["$date"].(string)
 		if !ok || date == "" {
 			// Try to handle the case where we get something that looks like
-			// this: {"$date":{"$numberLong":"-62075098782000"}}
-			if _, ok := v["$date"].(map[string]int64); ok {
-				logrus.Warn("saw a date in the format: { $date: { $numberLong: -000 } }")
+			// this: {"$date":-62075098782000}
+			if _, ok := v["$date"].(map[string]float64); ok {
+				logrus.Warn("saw a date in the format: { $date: -62075098782000 }")
 
 				return nil
 			}
 
-			return errors.Errorf("invalid format: %#v", v)
+			// Try to handle the case where we get something that looks like
+			// this: {"$date":{"$numberLong":"-62075098782000"}}
+			if obj, ok := v["$date"].(map[string]interface{}); ok {
+				if _, ok := obj["$numberLong"].(string); ok {
+					logrus.Warn("saw a date in the format: { $date: { $numberLong: \"-62075098782000\" } }")
+
+					return nil
+				}
+			}
+
+			return errors.Errorf("invalid format: %#v, %s", v["$date"], reflect.TypeOf(v["$date"]))
 		}
 
 		tt, err := time.Parse(time.RFC3339, date)

common/coral/time_test.go

@@ -0,0 +1,26 @@
+package coral_test
+
+import (
+	"coral-importer/common/coral"
+	"testing"
+)
+
+func TestTimeUnmarshalJSON(t *testing.T) {
+	var tm coral.Time
+
+	inputs := []struct {
+		name  string
+		value string
+	}{
+		{name: "normal date time", value: `{ "$date": "2018-10-25T23:57:49.053Z" }`},
+		{name: "negative number long", value: `{ "$date": { "$numberLong": "-62075098782000" } }`},
+	}
+
+	for _, input := range inputs {
+		t.Run(input.name, func(t *testing.T) {
+			if err := tm.UnmarshalJSON([]byte(input.value)); err != nil {
+				t.Errorf("expected no error with input %s, got %v", input.value, err)
+			}
+		})
+	}
+}