[pypy-commit] pypy hpy: add unit test for handles.py

[antocuni]

Author: Antonio Cuni <anto.cuni at gmail.com>
Branch: hpy
Changeset: r98088:3308dc2cfbc4
Date: 2019-11-17 00:49 +0100
http://bitbucket.org/pypy/pypy/changeset/3308dc2cfbc4/

Log:	add unit test for handles.py

diff --git a/pypy/module/hpy_universal/test/test_handles.py b/pypy/module/hpy_universal/test/test_handles.py
new file mode 100644
--- /dev/null
+++ b/pypy/module/hpy_universal/test/test_handles.py
@@ -0,0 +1,31 @@
+from pypy.module.hpy_universal.handles import HandleManager
+
+class TestHandleManager(object):
+
+    def test_new(self):
+        mgr = HandleManager(None)
+        h = mgr.new('hello')
+        assert mgr.handles_w[h] == 'hello'
+
+    def test_consume(self):
+        mgr = HandleManager(None)
+        h = mgr.new('hello')
+        assert mgr.consume(h) == 'hello'
+        assert mgr.handles_w[h] is None
+
+    def test_freelist(self):
+        mgr = HandleManager(None)
+        h0 = mgr.new('hello')
+        h1 = mgr.new('world')
+        assert mgr.consume(h0) == 'hello'
+        assert mgr.free_list == [h0]
+        h2 = mgr.new('hello2')
+        assert h2 == h0
+        assert mgr.free_list == []
+
+    def test_dup(self):
+        mgr = HandleManager(None)
+        h0 = mgr.new('hello')
+        h1 = mgr.dup(h0)
+        assert h1 != h0
+        assert mgr.consume(h0) == mgr.consume(h1) == 'hello'