def test_timestamps_return_none_if_set_to_none(self):
model = Model()
model.unguard()
timestamps = {
'created_at': datetime.datetime.now(),
'updated_at': datetime.datetime.now()
}
instance = model.new_instance(timestamps)
instance.created_at = None
self.assertIsNone(instance.created_at)
model.reguard()
def test_timestamps_are_returned_as_objects_on_create(self):
model = Model()
model.unguard()
timestamps = {
'created_at': datetime.datetime.now(),
'updated_at': datetime.datetime.now()
}
instance = model.new_instance(timestamps)
self.assertIsInstance(instance.created_at, Arrow)
self.assertIsInstance(instance.updated_at, Arrow)
model.reguard()