Add a snapshot/restore test