PCQRSCANER/venv/Lib/site-packages/imapclient/test/test_starttls.py
2019-12-22 21:51:47 +01:00

64 lines
2.1 KiB
Python

# Copyright (c) 2015, Menno Smits
# Released subject to the New BSD License
# Please see http://en.wikipedia.org/wiki/BSD_licenses
from __future__ import unicode_literals
from mock import Mock, patch, sentinel
from ..imapclient import IMAPClient
from .imapclient_test import IMAPClientTest
class TestStarttls(IMAPClientTest):
def setUp(self):
super(TestStarttls, self).setUp()
patcher = patch("imapclient.imapclient.tls")
self.tls = patcher.start()
self.addCleanup(patcher.stop)
self.client._imap.sock = sentinel.old_sock
self.new_sock = Mock()
self.new_sock.makefile.return_value = sentinel.file
self.tls.wrap_socket.return_value = self.new_sock
self.client.host = sentinel.host
self.client.ssl = False
self.client._starttls_done = False
self.client._imap._simple_command.return_value = "OK", [b'start TLS negotiation']
def test_works(self):
resp = self.client.starttls(sentinel.ssl_context)
self.tls.wrap_socket.assert_called_once_with(
sentinel.old_sock,
sentinel.ssl_context,
sentinel.host,
)
self.new_sock.makefile.assert_called_once_with()
self.assertEqual(self.client._imap.file, sentinel.file)
self.assertEqual(resp, b'start TLS negotiation')
def test_command_fails(self):
self.client._imap._simple_command.return_value = "NO", [b'sorry']
with self.assertRaises(IMAPClient.Error) as raised:
self.client.starttls(sentinel.ssl_context)
self.assertEqual(str(raised.exception), "starttls failed: sorry")
def test_fails_if_called_twice(self):
self.client.starttls(sentinel.ssl_context)
self.assert_tls_already_established()
def test_fails_if_ssl_true(self):
self.client.ssl = True
self.assert_tls_already_established()
def assert_tls_already_established(self):
with self.assertRaises(IMAPClient.AbortError) as raised:
self.client.starttls(sentinel.ssl_context)
self.assertEqual(str(raised.exception), "TLS session already established")