1ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot"""Test script for poplib module.""" 2ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 3ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot# Modified by Giampaolo Rodola' to give poplib.POP3 and poplib.POP3_SSL 4ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot# a real test suite 5ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 6ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robotimport poplib 7ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robotimport asyncore 8ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robotimport asynchat 9ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robotimport socket 10ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robotimport os 11ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robotimport time 12ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robotimport errno 13ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 14ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robotfrom unittest import TestCase 15ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robotfrom test import test_support 16ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robotfrom test.test_support import HOST 17ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robotthreading = test_support.import_module('threading') 18ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 19ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 20ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot# the dummy data returned by server when LIST and RETR commands are issued 21ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team RobotLIST_RESP = '1 1\r\n2 2\r\n3 3\r\n4 4\r\n5 5\r\n.\r\n' 22ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team RobotRETR_RESP = """From: postmaster@python.org\ 23ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot\r\nContent-Type: text/plain\r\n\ 24ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team RobotMIME-Version: 1.0\r\n\ 25ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team RobotSubject: Dummy\r\n\ 26ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot\r\n\ 27ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robotline1\r\n\ 28ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robotline2\r\n\ 29ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robotline3\r\n\ 30ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot.\r\n""" 31ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 32ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 33ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robotclass DummyPOP3Handler(asynchat.async_chat): 34ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 35ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot def __init__(self, conn): 36ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot asynchat.async_chat.__init__(self, conn) 37ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.set_terminator("\r\n") 38ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.in_buffer = [] 39ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.push('+OK dummy pop3 server ready.') 40ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 41ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot def collect_incoming_data(self, data): 42ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.in_buffer.append(data) 43ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 44ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot def found_terminator(self): 45ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot line = ''.join(self.in_buffer) 46ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.in_buffer = [] 47ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot cmd = line.split(' ')[0].lower() 48ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot space = line.find(' ') 49ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot if space != -1: 50ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot arg = line[space + 1:] 51ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot else: 52ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot arg = "" 53ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot if hasattr(self, 'cmd_' + cmd): 54ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot method = getattr(self, 'cmd_' + cmd) 55ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot method(arg) 56ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot else: 57ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.push('-ERR unrecognized POP3 command "%s".' %cmd) 58ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 59ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot def handle_error(self): 60ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot raise 61ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 62ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot def push(self, data): 63ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot asynchat.async_chat.push(self, data + '\r\n') 64ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 65ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot def cmd_echo(self, arg): 66ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot # sends back the received string (used by the test suite) 67ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.push(arg) 68ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 69ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot def cmd_user(self, arg): 70ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot if arg != "guido": 71ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.push("-ERR no such user") 72ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.push('+OK password required') 73ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 74ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot def cmd_pass(self, arg): 75ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot if arg != "python": 76ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.push("-ERR wrong password") 77ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.push('+OK 10 messages') 78ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 79ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot def cmd_stat(self, arg): 80ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.push('+OK 10 100') 81ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 82ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot def cmd_list(self, arg): 83ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot if arg: 84ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.push('+OK %s %s' %(arg, arg)) 85ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot else: 86ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.push('+OK') 87ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot asynchat.async_chat.push(self, LIST_RESP) 88ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 89ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot cmd_uidl = cmd_list 90ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 91ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot def cmd_retr(self, arg): 92ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.push('+OK %s bytes' %len(RETR_RESP)) 93ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot asynchat.async_chat.push(self, RETR_RESP) 94ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 95ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot cmd_top = cmd_retr 96ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 97ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot def cmd_dele(self, arg): 98ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.push('+OK message marked for deletion.') 99ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 100ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot def cmd_noop(self, arg): 101ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.push('+OK done nothing.') 102ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 103ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot def cmd_rpop(self, arg): 104ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.push('+OK done nothing.') 105ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 106ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 107ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robotclass DummyPOP3Server(asyncore.dispatcher, threading.Thread): 108ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 109ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot handler = DummyPOP3Handler 110ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 111ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot def __init__(self, address, af=socket.AF_INET): 112ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot threading.Thread.__init__(self) 113ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot asyncore.dispatcher.__init__(self) 114ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.create_socket(af, socket.SOCK_STREAM) 115ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.bind(address) 116ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.listen(5) 117ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.active = False 118ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.active_lock = threading.Lock() 119ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.host, self.port = self.socket.getsockname()[:2] 120ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 121ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot def start(self): 122ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot assert not self.active 123ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.__flag = threading.Event() 124ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot threading.Thread.start(self) 125ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.__flag.wait() 126ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 127ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot def run(self): 128ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.active = True 129ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.__flag.set() 130ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot while self.active and asyncore.socket_map: 131ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.active_lock.acquire() 132ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot asyncore.loop(timeout=0.1, count=1) 133ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.active_lock.release() 134ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot asyncore.close_all(ignore_all=True) 135ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 136ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot def stop(self): 137ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot assert self.active 138ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.active = False 139ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.join() 140ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 141ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot def handle_accept(self): 142ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot conn, addr = self.accept() 143ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.handler = self.handler(conn) 144ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.close() 145ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 146ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot def handle_connect(self): 147ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.close() 148ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot handle_read = handle_connect 149ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 150ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot def writable(self): 151ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot return 0 152ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 153ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot def handle_error(self): 154ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot raise 155ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 156ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 157ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robotclass TestPOP3Class(TestCase): 158ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 159ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot def assertOK(self, resp): 160ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.assertTrue(resp.startswith("+OK")) 161ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 162ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot def setUp(self): 163ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.server = DummyPOP3Server((HOST, 0)) 164ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.server.start() 165ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.client = poplib.POP3(self.server.host, self.server.port) 166ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 167ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot def tearDown(self): 168ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.client.quit() 169ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.server.stop() 170ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 171ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot def test_getwelcome(self): 172ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.assertEqual(self.client.getwelcome(), '+OK dummy pop3 server ready.') 173ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 174ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot def test_exceptions(self): 175ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.assertRaises(poplib.error_proto, self.client._shortcmd, 'echo -err') 176ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 177ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot def test_user(self): 178ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.assertOK(self.client.user('guido')) 179ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.assertRaises(poplib.error_proto, self.client.user, 'invalid') 180ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 181ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot def test_pass_(self): 182ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.assertOK(self.client.pass_('python')) 183ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.assertRaises(poplib.error_proto, self.client.user, 'invalid') 184ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 185ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot def test_stat(self): 186ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.assertEqual(self.client.stat(), (10, 100)) 187ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 188ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot def test_list(self): 189ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.assertEqual(self.client.list()[1:], 190ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot (['1 1', '2 2', '3 3', '4 4', '5 5'], 25)) 191ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.assertTrue(self.client.list('1').endswith("OK 1 1")) 192ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 193ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot def test_retr(self): 194ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot expected = ('+OK 116 bytes', 195ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot ['From: postmaster@python.org', 'Content-Type: text/plain', 196ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 'MIME-Version: 1.0', 'Subject: Dummy', 197ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot '', 'line1', 'line2', 'line3'], 198ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 113) 199ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.assertEqual(self.client.retr('foo'), expected) 200ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 201ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot def test_dele(self): 202ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.assertOK(self.client.dele('foo')) 203ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 204ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot def test_noop(self): 205ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.assertOK(self.client.noop()) 206ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 207ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot def test_rpop(self): 208ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.assertOK(self.client.rpop('foo')) 209ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 210ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot def test_top(self): 211ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot expected = ('+OK 116 bytes', 212ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot ['From: postmaster@python.org', 'Content-Type: text/plain', 213ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 'MIME-Version: 1.0', 'Subject: Dummy', '', 214ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 'line1', 'line2', 'line3'], 215ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 113) 216ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.assertEqual(self.client.top(1, 1), expected) 217ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 218ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot def test_uidl(self): 219ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.client.uidl() 220ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.client.uidl('foo') 221ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 222ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 223ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team RobotSUPPORTS_SSL = False 224ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robotif hasattr(poplib, 'POP3_SSL'): 225ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot import ssl 226ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 227ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot SUPPORTS_SSL = True 228ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot CERTFILE = os.path.join(os.path.dirname(__file__) or os.curdir, "keycert.pem") 229ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 230ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot class DummyPOP3_SSLHandler(DummyPOP3Handler): 231ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 232ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot def __init__(self, conn): 233ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot asynchat.async_chat.__init__(self, conn) 234ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.socket = ssl.wrap_socket(self.socket, certfile=CERTFILE, 235ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot server_side=True, 236ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot do_handshake_on_connect=False) 237ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot # Must try handshake before calling push() 238ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self._ssl_accepting = True 239ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self._do_ssl_handshake() 240ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.set_terminator("\r\n") 241ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.in_buffer = [] 242ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.push('+OK dummy pop3 server ready.') 243ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 244ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot def _do_ssl_handshake(self): 245ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot try: 246ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.socket.do_handshake() 247ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot except ssl.SSLError, err: 248ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot if err.args[0] in (ssl.SSL_ERROR_WANT_READ, 249ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot ssl.SSL_ERROR_WANT_WRITE): 250ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot return 251ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot elif err.args[0] == ssl.SSL_ERROR_EOF: 252ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot return self.handle_close() 253ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot raise 254ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot except socket.error, err: 255ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot if err.args[0] == errno.ECONNABORTED: 256ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot return self.handle_close() 257ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot else: 258ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self._ssl_accepting = False 259ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 260ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot def handle_read(self): 261ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot if self._ssl_accepting: 262ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self._do_ssl_handshake() 263ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot else: 264ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot DummyPOP3Handler.handle_read(self) 265ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 266ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot class TestPOP3_SSLClass(TestPOP3Class): 267ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot # repeat previous tests by using poplib.POP3_SSL 268ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 269ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot def setUp(self): 270ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.server = DummyPOP3Server((HOST, 0)) 271ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.server.handler = DummyPOP3_SSLHandler 272ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.server.start() 273ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.client = poplib.POP3_SSL(self.server.host, self.server.port) 274ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 275ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot def test__all__(self): 276ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.assertIn('POP3_SSL', poplib.__all__) 277ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 278ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 279ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robotclass TestTimeouts(TestCase): 280ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 281ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot def setUp(self): 282ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.evt = threading.Event() 283ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM) 284ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.sock.settimeout(60) # Safety net. Look issue 11812 285ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.port = test_support.bind_port(self.sock) 286ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.thread = threading.Thread(target=self.server, args=(self.evt,self.sock)) 287ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.thread.setDaemon(True) 288ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.thread.start() 289ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.evt.wait() 290ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 291ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot def tearDown(self): 292ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.thread.join() 293ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot del self.thread # Clear out any dangling Thread objects. 294ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 295ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot def server(self, evt, serv): 296ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot serv.listen(5) 297ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot evt.set() 298ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot try: 299ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot conn, addr = serv.accept() 300ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot conn.send("+ Hola mundo\n") 301ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot conn.close() 302ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot except socket.timeout: 303ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot pass 304ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot finally: 305ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot serv.close() 306ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 307ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot def testTimeoutDefault(self): 308ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.assertTrue(socket.getdefaulttimeout() is None) 309ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot socket.setdefaulttimeout(30) 310ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot try: 311ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot pop = poplib.POP3(HOST, self.port) 312ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot finally: 313ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot socket.setdefaulttimeout(None) 314ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.assertEqual(pop.sock.gettimeout(), 30) 315ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot pop.sock.close() 316ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 317ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot def testTimeoutNone(self): 318ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.assertTrue(socket.getdefaulttimeout() is None) 319ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot socket.setdefaulttimeout(30) 320ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot try: 321ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot pop = poplib.POP3(HOST, self.port, timeout=None) 322ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot finally: 323ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot socket.setdefaulttimeout(None) 324ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.assertTrue(pop.sock.gettimeout() is None) 325ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot pop.sock.close() 326ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 327ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot def testTimeoutValue(self): 328ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot pop = poplib.POP3(HOST, self.port, timeout=30) 329ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot self.assertEqual(pop.sock.gettimeout(), 30) 330ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot pop.sock.close() 331ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 332ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 333ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robotdef test_main(): 334ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot tests = [TestPOP3Class, TestTimeouts] 335ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot if SUPPORTS_SSL: 336ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot tests.append(TestPOP3_SSLClass) 337ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot thread_info = test_support.threading_setup() 338ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot try: 339ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot test_support.run_unittest(*tests) 340ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot finally: 341ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot test_support.threading_cleanup(*thread_info) 342ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 343ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot 344ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robotif __name__ == '__main__': 345ec1a0b3abe08fb9a3952e8f48231cda1f6d9b1fandroid-build-team Robot test_main() 346