The tests are less focused, as far as I can tell all you would need to mock is columns on this class. This would simplify the tests a fair bit I think. I don't care that much, the migrations just seem unnecessary to me.