diff options
Diffstat (limited to 'sources/pyside6/tests/QtAsyncio/qasyncio_test.py')
-rw-r--r-- | sources/pyside6/tests/QtAsyncio/qasyncio_test.py | 51 |
1 files changed, 51 insertions, 0 deletions
diff --git a/sources/pyside6/tests/QtAsyncio/qasyncio_test.py b/sources/pyside6/tests/QtAsyncio/qasyncio_test.py new file mode 100644 index 000000000..f3c971285 --- /dev/null +++ b/sources/pyside6/tests/QtAsyncio/qasyncio_test.py @@ -0,0 +1,51 @@ +# Copyright (C) 2023 The Qt Company Ltd. +# SPDX-License-Identifier: LicenseRef-Qt-Commercial OR GPL-3.0-only WITH Qt-GPL-exception-1.0 + +'''Test cases for QtAsyncio''' + +import unittest +import asyncio + +from PySide6.QtAsyncio import QAsyncioEventLoopPolicy + + +class QAsyncioTestCase(unittest.TestCase): + async def sleep(self, output): + output += "Hello" + await asyncio.sleep(0.2) + output += "World" + + async def gather(self, output): + await asyncio.gather(self.sleep(output), self.sleep(output), self.sleep(output)) + + def test_sleep(self): + outputs_expected = [] + outputs_real = [] + + # Run the code without QAsyncioEventLoopPolicy + asyncio.set_event_loop_policy(asyncio.DefaultEventLoopPolicy()) + asyncio.run(self.sleep(outputs_expected)) + + # Run the code with QAsyncioEventLoopPolicy and QtEventLoop + asyncio.set_event_loop_policy(QAsyncioEventLoopPolicy()) + asyncio.run(self.sleep(outputs_real)) + + self.assertEqual(outputs_expected, outputs_real) + + def test_gather(self): + outputs_expected = [] + outputs_real = [] + + # Run the code without QAsyncioEventLoopPolicy + asyncio.set_event_loop_policy(asyncio.DefaultEventLoopPolicy()) + asyncio.run(self.gather(outputs_expected)) + + # Run the code with QAsyncioEventLoopPolicy and QtEventLoop + asyncio.set_event_loop_policy(QAsyncioEventLoopPolicy()) + asyncio.run(self.gather(outputs_real)) + + self.assertEqual(outputs_expected, outputs_real) + + +if __name__ == '__main__': + unittest.main() |