shell bypass 403
UnknownSec Shell
:
/
lib
/
python3
/
dist-packages
/
twisted
/
test
/ [
drwxr-xr-x
]
upload
mass deface
mass delete
console
info server
name :
test_monkey.py
# 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")
© 2026 UnknownSec