def test_get_version(unpack_version): """ Returns migration version based on database table """ db = mock.Mock() assert mod.get_version(db) == unpack_version.return_value db.query.assert_any_call(mod.GET_VERSION_SQL) unpack_version.assert_called_once_with( db.query.return_value.result.user_version)
def test_get_version_drop_db(drop_db): """ Version-tracking table is crated if it doesn't exist """ db = mock.Mock() db.query.return_value.result.user_version = None assert mod.get_version(db) == (0, 0) drop_db.assert_called_once_with(db)