Add tests for the squashmigrations command