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()