Refactoring and adding tests around DruidNodeConnection