Sounds like you are coming out smelling like a rose! And having the bearings changed for 1 hour of shop time is a bargain. I had much more time than that just getting the bearing cone out of the housing, but maybe they have the proper tools. Or maybe there is a trick I didn"t know about. Working with a die grinder worked for me, but it was WAY slow.
With the looseness you described, I would advise replacing the roller bearing AND the needle bearings, as the old ones are probably destroyed. I think I would ask to see the removed parts, "just to see how bad they are", to make sure they actually change both bearings.
I agree with you, listening to your machines is really important. I would be slightly concerned about wear to the rear camshaft gear from the poor mesh the old gear on the pump had with it. When you get it back together, if it doesn"t sound right, I would suggest shutting it right down, as you still may need to replace that drive gear to avoid further damage. Hope you don"t, and it will work OK. Good luck!