blob: dd5086b366b4be89ea65e5dc920a1ac83724cb19 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
|
import aioeventlet
import sys
try:
import asyncio
except ImportError:
import trollius as asyncio
try:
# On Python 2.6, unittest2 is needed to get new features like addCleanup()
import unittest2
unittest = unittest2
except ImportError:
import unittest
unittest2 = None
try:
from unittest import mock
except ImportError:
import mock
class TestCase(unittest.TestCase):
def setUp(self):
policy = aioeventlet.EventLoopPolicy()
asyncio.set_event_loop_policy(policy)
self.addCleanup(asyncio.set_event_loop_policy, None)
self.loop = policy.get_event_loop()
self.addCleanup(self.loop.close)
self.addCleanup(asyncio.set_event_loop, None)
if sys.version_info < (3,) and unittest2 is None:
def assertRaisesRegex(self, *args):
return self.assertRaisesRegexp(*args)
|