Added more tests for Connection class