added some tests for migration implementaion