commit 732324a3f86a352217b01ae2438b6db7691ae0b1
author Georg Brandl <georg@python.org> Sat Dec 04 11:12:43 2010 +0000
committer Georg Brandl <georg@python.org> Sat Dec 04 11:12:43 2010 +0000
tree 8751b652a09b2fe8ac354edd571c431210dda3b2
parent c9fb3c64177f8efefd18a11f86bcf2f94d0f7c90

#7905: Actually respect the keyencoding parameter to shelve.Shelf.

Lib/test/test_shelve.py

diff --git a/Lib/test/test_shelve.py b/Lib/test/test_shelve.py
index 066208a..3e73f52 100644
--- a/Lib/test/test_shelve.py
+++ b/Lib/test/test_shelve.py
@@ -122,6 +122,19 @@
         self.assertEqual(len(d1), 1)
         self.assertEqual(len(d2), 1)
 
+    def test_keyencoding(self):
+        d = {}
+        key = 'Pöp'
+        # the default keyencoding is utf-8
+        shelve.Shelf(d)[key] = [1]
+        self.assertIn(key.encode('utf-8'), d)
+        # but a different one can be given
+        shelve.Shelf(d, keyencoding='latin1')[key] = [1]
+        self.assertIn(key.encode('latin1'), d)
+        # with all consequences
+        s = shelve.Shelf(d, keyencoding='ascii')
+        self.assertRaises(UnicodeEncodeError, s.__setitem__, key, [1])
+
     def test_writeback_also_writes_immediately(self):
         # Issue 5754
         d = {}