Editing: test_extension_interface.py

import sys import greenlet from . import _test_extension from . import TestCase from .leakcheck import ignores_leakcheck # pylint:disable=c-extension-no-member class CAPITests(TestCase): def test_switch(self): self.assertEqual( 50, _test_extension.test_switch(greenlet.greenlet(lambda: 50))) def test_switch_kwargs(self): def adder(x, y): return x * y g = greenlet.greenlet(adder) self.assertEqual(6, _test_extension.test_switch_kwargs(g, x=3, y=2)) with self.assertRaisesRegex(TypeError, "argument 1 must be greenlet"): _test_extension.test_switch_kwargs("not a greenlet") def test_setparent(self): # pylint:disable=disallowed-name def foo(): def bar(): greenlet.getcurrent().parent.switch() # This final switch should go back to the main greenlet, since # the test_setparent() function in the C extension should have # reparented this greenlet. greenlet.getcurrent().parent.switch() raise AssertionError("Should never have reached this code") child = greenlet.greenlet(bar) child.switch() greenlet.getcurrent().parent.switch(child) greenlet.getcurrent().parent.throw( AssertionError("Should never reach this code")) foo_child = greenlet.greenlet(foo).switch() self.assertEqual(None, _test_extension.test_setparent(foo_child)) def test_getcurrent(self): _test_extension.test_getcurrent() def test_new_greenlet(self): self.assertEqual(-15, _test_extension.test_new_greenlet(lambda: -15)) def test_raise_greenlet_dead(self): self.assertRaises( greenlet.GreenletExit, _test_extension.test_raise_dead_greenlet) def test_raise_greenlet_error(self): self.assertRaises( greenlet.error, _test_extension.test_raise_greenlet_error) def test_throw(self): seen = [] def foo(): # pylint:disable=disallowed-name try: greenlet.getcurrent().parent.switch() except ValueError: seen.append(sys.exc_info()[1]) except greenlet.GreenletExit: raise AssertionError g = greenlet.greenlet(foo) g.switch() _test_extension.test_throw(g) self.assertEqual(len(seen), 1) self.assertTrue( isinstance(seen[0], ValueError), "ValueError was not raised in foo()") self.assertEqual( str(seen[0]), 'take that sucka!', "message doesn't match") def test_non_traceback_param(self): with self.assertRaises(TypeError) as exc: _test_extension.test_throw_exact( greenlet.getcurrent(), Exception, Exception(), self ) self.assertEqual(str(exc.exception), "throw() third argument must be a traceback object") def test_instance_of_wrong_type(self): with self.assertRaises(TypeError) as exc: _test_extension.test_throw_exact( greenlet.getcurrent(), Exception(), BaseException(), None, ) self.assertEqual(str(exc.exception), "instance exception may not have a separate value") def test_not_throwable(self): with self.assertRaises(TypeError) as exc: _test_extension.test_throw_exact( greenlet.getcurrent(), "abc", None, None, ) self.assertEqual(str(exc.exception), "exceptions must be classes, or instances, not str") @ignores_leakcheck def test_leaks(self): from . import PY314 iters = 100 if PY314: expected_refs = [1] * iters else: expected_refs = [2] * iters for name, caller in ( ("test_switch", lambda: _test_extension.test_switch(greenlet.greenlet(object))), ("test_switch_kwargs", lambda: _test_extension.test_switch_kwargs(greenlet.greenlet(object))), ("test_new_greenlet", lambda: _test_extension.test_new_greenlet(object)), ): with self.subTest(name): results = [caller() for _ in range(iters)] refs = [ sys.getrefcount(i) - 1 # ignore ref in ``i`` for i in results ] self.assertEqual(refs, expected_refs) if __name__ == '__main__': import unittest unittest.main()

SIMPAN BATAL