Add test for the custom timestamp feature.