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

Use transient_internet() consistently in test_imaplib

parent b1aef233
...@@ -219,9 +219,11 @@ class RemoteIMAPTest(unittest.TestCase): ...@@ -219,9 +219,11 @@ class RemoteIMAPTest(unittest.TestCase):
def tearDown(self): def tearDown(self):
if self.server is not None: if self.server is not None:
with transient_internet(self.host):
self.server.logout() self.server.logout()
def test_logincapa(self): def test_logincapa(self):
with transient_internet(self.host):
for cap in self.server.capabilities: for cap in self.server.capabilities:
self.assertIsInstance(cap, str) self.assertIsInstance(cap, str)
self.assertTrue('LOGINDISABLED' in self.server.capabilities) self.assertTrue('LOGINDISABLED' in self.server.capabilities)
...@@ -230,6 +232,7 @@ class RemoteIMAPTest(unittest.TestCase): ...@@ -230,6 +232,7 @@ class RemoteIMAPTest(unittest.TestCase):
self.assertEqual(rs[0], 'OK') self.assertEqual(rs[0], 'OK')
def test_logout(self): def test_logout(self):
with transient_internet(self.host):
rs = self.server.logout() rs = self.server.logout()
self.server = None self.server = None
self.assertEqual(rs[0], 'BYE') self.assertEqual(rs[0], 'BYE')
...@@ -240,6 +243,7 @@ class RemoteIMAP_STARTTLSTest(RemoteIMAPTest): ...@@ -240,6 +243,7 @@ class RemoteIMAP_STARTTLSTest(RemoteIMAPTest):
def setUp(self): def setUp(self):
super().setUp() super().setUp()
with transient_internet(self.host):
rs = self.server.starttls() rs = self.server.starttls()
self.assertEqual(rs[0], 'OK') self.assertEqual(rs[0], 'OK')
......
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