Add mockery to remove the need for __purge