You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
81 lines
2.1 KiB
Python
81 lines
2.1 KiB
Python
5 years ago
|
# -*- coding: utf-8 -*-
|
||
|
|
# copyright 2018 gevent contributors. See LICENSE for details.
|
||
|
|
|
||
|
|
from __future__ import absolute_import
|
||
|
|
from __future__ import division
|
||
|
|
from __future__ import print_function
|
||
|
|
|
||
|
|
import gc
|
||
|
|
|
||
|
|
|
||
|
|
import gevent.testing as greentest
|
||
|
|
from gevent._ident import IdentRegistry
|
||
|
|
from gevent._compat import PYPY
|
||
|
|
|
||
|
|
class Target(object):
|
||
|
|
pass
|
||
|
|
|
||
|
|
class TestIdent(greentest.TestCase):
|
||
|
|
|
||
|
|
def setUp(self):
|
||
|
|
self.reg = IdentRegistry()
|
||
|
|
|
||
|
|
def tearDown(self):
|
||
|
|
self.reg = None
|
||
|
|
|
||
|
|
def test_basic(self):
|
||
|
|
target = Target()
|
||
|
|
self.assertEqual(0, self.reg.get_ident(target))
|
||
|
|
self.assertEqual(1, len(self.reg))
|
||
|
|
|
||
|
|
self.assertEqual(0, self.reg.get_ident(target))
|
||
|
|
self.assertEqual(1, len(self.reg))
|
||
|
|
|
||
|
|
target2 = Target()
|
||
|
|
self.assertEqual(1, self.reg.get_ident(target2))
|
||
|
|
self.assertEqual(2, len(self.reg))
|
||
|
|
|
||
|
|
self.assertEqual(1, self.reg.get_ident(target2))
|
||
|
|
self.assertEqual(2, len(self.reg))
|
||
|
|
|
||
|
|
self.assertEqual(0, self.reg.get_ident(target))
|
||
|
|
|
||
|
|
# When an object dies, we can re-use
|
||
|
|
# its id. Under PyPy we need to collect garbage first.
|
||
|
|
del target
|
||
|
|
if PYPY:
|
||
|
|
for _ in range(3):
|
||
|
|
gc.collect()
|
||
|
|
|
||
|
|
self.assertEqual(1, len(self.reg))
|
||
|
|
|
||
|
|
target3 = Target()
|
||
|
|
self.assertEqual(1, self.reg.get_ident(target2))
|
||
|
|
self.assertEqual(0, self.reg.get_ident(target3))
|
||
|
|
self.assertEqual(2, len(self.reg))
|
||
|
|
|
||
|
|
@greentest.skipOnPyPy("This would need to GC very frequently")
|
||
|
|
def test_circle(self):
|
||
|
|
keep_count = 3
|
||
|
|
keepalive = [None] * keep_count
|
||
|
|
|
||
|
|
for i in range(1000):
|
||
|
|
target = Target()
|
||
|
|
# Drop an old one.
|
||
|
|
keepalive[i % keep_count] = target
|
||
|
|
self.assertLessEqual(self.reg.get_ident(target), keep_count)
|
||
|
|
|
||
|
|
|
||
|
|
@greentest.skipOnPurePython("Needs C extension")
|
||
|
|
class TestCExt(greentest.TestCase):
|
||
|
|
|
||
|
|
def test_c_extension(self):
|
||
|
|
self.assertEqual(IdentRegistry.__module__,
|
||
|
|
'gevent.__ident')
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
greentest.main()
|