Commit 924cbea4 authored by Antoine Pitrou's avatar Antoine Pitrou

Use transient_internet() consistently in test_imaplib

parent b1aef233
...@@ -219,20 +219,23 @@ class RemoteIMAPTest(unittest.TestCase): ...@@ -219,20 +219,23 @@ class RemoteIMAPTest(unittest.TestCase):
def tearDown(self): def tearDown(self):
if self.server is not None: if self.server is not None:
self.server.logout() with transient_internet(self.host):
self.server.logout()
def test_logincapa(self): def test_logincapa(self):
for cap in self.server.capabilities: with transient_internet(self.host):
self.assertIsInstance(cap, str) for cap in self.server.capabilities:
self.assertTrue('LOGINDISABLED' in self.server.capabilities) self.assertIsInstance(cap, str)
self.assertTrue('AUTH=ANONYMOUS' in self.server.capabilities) self.assertTrue('LOGINDISABLED' in self.server.capabilities)
rs = self.server.login(self.username, self.password) self.assertTrue('AUTH=ANONYMOUS' in self.server.capabilities)
self.assertEqual(rs[0], 'OK') rs = self.server.login(self.username, self.password)
self.assertEqual(rs[0], 'OK')
def test_logout(self): def test_logout(self):
rs = self.server.logout() with transient_internet(self.host):
self.server = None rs = self.server.logout()
self.assertEqual(rs[0], 'BYE') self.server = None
self.assertEqual(rs[0], 'BYE')
@unittest.skipUnless(ssl, "SSL not available") @unittest.skipUnless(ssl, "SSL not available")
...@@ -240,8 +243,9 @@ class RemoteIMAP_STARTTLSTest(RemoteIMAPTest): ...@@ -240,8 +243,9 @@ class RemoteIMAP_STARTTLSTest(RemoteIMAPTest):
def setUp(self): def setUp(self):
super().setUp() super().setUp()
rs = self.server.starttls() with transient_internet(self.host):
self.assertEqual(rs[0], 'OK') rs = self.server.starttls()
self.assertEqual(rs[0], 'OK')
def test_logincapa(self): def test_logincapa(self):
for cap in self.server.capabilities: for cap in self.server.capabilities:
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment