Add test for recursive type

Thriftpy · Nov 22, 2016 · afd9acd · afd9acd
1 parent 4032763
commit afd9acd
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/tests/test_hook.py b/tests/test_hook.py
@@ -130,3 +130,10 @@ def test_load_slots():
     p_str = pickle.dumps(bob)
 
     assert pickle.loads(p_str) == bob
+
+    # works for recursive types too
+    rec = thriftpy.load('parser-cases/recursive_union.thrift', use_slots=True)
+    rec_slots = rec.Dynamic.__slots__
+    assert rec_slots == ['boolean', 'integer', 'doubl', 'str', 'arr', 'object']
+    with pytest.raises(AttributeError):
+        rec.Dynamic.attr_not_exist = "shouldn't work"