#8739: fix omission of DEBUGSTREAM reset in new test in test_smtpd.

This clears up an error in detected by refleak mode that showed up when
test_smtplib was run after test_smtpd in the same refleak run.
This commit is contained in:
R David Murray 2012-06-04 15:55:51 -04:00
parent bd7b5dd816
commit 05cab75d59
1 changed files with 2 additions and 0 deletions

View File

@ -507,6 +507,7 @@ class SMTPDChannelWithDataSizeLimitTest(unittest.TestCase):
def setUp(self): def setUp(self):
smtpd.socket = asyncore.socket = mock_socket smtpd.socket = asyncore.socket = mock_socket
self.old_debugstream = smtpd.DEBUGSTREAM
self.debug = smtpd.DEBUGSTREAM = io.StringIO() self.debug = smtpd.DEBUGSTREAM = io.StringIO()
self.server = DummyServer('a', 'b') self.server = DummyServer('a', 'b')
conn, addr = self.server.accept() conn, addr = self.server.accept()
@ -516,6 +517,7 @@ class SMTPDChannelWithDataSizeLimitTest(unittest.TestCase):
def tearDown(self): def tearDown(self):
asyncore.close_all() asyncore.close_all()
asyncore.socket = smtpd.socket = socket asyncore.socket = smtpd.socket = socket
smtpd.DEBUGSTREAM = self.old_debugstream
def write_line(self, line): def write_line(self, line):
self.channel.socket.queue_recv(line) self.channel.socket.queue_recv(line)