Module rsyscall.tests.trio_test_case

A trio-enabled variant of unittest.TestCase

"A trio-enabled variant of unittest.TestCase"
import trio
import unittest
import functools
import types
from trio._core._run import Nursery

class TrioTestCase(unittest.TestCase):
    "A trio-enabled variant of unittest.TestCase"
    nursery: Nursery

    async def asyncSetUp(self) -> None:
        "Asynchronously set up resources for tests in this TestCase"
        pass

    async def asyncTearDown(self) -> None:
        "Asynchronously clean up resources for tests in this TestCase"
        pass

    def __init__(self, methodName='runTest') -> None:
        test = getattr(type(self), methodName)
        @functools.wraps(test)
        async def test_with_setup() -> None:
            async with trio.open_nursery() as nursery:
                self.nursery = nursery
                await self.asyncSetUp()
                try:
                    await test(self)
                except BaseException as exn:
                    try:
                        await self.asyncTearDown()
                    except BaseException as teardown_exn:
                        # have to merge the exceptions if they both throw;
                        # might as well do this with trio.MultiError since we have it
                        raise trio.MultiError([exn, teardown_exn])
                    else:
                        raise
                else:
                    await self.asyncTearDown()
                nursery.cancel_scope.cancel()
        @functools.wraps(test_with_setup)
        def sync_test_with_setup(self) -> None:
            trio.run(test_with_setup)
        setattr(self, methodName, types.MethodType(sync_test_with_setup, self))
        super().__init__(methodName)

Classes

class TrioTestCase (methodName='runTest')

A trio-enabled variant of unittest.TestCase

Create an instance of the class that will use the named test method when executed. Raises a ValueError if the instance does not have a method with the specified name.

Expand source code Browse git

class TrioTestCase(unittest.TestCase):
    "A trio-enabled variant of unittest.TestCase"
    nursery: Nursery

    async def asyncSetUp(self) -> None:
        "Asynchronously set up resources for tests in this TestCase"
        pass

    async def asyncTearDown(self) -> None:
        "Asynchronously clean up resources for tests in this TestCase"
        pass

    def __init__(self, methodName='runTest') -> None:
        test = getattr(type(self), methodName)
        @functools.wraps(test)
        async def test_with_setup() -> None:
            async with trio.open_nursery() as nursery:
                self.nursery = nursery
                await self.asyncSetUp()
                try:
                    await test(self)
                except BaseException as exn:
                    try:
                        await self.asyncTearDown()
                    except BaseException as teardown_exn:
                        # have to merge the exceptions if they both throw;
                        # might as well do this with trio.MultiError since we have it
                        raise trio.MultiError([exn, teardown_exn])
                    else:
                        raise
                else:
                    await self.asyncTearDown()
                nursery.cancel_scope.cancel()
        @functools.wraps(test_with_setup)
        def sync_test_with_setup(self) -> None:
            trio.run(test_with_setup)
        setattr(self, methodName, types.MethodType(sync_test_with_setup, self))
        super().__init__(methodName)

Ancestors

• unittest.case.TestCase

Subclasses

• TestBenchmark
• TestCat
• TestChroot
• TestClone
• TestCloneUnshareFiles
• TestConcur
• TestConcurrency
• TestEpoller
• TestEventfd
• TestFS
• TestFS
• TestFUSE
• TestInotify
• TestIP
• TestMisc
• TestNet
• TestNix
• TestPersistent
• TestPidns
• TestPipe
• TestPointer
• TestUser
• TestProc
• TestREPL
• TestResource
• TestEventfd
• TestSeek
• TestSignal
• TestSocket
• TestSSH
• TestStdinboot
• TestStub
• TestTimerfd
• TestUser

Class variables

var nursery : trio.Nursery

Methods

async def asyncSetUp(self) ‑> NoneType

Asynchronously set up resources for tests in this TestCase

Expand source code Browse git

async def asyncSetUp(self) -> None:
    "Asynchronously set up resources for tests in this TestCase"
    pass

async def asyncTearDown(self) ‑> NoneType

Asynchronously clean up resources for tests in this TestCase

Expand source code Browse git

async def asyncTearDown(self) -> None:
    "Asynchronously clean up resources for tests in this TestCase"
    pass