Python ssl_handshake Examples

Example #1

File: tests_link.py Project: bussiere/snakemq

 def test_ssl_handshake_none_sslobj(self):
     link = snakemq.link.Link()
     sock = mock.Mock()
     sock.sock._sslobj = None
     link._in_ssl_handshake.add(sock)
     link.poller = mock.Mock()
     self.assertEqual(link.ssl_handshake(sock), snakemq.link.SSL_HANDSHAKE_FAILED)
     self.assertNotIn(sock, link._in_ssl_handshake)

Example #2

File: tests_link.py Project: windhl/snakemq

 def test_ssl_handshake_none_sslobj(self):
     link = snakemq.link.Link()
     sock = mock.Mock()
     sock.sock._sslobj = None
     link._in_ssl_handshake.add(sock)
     link.poller = mock.Mock()
     self.assertEqual(link.ssl_handshake(sock),
                      snakemq.link.SSL_HANDSHAKE_FAILED)
     self.assertNotIn(sock, link._in_ssl_handshake)

Example #3

File: tests_link.py Project: bussiere/snakemq

 def test_failed_handshake_cleanup(self):
     link = snakemq.link.Link()
     sock = mock.Mock()
     sock.sock._sslobj = mock.Mock()  # anything but None
     sock.sock.do_handshake.side_effect = socket.error()
     link._in_ssl_handshake.add(sock)
     link.handle_close = mock.Mock()
     link.poller = mock.Mock()
     self.assertEqual(link.ssl_handshake(sock), snakemq.link.SSL_HANDSHAKE_FAILED)
     self.assertNotIn(sock, link._in_ssl_handshake)
     self.assertEqual(link.handle_close.call_count, 1)

Example #4

File: tests_link.py Project: mobilipia/snakemq

 def test_failed_handshake_cleanup(self):
     link = snakemq.link.Link()
     sock = mock.Mock()
     sock.sock._sslobj = mock.Mock()  # anything but None
     sock.sock.do_handshake.side_effect = socket.error()
     link._in_ssl_handshake.add(sock)
     link.handle_close = mock.Mock()
     link.poller = mock.Mock()
     self.assertEqual(link.ssl_handshake(sock),
                      snakemq.link.SSL_HANDSHAKE_FAILED)
     self.assertNotIn(sock, link._in_ssl_handshake)
     self.assertEqual(link.handle_close.call_count, 1)

Example #5

File: tests_link.py Project: dsiroky/snakemq

 def test_failed_handshake_cleanup(self):
     link = snakemq.link.Link()
     sock = mock.Mock()
     sock.sock._sslobj = mock.Mock()  # anything but None
     sock.sock.do_handshake.side_effect = socket.error()
     link._in_ssl_handshake.add(sock)
     def handle_close(sock):
         # during handle_close() the socket must be still have
         # the "handshake flag"
         self.assertIn(sock, link._in_ssl_handshake)
     link.handle_close = mock.Mock(wraps=handle_close)
     link.poller = mock.Mock()
     self.assertEqual(link.ssl_handshake(sock), snakemq.link.SSL_HANDSHAKE_FAILED)
     self.assertNotIn(sock, link._in_ssl_handshake)
     self.assertEqual(link.handle_close.call_count, 1)

Example #6

File: tests_link.py Project: windhl/snakemq

    def test_failed_handshake_cleanup(self):
        link = snakemq.link.Link()
        sock = mock.Mock()
        sock.sock._sslobj = mock.Mock()  # anything but None
        sock.sock.do_handshake.side_effect = socket.error()
        link._in_ssl_handshake.add(sock)

        def handle_close(sock):
            # during handle_close() the socket must be still have
            # the "handshake flag"
            self.assertIn(sock, link._in_ssl_handshake)

        link.handle_close = mock.Mock(wraps=handle_close)
        link.poller = mock.Mock()
        self.assertEqual(link.ssl_handshake(sock),
                         snakemq.link.SSL_HANDSHAKE_FAILED)
        self.assertNotIn(sock, link._in_ssl_handshake)
        self.assertEqual(link.handle_close.call_count, 1)