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

Update proxy keys on CaseInsensitiveDict.update()

2 files changed, 10 insertions, 0 deletions

diff --git a/oauthlib/common.py b/oauthlib/common.py
index 5d999b2..705cbd2 100644
--- a/oauthlib/common.py
+++ b/oauthlib/common.py
@@ -354,6 +354,11 @@ class CaseInsensitiveDict(dict):
         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
index 078b67b..4f95cd8 100644
--- a/tests/test_common.py
+++ b/tests/test_common.py
@@ -228,3 +228,8 @@ class CaseInsensitiveDictTest(TestCase):
         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')