add test for the checkpoint migration case