add connection wrapper unit test