REDROOM

Navigation

Breadcrumbs

/ / lib / python3 / dist-packages / twisted / test

Create

Create File | Create Folder | Upload Files | Run Command

Edit File Size: 6.38 KB

//lib/python3/dist-packages/twisted/test/test_monkey.py

Text Base64

# Copyright (c) Twisted Matrix Laboratories.
# See LICENSE for details.

"""
Tests for L{twisted.python.monkey}.
"""
from __future__ import annotations

from typing import NoReturn

from twisted.python.monkey import MonkeyPatcher
from twisted.trial import unittest

class TestObj:
    def __init__(self) -> None:
        self.foo = "foo value"
        self.bar = "bar value"
        self.baz = "baz value"

class MonkeyPatcherTests(unittest.SynchronousTestCase):
    """
    Tests for L{MonkeyPatcher} monkey-patching class.
    """

def setUp(self) -> None:
        self.testObject = TestObj()
        self.originalObject = TestObj()
        self.monkeyPatcher = MonkeyPatcher()

def test_empty(self) -> None:
        """
        A monkey patcher without patches shouldn't change a thing.
        """
        self.monkeyPatcher.patch()

# We can't assert that all state is unchanged, but at least we can
        # check our test object.
        self.assertEqual(self.originalObject.foo, self.testObject.foo)
        self.assertEqual(self.originalObject.bar, self.testObject.bar)
        self.assertEqual(self.originalObject.baz, self.testObject.baz)

def test_constructWithPatches(self) -> None:
        """
        Constructing a L{MonkeyPatcher} with patches should add all of the
        given patches to the patch list.
        """
        patcher = MonkeyPatcher(
            (self.testObject, "foo", "haha"), (self.testObject, "bar", "hehe")
        )
        patcher.patch()
        self.assertEqual("haha", self.testObject.foo)
        self.assertEqual("hehe", self.testObject.bar)
        self.assertEqual(self.originalObject.baz, self.testObject.baz)

def test_patchExisting(self) -> None:
        """
        Patching an attribute that exists sets it to the value defined in the
        patch.
        """
        self.monkeyPatcher.addPatch(self.testObject, "foo", "haha")
        self.monkeyPatcher.patch()
        self.assertEqual(self.testObject.foo, "haha")

def test_patchNonExisting(self) -> None:
        """
        Patching a non-existing attribute fails with an C{AttributeError}.
        """
        self.monkeyPatcher.addPatch(self.testObject, "nowhere", "blow up please")
        self.assertRaises(AttributeError, self.monkeyPatcher.patch)

def test_patchAlreadyPatched(self) -> None:
        """
        Adding a patch for an object and attribute that already have a patch
        overrides the existing patch.
        """
        self.monkeyPatcher.addPatch(self.testObject, "foo", "blah")
        self.monkeyPatcher.addPatch(self.testObject, "foo", "BLAH")
        self.monkeyPatcher.patch()
        self.assertEqual(self.testObject.foo, "BLAH")
        self.monkeyPatcher.restore()
        self.assertEqual(self.testObject.foo, self.originalObject.foo)

def test_restoreTwiceIsANoOp(self) -> None:
        """
        Restoring an already-restored monkey patch is a no-op.
        """
        self.monkeyPatcher.addPatch(self.testObject, "foo", "blah")
        self.monkeyPatcher.patch()
        self.monkeyPatcher.restore()
        self.assertEqual(self.testObject.foo, self.originalObject.foo)
        self.monkeyPatcher.restore()
        self.assertEqual(self.testObject.foo, self.originalObject.foo)

def test_runWithPatchesDecoration(self) -> None:
        """
        runWithPatches should run the given callable, passing in all arguments
        and keyword arguments, and return the return value of the callable.
        """
        log: list[tuple[int, int, int | None]] = []

def f(a: int, b: int, c: int | None = None) -> str:
            log.append((a, b, c))
            return "foo"

result = self.monkeyPatcher.runWithPatches(f, 1, 2, c=10)
        self.assertEqual("foo", result)
        self.assertEqual([(1, 2, 10)], log)

def test_repeatedRunWithPatches(self) -> None:
        """
        We should be able to call the same function with runWithPatches more
        than once. All patches should apply for each call.
        """

def f() -> tuple[str, str, str]:
            return (self.testObject.foo, self.testObject.bar, self.testObject.baz)

self.monkeyPatcher.addPatch(self.testObject, "foo", "haha")
        result = self.monkeyPatcher.runWithPatches(f)
        self.assertEqual(
            ("haha", self.originalObject.bar, self.originalObject.baz), result
        )
        result = self.monkeyPatcher.runWithPatches(f)
        self.assertEqual(
            ("haha", self.originalObject.bar, self.originalObject.baz), result
        )

def test_runWithPatchesRestores(self) -> None:
        """
        C{runWithPatches} should restore the original values after the function
        has executed.
        """
        self.monkeyPatcher.addPatch(self.testObject, "foo", "haha")
        self.assertEqual(self.originalObject.foo, self.testObject.foo)
        self.monkeyPatcher.runWithPatches(lambda: None)
        self.assertEqual(self.originalObject.foo, self.testObject.foo)

def test_runWithPatchesRestoresOnException(self) -> None:
        """
        Test runWithPatches restores the original values even when the function
        raises an exception.
        """

def _() -> NoReturn:
            self.assertEqual(self.testObject.foo, "haha")
            self.assertEqual(self.testObject.bar, "blahblah")
            raise RuntimeError("Something went wrong!")

self.monkeyPatcher.addPatch(self.testObject, "foo", "haha")
        self.monkeyPatcher.addPatch(self.testObject, "bar", "blahblah")

self.assertRaises(RuntimeError, self.monkeyPatcher.runWithPatches, _)
        self.assertEqual(self.testObject.foo, self.originalObject.foo)
        self.assertEqual(self.testObject.bar, self.originalObject.bar)

def test_contextManager(self) -> None:
        """
        L{MonkeyPatcher} is a context manager that applies its patches on
        entry and restore original values on exit.
        """
        self.monkeyPatcher.addPatch(self.testObject, "foo", "patched value")
        with self.monkeyPatcher:
            self.assertEqual(self.testObject.foo, "patched value")
        self.assertEqual(self.testObject.foo, self.originalObject.foo)

def test_contextManagerPropagatesExceptions(self) -> None:
        """
        Exceptions propagate through the L{MonkeyPatcher} context-manager
        exit method.
        """
        with self.assertRaises(RuntimeError):
            with self.monkeyPatcher:
                raise RuntimeError("something")

Exit & Reset Text mode: syntax highlighting auto-detects file type.

Directory Contents

Dirs: 1 × Files: 104

Sort:

Enable drag-to-move

Name	Size	Perms	Modified	Actions
__pycache__ DIR	-	drwxr-xr-x	2026-01-08 12:56:24	Edit Download recursive recursive
cert.pem.no_trailing_newline	1.46 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
crash_test_dummy.py	654 B	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
ignore_test_failure.py	34.88 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
iosim.py	18.40 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
key.pem.no_trailing_newline	1.63 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
mock_win32process.py	1.25 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
myrebuilder1.py	172 B	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
myrebuilder2.py	172 B	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
plugin_basic.py	925 B	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
plugin_extra1.py	400 B	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
plugin_extra2.py	566 B	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
process_cmdline.py	123 B	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
process_echoer.py	214 B	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
process_fds.py	983 B	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
process_getargv.py	233 B	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
process_getenv.py	268 B	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
process_linger.py	297 B	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
process_reader.py	178 B	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
process_signal.py	220 B	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
process_stdinreader.py	739 B	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
process_tester.py	787 B	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
process_tty.py	130 B	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
process_twisted.py	1.15 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
proto_helpers.py	1.34 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
reflect_helper_IE.py	60 B	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
reflect_helper_VE.py	81 B	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
reflect_helper_ZDE.py	48 B	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
server.pem	5.23 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
ssl_helpers.py	1.72 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
stdio_test_consumer.py	1.14 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
stdio_test_halfclose.py	2.00 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
stdio_test_hostpeer.py	1.06 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
stdio_test_lastwrite.py	1.13 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
stdio_test_loseconn.py	1.55 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
stdio_test_producer.py	1.45 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
stdio_test_write.py	902 B	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
stdio_test_writeseq.py	894 B	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
testutils.py	5.06 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
test_abstract.py	3.66 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
test_adbapi.py	25.47 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
test_amp.py	108.04 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
test_application.py	33.34 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
test_compat.py	17.75 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
test_context.py	1.43 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
test_cooperator.py	20.84 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
test_defer.py	143.37 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
test_defgen.py	13.01 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
test_dirdbm.py	6.90 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
test_error.py	9.60 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
test_factories.py	4.46 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
test_fdesc.py	7.28 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
test_finger.py	1.89 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
test_formmethod.py	4.27 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
test_ftp.py	126.96 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
test_ftp_options.py	2.65 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
test_htb.py	3.19 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
test_ident.py	6.56 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
test_internet.py	45.38 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
test_iosim.py	9.58 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
test_iutils.py	13.31 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
test_lockfile.py	15.54 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
test_log.py	36.86 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
test_logfile.py	17.79 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
test_loopback.py	13.99 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
test_main.py	2.12 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
test_memcache.py	24.69 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
test_modules.py	17.84 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
test_monkey.py	6.38 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
test_paths.py	73.64 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
test_pcp.py	12.23 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
test_persisted.py	14.73 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
test_plugin.py	26.02 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
test_policies.py	32.28 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
test_postfix.py	4.32 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
test_process.py	86.29 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
test_protocols.py	7.16 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
test_randbytes.py	3.63 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
test_rebuild.py	7.40 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
test_reflect.py	23.89 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
test_roots.py	1.65 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
test_shortcut.py	1.91 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
test_sip.py	24.90 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
test_sob.py	5.53 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
test_socks.py	17.09 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
test_ssl.py	22.73 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
test_sslverify.py	113.84 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
test_stateful.py	1.97 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
test_stdio.py	12.43 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
test_strerror.py	5.10 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
test_strports.py	1.67 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
test_task.py	47.73 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
test_tcp.py	64.26 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
test_tcp_internals.py	12.73 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
test_text.py	6.47 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
test_threadable.py	3.26 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
test_threadpool.py	21.64 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
test_threads.py	12.90 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
test_tpfile.py	1.69 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
test_twistd.py	72.29 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
test_twisted.py	6.13 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
test_udp.py	26.79 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
test_unix.py	13.26 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
test_usage.py	22.76 KB	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive
__init__.py	475 B	lrw-r--r--	2024-08-27 10:30:39	Edit Download recursive recursive