Add migration tests

Confirms already working code.