Clean up.

artie-labs · Oct 24, 2024 · c062c5c · c062c5c
1 parent eb48521
commit c062c5c
Showing 1 changed file with 7 additions and 11 deletions.
diff --git a/lib/typing/mongo/bson_test.go b/lib/typing/mongo/bson_test.go
@@ -116,16 +116,13 @@ func TestJSONEToMap(t *testing.T) {
 	}
 	{
 		// Date
-		extendedTime, isOk := result["order_date"].(*ext.ExtendedTime)
-		assert.True(t, isOk)
-		assert.Equal(t, ext.NewExtendedTime(time.Date(2016, time.February, 21, 0, 0, 0, 0, time.UTC), ext.TimestampTZKindType, ext.ISO8601), extendedTime)
+		assert.Equal(t, time.Date(2016, time.February, 21, 0, 0, 0, 0, time.UTC), result["order_date"].(time.Time))
 	}
 	{
 		// Timestamp
-		extendedTime, isOk := result["test_timestamp"]
-		assert.True(t, isOk)
-		assert.Equal(t, ext.NewExtendedTime(time.Date(2023, time.March, 16, 1, 18, 37, 0, time.UTC), ext.TimestampTZKindType, ext.ISO8601), extendedTime)
-		assert.Equal(t, "2023-03-16T01:18:37+00:00", extendedTime.(*ext.ExtendedTime).GetTime().Format(ext.ISO8601))
+		_ts := result["test_timestamp"].(time.Time)
+		assert.Equal(t, time.Date(2023, time.March, 16, 1, 18, 37, 0, time.UTC), _ts)
+		assert.Equal(t, "2023-03-16T01:18:37+00:00", _ts.Format(ext.ISO8601))
 	}
 	{
 		// Boolean
@@ -220,10 +217,9 @@ func TestBsonValueToGoValue(t *testing.T) {
 		result, err := bsonValueToGoValue(dateTime)
 		assert.NoError(t, err)
 
-		extendedTime, isOk := result.(*ext.ExtendedTime)
-		assert.True(t, isOk)
-		assert.Equal(t, ext.NewExtendedTime(time.Date(2021, time.January, 1, 0, 0, 0, 0, time.UTC), ext.TimestampTZKindType, ext.ISO8601), extendedTime)
-		assert.Equal(t, "2021-01-01T00:00:00+00:00", extendedTime.GetTime().Format(ext.ISO8601))
+		_ts := result.(time.Time)
+		assert.Equal(t, time.Date(2021, time.January, 1, 0, 0, 0, 0, time.UTC), _ts)
+		assert.Equal(t, "2021-01-01T00:00:00+00:00", _ts.Format(ext.ISO8601))
 	}
 	{
 		// primitive.ObjectID