def test_allow_for_null_created_removed_and_archived_dates():
dateOne = "2019-12-14T15:01:02.000Z"
oldJson = {
"_id": {
"test_key_a": "test_value_a",
"test_key_b": "test_value_b"
},
"_lastModifiedDateTime": dateOne,
"createdDateTime": None,
"_removedDateTime": None,
"_archivedDateTime": None
}
newJson = {
"_id": {
"test_key_a": "test_value_a",
"test_key_b": "test_value_b"
},
"_lastModifiedDateTime": {
"$date": dateOne
},
"createdDateTime": None,
"_removedDateTime": None,
"_archivedDateTime": None
}
actual = wrap_dates(oldJson)
assert newJson == actual
def test_format_all_wrapped_dates():
dateOne = "2019-12-14T15:01:02.000+0000"
dateTwo = "2018-12-14T15:01:02.000+0000"
dateThree = "2017-12-14T15:01:02.000+0000"
dateFour = "2016-12-14T15:01:02.000+0000"
oldJson = {
"_id": {
"test_key_a": "test_value_a",
"test_key_b": "test_value_b"
},
"_lastModifiedDateTime": {
"$date": dateOne
},
"createdDateTime": {
"$date": dateTwo
},
"_removedDateTime": {
"$date": dateThree
},
"_archivedDateTime": {
"$date": dateFour
}
}
formattedDateOne = "2019-12-14T15:01:02.000Z"
formattedDateTwo = "2018-12-14T15:01:02.000Z"
formattedDateThree = "2017-12-14T15:01:02.000Z"
formattedDateFour = "2016-12-14T15:01:02.000Z"
newJson = {
"_id": {
"test_key_a": "test_value_a",
"test_key_b": "test_value_b"
},
"_lastModifiedDateTime": {
"$date": formattedDateOne
},
"createdDateTime": {
"$date": formattedDateTwo
},
"_removedDateTime": {
"$date": formattedDateThree
},
"_archivedDateTime": {
"$date": formattedDateFour
}
}
actual = wrap_dates(oldJson)
assert newJson == actual
def test_create_last_modified_if_missing_dates():
epoch = "1980-01-01T00:00:00.000Z"
oldJson = {
"_id": {
"test_key_a": "test_value_a",
"test_key_b": "test_value_b"
}
}
newJson = {
"_id": {
"test_key_a": "test_value_a",
"test_key_b": "test_value_b"
},
"_lastModifiedDateTime": {
"$date": epoch
}
}
actual = wrap_dates(oldJson)
assert newJson == actual
def test_wrap_all_dates():
date_one = "2019-12-14T15:01:02.000Z"
date_two = "2018-12-14T15:01:02.000Z"
date_three = "2017-12-14T15:01:02.000Z"
date_four = "2016-12-14T15:01:02.000Z"
old_json = {
"_id": {
"test_key_a": "test_value_a",
"test_key_b": "test_value_b"
},
"_lastModifiedDateTime": date_one,
"createdDateTime": date_two,
"_removedDateTime": date_three,
"_archivedDateTime": date_four
}
new_json = {
"_id": {
"test_key_a": "test_value_a",
"test_key_b": "test_value_b"
},
"_lastModifiedDateTime": {
"$date": date_one
},
"createdDateTime": {
"$date": date_two
},
"_removedDateTime": {
"$date": date_three
},
"_archivedDateTime": {
"$date": date_four
}
}
# TODO do we need to return last modified time as tuple
actual = wrap_dates(old_json)
assert new_json == actual
