add test case for migration class