+++ /dev/null
-import os\r
-import unittest\r
-import shelve\r
-import glob\r
-from test import test_support\r
-\r
-test_support.import_module('anydbm', deprecated=True)\r
-\r
-class TestCase(unittest.TestCase):\r
-\r
- fn = "shelftemp" + os.extsep + "db"\r
-\r
- def test_close(self):\r
- d1 = {}\r
- s = shelve.Shelf(d1, protocol=2, writeback=False)\r
- s['key1'] = [1,2,3,4]\r
- self.assertEqual(s['key1'], [1,2,3,4])\r
- self.assertEqual(len(s), 1)\r
- s.close()\r
- self.assertRaises(ValueError, len, s)\r
- try:\r
- s['key1']\r
- except ValueError:\r
- pass\r
- else:\r
- self.fail('Closed shelf should not find a key')\r
-\r
- def test_ascii_file_shelf(self):\r
- try:\r
- s = shelve.open(self.fn, protocol=0)\r
- s['key1'] = (1,2,3,4)\r
- self.assertEqual(s['key1'], (1,2,3,4))\r
- s.close()\r
- finally:\r
- for f in glob.glob(self.fn+"*"):\r
- os.unlink(f)\r
-\r
- def test_binary_file_shelf(self):\r
- try:\r
- s = shelve.open(self.fn, protocol=1)\r
- s['key1'] = (1,2,3,4)\r
- self.assertEqual(s['key1'], (1,2,3,4))\r
- s.close()\r
- finally:\r
- for f in glob.glob(self.fn+"*"):\r
- os.unlink(f)\r
-\r
- def test_proto2_file_shelf(self):\r
- try:\r
- s = shelve.open(self.fn, protocol=2)\r
- s['key1'] = (1,2,3,4)\r
- self.assertEqual(s['key1'], (1,2,3,4))\r
- s.close()\r
- finally:\r
- for f in glob.glob(self.fn+"*"):\r
- os.unlink(f)\r
-\r
- def test_in_memory_shelf(self):\r
- d1 = {}\r
- s = shelve.Shelf(d1, protocol=0)\r
- s['key1'] = (1,2,3,4)\r
- self.assertEqual(s['key1'], (1,2,3,4))\r
- s.close()\r
- d2 = {}\r
- s = shelve.Shelf(d2, protocol=1)\r
- s['key1'] = (1,2,3,4)\r
- self.assertEqual(s['key1'], (1,2,3,4))\r
- s.close()\r
-\r
- self.assertEqual(len(d1), 1)\r
- self.assertNotEqual(d1, d2)\r
-\r
- def test_mutable_entry(self):\r
- d1 = {}\r
- s = shelve.Shelf(d1, protocol=2, writeback=False)\r
- s['key1'] = [1,2,3,4]\r
- self.assertEqual(s['key1'], [1,2,3,4])\r
- s['key1'].append(5)\r
- self.assertEqual(s['key1'], [1,2,3,4])\r
- s.close()\r
-\r
- d2 = {}\r
- s = shelve.Shelf(d2, protocol=2, writeback=True)\r
- s['key1'] = [1,2,3,4]\r
- self.assertEqual(s['key1'], [1,2,3,4])\r
- s['key1'].append(5)\r
- self.assertEqual(s['key1'], [1,2,3,4,5])\r
- s.close()\r
-\r
- self.assertEqual(len(d1), 1)\r
- self.assertEqual(len(d2), 1)\r
-\r
- def test_writeback_also_writes_immediately(self):\r
- # Issue 5754\r
- d = {}\r
- s = shelve.Shelf(d, writeback=True)\r
- s['key'] = [1]\r
- p1 = d['key'] # Will give a KeyError if backing store not updated\r
- s['key'].append(2)\r
- s.close()\r
- p2 = d['key']\r
- self.assertNotEqual(p1, p2) # Write creates new object in store\r
-\r
-\r
-from test import mapping_tests\r
-\r
-class TestShelveBase(mapping_tests.BasicTestMappingProtocol):\r
- fn = "shelftemp.db"\r
- counter = 0\r
- def __init__(self, *args, **kw):\r
- self._db = []\r
- mapping_tests.BasicTestMappingProtocol.__init__(self, *args, **kw)\r
- type2test = shelve.Shelf\r
- def _reference(self):\r
- return {"key1":"value1", "key2":2, "key3":(1,2,3)}\r
- def _empty_mapping(self):\r
- if self._in_mem:\r
- x= shelve.Shelf({}, **self._args)\r
- else:\r
- self.counter+=1\r
- x= shelve.open(self.fn+str(self.counter), **self._args)\r
- self._db.append(x)\r
- return x\r
- def tearDown(self):\r
- for db in self._db:\r
- db.close()\r
- self._db = []\r
- if not self._in_mem:\r
- for f in glob.glob(self.fn+"*"):\r
- test_support.unlink(f)\r
-\r
-class TestAsciiFileShelve(TestShelveBase):\r
- _args={'protocol':0}\r
- _in_mem = False\r
-class TestBinaryFileShelve(TestShelveBase):\r
- _args={'protocol':1}\r
- _in_mem = False\r
-class TestProto2FileShelve(TestShelveBase):\r
- _args={'protocol':2}\r
- _in_mem = False\r
-class TestAsciiMemShelve(TestShelveBase):\r
- _args={'protocol':0}\r
- _in_mem = True\r
-class TestBinaryMemShelve(TestShelveBase):\r
- _args={'protocol':1}\r
- _in_mem = True\r
-class TestProto2MemShelve(TestShelveBase):\r
- _args={'protocol':2}\r
- _in_mem = True\r
-\r
-def test_main():\r
- test_support.run_unittest(\r
- TestAsciiFileShelve,\r
- TestBinaryFileShelve,\r
- TestProto2FileShelve,\r
- TestAsciiMemShelve,\r
- TestBinaryMemShelve,\r
- TestProto2MemShelve,\r
- TestCase\r
- )\r
-\r
-if __name__ == "__main__":\r
- test_main()\r