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

jdufresne · encukou · commit 28528ea20114 · 2018-03-14T15:13:33.000+01:00
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"]
