def test_connect_auth(self):
with patch("ssl.SSLContext") as SSHContextMock:
context_mock = SSHContextMock.return_value
cert_file = Mock()
ca_file = Mock()
key_file = Mock()
connection = TLSConnection(certfile=cert_file,
cafile=ca_file,
keyfile=key_file)
connection.connect()
context_mock.load_cert_chain.assert_called_once()
context_mock.wrap_socket.assert_called_once()
self.assertFalse(context_mock.check_hostname)
def test_connect(self):
with patch("ssl.SSLContext") as SSHContextMock:
context_mock = SSHContextMock.return_value
connection = TLSConnection()
connection.connect()
context_mock.wrap_socket.assert_called_once()
