home *** CD-ROM | disk | FTP | other *** search
/ Maximum CD 2011 July / maximum-cd-2011-07.iso / DiscContents / LibO_3.3.2_Win_x86_install_multi.exe / libreoffice1.cab / test_gdbm.py < prev    next >
Encoding:
Python Source  |  2011-03-15  |  2.4 KB  |  83 lines
  1. import gdbm
  2. import unittest
  3. import os
  4. from test.test_support import verbose, TESTFN, run_unittest, unlink
  5.  
  6.  
  7. filename = TESTFN
  8.  
  9. class TestGdbm(unittest.TestCase):
  10.  
  11.     def setUp(self):
  12.         self.g = None
  13.  
  14.     def tearDown(self):
  15.         if self.g is not None:
  16.             self.g.close()
  17.         unlink(filename)
  18.  
  19.     def test_key_methods(self):
  20.         self.g = gdbm.open(filename, 'c')
  21.         self.assertEqual(self.g.keys(), [])
  22.         self.g['a'] = 'b'
  23.         self.g['12345678910'] = '019237410982340912840198242'
  24.         key_set = set(self.g.keys())
  25.         self.assertEqual(key_set, frozenset(['a', '12345678910']))
  26.         self.assert_(self.g.has_key('a'))
  27.         key = self.g.firstkey()
  28.         while key:
  29.             self.assert_(key in key_set)
  30.             key_set.remove(key)
  31.             key = self.g.nextkey(key)
  32.         self.assertRaises(KeyError, lambda: self.g['xxx'])
  33.  
  34.     def test_error_conditions(self):
  35.         # Try to open a non-existent database.
  36.         unlink(filename)
  37.         self.assertRaises(gdbm.error, gdbm.open, filename, 'r')
  38.         # Try to access a closed database.
  39.         self.g = gdbm.open(filename, 'c')
  40.         self.g.close()
  41.         self.assertRaises(gdbm.error, lambda: self.g['a'])
  42.         # try pass an invalid open flag
  43.         self.assertRaises(gdbm.error, lambda: gdbm.open(filename, 'rx').close())
  44.  
  45.     def test_flags(self):
  46.         # Test the flag parameter open() by trying all supported flag modes.
  47.         all = set(gdbm.open_flags)
  48.         # Test standard flags (presumably "crwn").
  49.         modes = all - set('fsu')
  50.         for mode in modes:
  51.             self.g = gdbm.open(filename, mode)
  52.             self.g.close()
  53.  
  54.         # Test additional flags (presumably "fsu").
  55.         flags = all - set('crwn')
  56.         for mode in modes:
  57.             for flag in flags:
  58.                 self.g = gdbm.open(filename, mode + flag)
  59.                 self.g.close()
  60.  
  61.     def test_reorganize(self):
  62.         self.g = gdbm.open(filename, 'c')
  63.         size0 = os.path.getsize(filename)
  64.  
  65.         self.g['x'] = 'x' * 10000
  66.         size1 = os.path.getsize(filename)
  67.         self.assert_(size0 < size1)
  68.  
  69.         del self.g['x']
  70.         # 'size' is supposed to be the same even after deleting an entry.
  71.         self.assertEqual(os.path.getsize(filename), size1)
  72.  
  73.         self.g.reorganize()
  74.         size2 = os.path.getsize(filename)
  75.         self.assert_(size1 > size2 >= size0)
  76.  
  77.  
  78. def test_main():
  79.     run_unittest(TestGdbm)
  80.  
  81. if __name__ == '__main__':
  82.     test_main()
  83.