mirror of
https://github.com/home-assistant/core.git
synced 2025-04-25 09:47:52 +00:00
Remove pytest warning message (#21713)
This commit is contained in:
parent
1891d5bf22
commit
ba70459e1e
@ -1,5 +1,6 @@
|
|||||||
"""Tests for async util methods from Python source."""
|
"""Tests for async util methods from Python source."""
|
||||||
import asyncio
|
import asyncio
|
||||||
|
import sys
|
||||||
from unittest.mock import MagicMock, patch
|
from unittest.mock import MagicMock, patch
|
||||||
from unittest import TestCase
|
from unittest import TestCase
|
||||||
|
|
||||||
@ -144,7 +145,11 @@ class RunThreadsafeTests(TestCase):
|
|||||||
"""Wait 0.05 second and return a + b."""
|
"""Wait 0.05 second and return a + b."""
|
||||||
yield from asyncio.sleep(0.05, loop=self.loop)
|
yield from asyncio.sleep(0.05, loop=self.loop)
|
||||||
if cancel:
|
if cancel:
|
||||||
asyncio.tasks.Task.current_task(self.loop).cancel()
|
if sys.version_info[:2] >= (3, 7):
|
||||||
|
current_task = asyncio.current_task
|
||||||
|
else:
|
||||||
|
current_task = asyncio.tasks.Task.current_task
|
||||||
|
current_task(self.loop).cancel()
|
||||||
yield
|
yield
|
||||||
return self.add_callback(a, b, fail, invalid)
|
return self.add_callback(a, b, fail, invalid)
|
||||||
|
|
||||||
@ -205,7 +210,11 @@ class RunThreadsafeTests(TestCase):
|
|||||||
self.loop.run_until_complete(future)
|
self.loop.run_until_complete(future)
|
||||||
self.run_briefly(self.loop)
|
self.run_briefly(self.loop)
|
||||||
# Check that there's no pending task (add has been cancelled)
|
# Check that there's no pending task (add has been cancelled)
|
||||||
for task in asyncio.Task.all_tasks(self.loop):
|
if sys.version_info[:2] >= (3, 7):
|
||||||
|
all_tasks = asyncio.all_tasks
|
||||||
|
else:
|
||||||
|
all_tasks = asyncio.Task.all_tasks
|
||||||
|
for task in all_tasks(self.loop):
|
||||||
self.assertTrue(task.done())
|
self.assertTrue(task.done())
|
||||||
|
|
||||||
def test_run_coroutine_threadsafe_task_cancelled(self):
|
def test_run_coroutine_threadsafe_task_cancelled(self):
|
||||||
|
Loading…
x
Reference in New Issue
Block a user