Unittests for Loadbalancer wrapper plus minor fixes based on feedback.