def test_reconfigure_resolver(self):
with unittest.mock.patch("dns.resolver.Resolver") as Resolver:
network.reconfigure_resolver()
Resolver.assert_called_with()
self.assertEqual(network.get_resolver(), Resolver())
def setUp(self):
self.tlr = MockResolver(self)
base = unittest.mock.Mock()
self.base = base
self.run_in_executor = unittest.mock.Mock()
@asyncio.coroutine
def run_in_executor(executor, func, *args):
self.run_in_executor(executor, func, *args)
return func(*args)
self.patches = [
unittest.mock.patch(
"aioxmpp.network.get_resolver",
new=base.get_resolver,
),
unittest.mock.patch(
"aioxmpp.network.reconfigure_resolver",
new=base.reconfigure_resolver,
),
unittest.mock.patch.object(asyncio.get_event_loop(),
"run_in_executor",
new=run_in_executor)
]
self.answer = MockAnswer([])
# ensure consistent state
network.reconfigure_resolver()
for patch in self.patches:
patch.start()
base.get_resolver.return_value = self.tlr
def setUp(self):
self.tlr = MockResolver(self)
base = unittest.mock.Mock()
self.base = base
self.run_in_executor = unittest.mock.Mock()
@asyncio.coroutine
def run_in_executor(executor, func, *args):
self.run_in_executor(executor, func, *args)
return func(*args)
self.patches = [
unittest.mock.patch(
"aioxmpp.network.get_resolver",
new=base.get_resolver,
),
unittest.mock.patch(
"aioxmpp.network.reconfigure_resolver",
new=base.reconfigure_resolver,
),
unittest.mock.patch.object(
asyncio.get_event_loop(),
"run_in_executor",
new=run_in_executor
)
]
self.answer = MockAnswer([])
# ensure consistent state
network.reconfigure_resolver()
for patch in self.patches:
patch.start()
base.get_resolver.return_value = self.tlr
def tearDown(self):
for patch in self.patches:
patch.stop()
# ensure consistent state
network.reconfigure_resolver()
def tearDown(self):
network.reconfigure_resolver()
def tearDown(self):
network.reconfigure_resolver()
