Adding tests for connection