Added unit tests for Connection class