summaryrefslogtreecommitdiffstats
path: root/tests/unit/test_jsonutils.py
diff options
context:
space:
mode:
authorRussell Bryant <rbryant@redhat.com>2012-07-18 16:15:52 -0400
committerRussell Bryant <rbryant@redhat.com>2012-07-18 16:15:52 -0400
commit2d6f84742a3e8ea51ebbfb82cbeacefe97e199d5 (patch)
treef35bda4f63f632f622d1c620d9ab6246aa22eeaa /tests/unit/test_jsonutils.py
parent1c1b36985be5a4c49b9cfc808edcdfd288c6d0cc (diff)
downloadoslo-2d6f84742a3e8ea51ebbfb82cbeacefe97e199d5.tar.gz
oslo-2d6f84742a3e8ea51ebbfb82cbeacefe97e199d5.tar.xz
oslo-2d6f84742a3e8ea51ebbfb82cbeacefe97e199d5.zip
Track to_primitive() depth after iteritems().
Change jsonutils.to_primitive() to increase the recursion depth counter when calling to_primitive() on the result of iteritems() from the current element. Previously, the only time the counter was increased was when converting the __dict__ from an object. The iteritems() case risks cycles, as well. I hit a problem with this when trying to call to_primitive on an instance of nova.db.sqlalchemy.models.Instance. An Instance includes a reference to InstanceInfoCache, which has a reference back to the Instance. Without this change, to_primitive() would raise an exception for an Instance due to excessive recursion. Related to nova blueprint no-db-messaging. Change-Id: Ifb878368d97e92ab6c361a4dd5f5ab2e68fc16e2
Diffstat (limited to 'tests/unit/test_jsonutils.py')
-rw-r--r--tests/unit/test_jsonutils.py18
1 files changed, 18 insertions, 0 deletions
diff --git a/tests/unit/test_jsonutils.py b/tests/unit/test_jsonutils.py
index fe25697..46b5b36 100644
--- a/tests/unit/test_jsonutils.py
+++ b/tests/unit/test_jsonutils.py
@@ -95,6 +95,24 @@ class ToPrimitiveTestCase(unittest.TestCase):
p = jsonutils.to_primitive(x)
self.assertEquals(p, {'a': 1, 'b': 2, 'c': 3})
+ def test_iteritems_with_cycle(self):
+ class IterItemsClass(object):
+ def __init__(self):
+ self.data = dict(a=1, b=2, c=3)
+ self.index = 0
+
+ def iteritems(self):
+ return self.data.items()
+
+ x = IterItemsClass()
+ x2 = IterItemsClass()
+ x.data['other'] = x2
+ x2.data['other'] = x
+
+ # If the cycle isn't caught, to_primitive() will eventually result in
+ # an exception due to excessive recursion depth.
+ p = jsonutils.to_primitive(x)
+
def test_instance(self):
class MysteryClass(object):
a = 10