Add connection integration tests