Merge pull request #464 from cknave/case-insensitive-dict-update

Update proxy keys on CaseInsensitiveDict.update()
oauthlib · Mar 31, 2017 · e6c6ddb · e6c6ddb
2 parents 0573457 + 10308cc
commit e6c6ddb
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/oauthlib/common.py b/oauthlib/common.py
@@ -354,6 +354,11 @@ def __setitem__(self, k, v):
         super(CaseInsensitiveDict, self).__setitem__(k, v)
         self.proxy[k.lower()] = k
 
+    def update(self, *args, **kwargs):
+        super(CaseInsensitiveDict, self).update(*args, **kwargs)
+        for k in dict(*args, **kwargs):
+            self.proxy[k.lower()] = k
+
 
 class Request(object):
 

diff --git a/tests/test_common.py b/tests/test_common.py
@@ -228,3 +228,8 @@ def test_basic(self):
         del cid['c']
         self.assertEqual(cid['A'], 'b')
         self.assertEqual(cid['a'], 'b')
+
+    def test_update(self):
+        cid = CaseInsensitiveDict({})
+        cid.update({'KeY': 'value'})
+        self.assertEqual(cid['kEy'], 'value')