Cleanup tests
This commit is contained in:
parent
cbefa18887
commit
783b0a8526
|
@ -17,8 +17,7 @@ class TrolliusEventLoopTests(TestCase):
|
||||||
asyncio.set_event_loop_policy(policy)
|
asyncio.set_event_loop_policy(policy)
|
||||||
trollius.set_event_loop_policy(policy)
|
trollius.set_event_loop_policy(policy)
|
||||||
self.loop = trollius.new_event_loop()
|
self.loop = trollius.new_event_loop()
|
||||||
asyncio.set_event_loop(self.loop)
|
policy.set_event_loop(self.loop)
|
||||||
trollius.set_event_loop(self.loop)
|
|
||||||
|
|
||||||
def tearDown(self):
|
def tearDown(self):
|
||||||
self.loop.close()
|
self.loop.close()
|
||||||
|
|
|
@ -4,12 +4,12 @@
|
||||||
##
|
##
|
||||||
|
|
||||||
|
|
||||||
import greenio
|
|
||||||
import asyncio
|
import asyncio
|
||||||
from unittest import TestCase
|
import greenio
|
||||||
|
import unittest
|
||||||
|
|
||||||
|
|
||||||
class TaskTests(TestCase):
|
class TaskTests(unittest.TestCase):
|
||||||
def setUp(self):
|
def setUp(self):
|
||||||
asyncio.set_event_loop_policy(greenio.GreenEventLoopPolicy())
|
asyncio.set_event_loop_policy(greenio.GreenEventLoopPolicy())
|
||||||
self.loop = asyncio.new_event_loop()
|
self.loop = asyncio.new_event_loop()
|
||||||
|
@ -40,7 +40,7 @@ class TaskTests(TestCase):
|
||||||
self.assertEqual(fut.result(), 42)
|
self.assertEqual(fut.result(), 42)
|
||||||
|
|
||||||
def test_task_yield_from_exception_propagation(self):
|
def test_task_yield_from_exception_propagation(self):
|
||||||
non_local = {'CHK': 0}
|
CHK = 0
|
||||||
|
|
||||||
@asyncio.coroutine
|
@asyncio.coroutine
|
||||||
def bar():
|
def bar():
|
||||||
|
@ -57,10 +57,11 @@ class TaskTests(TestCase):
|
||||||
try:
|
try:
|
||||||
return (yield from foo())
|
return (yield from foo())
|
||||||
except ZeroDivisionError:
|
except ZeroDivisionError:
|
||||||
non_local['CHK'] += 1
|
nonlocal CHK
|
||||||
|
CHK += 1
|
||||||
|
|
||||||
self.loop.run_until_complete(test())
|
self.loop.run_until_complete(test())
|
||||||
self.assertEqual(non_local['CHK'], 1)
|
self.assertEqual(CHK, 1)
|
||||||
|
|
||||||
def test_task_yield_from_coroutine(self):
|
def test_task_yield_from_coroutine(self):
|
||||||
@asyncio.coroutine
|
@asyncio.coroutine
|
||||||
|
|
|
@ -16,11 +16,12 @@ except ImportError:
|
||||||
|
|
||||||
class TrolliusTaskTests(TestCase):
|
class TrolliusTaskTests(TestCase):
|
||||||
def setUp(self):
|
def setUp(self):
|
||||||
trollius.set_event_loop_policy(greenio.GreenTrolliusEventLoopPolicy())
|
policy = greenio.GreenTrolliusEventLoopPolicy()
|
||||||
self.loop = trollius.new_event_loop()
|
trollius.set_event_loop_policy(policy)
|
||||||
trollius.set_event_loop(self.loop)
|
|
||||||
if asyncio is not None:
|
if asyncio is not None:
|
||||||
asyncio.set_event_loop(self.loop)
|
asyncio.set_event_loop_policy(policy)
|
||||||
|
self.loop = trollius.new_event_loop()
|
||||||
|
policy.set_event_loop(self.loop)
|
||||||
|
|
||||||
def tearDown(self):
|
def tearDown(self):
|
||||||
self.loop.close()
|
self.loop.close()
|
||||||
|
|
Loading…
Reference in New Issue