Allow for test cases without database
Added unit test for migration step interface
Reduce duplicate code