cidict: Make iteration over cidict yield same values as keys()

ECN · Mar 14, 2018 · 28528ea · 28528ea
1 parent 60b761b
commit 28528ea
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/Lib/ldap/cidict.py b/Lib/ldap/cidict.py
@@ -44,6 +44,9 @@ def has_key(self,key):
   def __contains__(self,key):
     return IterableUserDict.__contains__(self, key.lower())
 
+  def __iter__(self):
+    return iter(self.keys())
+
   def keys(self):
     return self._keys.values()
 

diff --git a/Tests/t_cidict.py b/Tests/t_cidict.py
@@ -36,6 +36,8 @@ def test_cidict(self):
         self.assertEqual(cix.get("xyz", None), 987)
         cix_keys = sorted(cix.keys())
         self.assertEqual(cix_keys, ['AbCDeF','xYZ'])
+        cix_keys = sorted(cix)
+        self.assertEqual(cix_keys, ['AbCDeF','xYZ'])
         cix_items = sorted(cix.items())
         self.assertEqual(cix_items, [('AbCDeF',123), ('xYZ',987)])
         del cix["abcdEF"]